Towards Load-Aware Prefill Deflection for Disaggregated LLM Serving
Summary
A proactive prefill-deflecting scheduler for disaggregated LLM serving addresses the common issue of prefill node saturation and underutilized decode nodes under bursty workloads. In a production-style A100 cluster with 2 prefill and 2 decode nodes (2P2D), 77-98% of P95 Time-to-First-Token (TTFT) is attributed to queuing and inter-node KV-cache transfer, with prefill execution accounting for only 2-23%. The proposed scheduler enables decode nodes to serve prefill phases of requests as chunked steps, interleaved with their in-flight decode batches. It estimates TTFT on prefill nodes and identifies optimal chunk schedules on decode nodes that maintain Time-Between-Tokens (TBT) SLOs for existing decodes. Deflection occurs if the decode path improves tail latency, eliminating inter-node KV transfer. Implemented on vLLM and evaluated with DeepSeek-V2-Lite on production traces, this approach reduces P95 TTFT by up to 81% and raises SLO attainment by up to 79%, incurring a sub-millisecond routing cost.
Key takeaway
For MLOps Engineers managing disaggregated LLM serving performance, especially under bursty workloads, your current disaggregated LLM serving setup likely suffers from high P95 TTFT due to prefill node saturation and KV-cache transfer. Implement a prefill-deflecting scheduler to offload prefill tasks to underutilized decode nodes. This strategy can significantly reduce tail latency and improve service level objective attainment by up to 79%, while minimizing inter-node data transfer.
Key insights
Disaggregated LLM serving benefits from deflecting prefill tasks to underutilized decode nodes to reduce tail latency and KV-cache transfer.
Principles
- Prefill phase often bottlenecks disaggregated LLM serving.
- Interleaving prefill with decode batches improves efficiency.
- Proactive scheduling can mitigate tail latency.
Method
Estimate request TTFT on prefill nodes. Search decode nodes for largest chunk schedule maintaining TBT SLOs. Deflect prefill to decode if it improves tail latency, eliminating inter-node KV transfer.
In practice
- Implement prefill deflection in vLLM-based systems.
- Monitor P95 TTFT and SLO attainment for improvements.
- Consider chunked prefill for bursty LLM workloads.
Topics
- Disaggregated LLM Serving
- Prefill Deflection
- Tail Latency Optimization
- vLLM
- DeepSeek-V2-Lite
- GPU Scheduling
Best for: AI Scientist, Research Scientist, AI Engineer, Machine Learning Engineer, MLOps Engineer
Related on AIssential
Editorial summary, takeaway, and curation by AIssential. Original article published by Artificial Intelligence.