用随机预测调度提升大模型服务负载均衡,减少延迟和预中断。
Astrolabe: Balancing Load in LLM Serving with Randomized Prediction-Guided Scheduling
- 结合预测长度与实例模拟,用双选择策略分配请求
- 比基准快31.6 QPS,首令牌延迟降低16%~77%
- 适合高并发、突发流量场景,降低资源开销
本文提出Astrolabe,一种用于多实例大语言模型服务的一次性请求调度器。它通过响应长度估计、基于实例模拟的延迟预测和幂次二选择调度策略,在不依赖迁移重平衡的前提下实现负载均衡,避免了因KV缓存传输带来的高开销与网络竞争。在默认的Llama-2-7B/ShareGPT配置下,其SLO容量达到31.6 QPS(对比最佳基线31.5 QPS),平均首令牌时间(TTFT)降低8%~36%,P99 TTFT降低16%~77%,平均端到端(E2E)延迟降低最多5.6%,达到容量后预中断次数减少约6倍。在配置变化下,对Qwen2-7B的SLO容量提升达6%,紧批量下提升7.1%;突发请求下平均E2E延迟降低6%~9%;相比全扇出,每预测器CPU使用量减少约2.8倍。在启用迁移的A100 GPU上,吞吐量最高较Llumnix提升2.6倍,饱和时单令牌延迟降低数个数量级。
原文摘要 · Abstract (English)
This paper presents Astrolabe, a randomized prediction-guided scheduler for one-shot request dispatch in multi-instance large language model (LLM) serving. Astrolabe improves load balancing without relying on migration-based rebalancing, whose KV-cache transfers can introduce substantial overhead and network contention under high load. It combines response-length estimation, per-instance simulation-based latency prediction, and a power-of-two-choices dispatch policy to balance load while avoiding request herding. On the default Llama-2-7B/ShareGPT setup, Astrolabe matches the SLO capacity of the best load-aware baseline (31.6 versus 31.5 QPS), while reducing mean time-to-first-token (TTFT) by 8 to 36 percent, P99 TTFT by 16 to 77 percent, and mean end-to-end (E2E) latency by up to 5.6 percent, with approximately six times fewer preemptions once capacity is reached. Under configuration shifts, Astrolabe improves SLO capacity by up to 6 percent on Qwen2-7B and 7.1 percent under tight batching, reduces mean E2E latency by 6 to 9 percent under bursty arrivals, and achieves an approximately 2.8-fold reduction in per-predictor CPU usage relative to full fanout. With migration enabled on A100 GPUs, Astrolabe outperforms Llumnix by up to 2.6 times in throughput while achieving orders-of-magnitude lower per-token latency at saturation.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。