arXiv:2412.18169cs.DCcs.AI2024-12被引 7

通过丢弃冗余参数快速释放显存,显著降低大模型服务延迟

KunServe: Parameter-centric Memory Management for Efficient Memory Overloading Handling in LLM Serving

  • 基于参数冗余特性,选择性丢弃副本释放显存,而非传统缓存管理
  • 在负载高峰时,尾部请求的首次令牌延迟降低72.2倍
  • 适合高并发大模型服务场景,尤其对低延迟要求严格的系统

当前大模型集群服务需满足严格延迟SLO,但其状态化特性导致需在有限显存中维护大量键值缓存(KVCache)。真实负载突增时,显存易被占满,引发排队,导致响应延迟上升数个数量级。现有基于KVCache的处理方法(如丢弃、迁移、交换)无法快速释放足够内存。本文提出首个以参数为中心的显存过载应对方案:利用大模型服务中参数常跨GPU冗余复制的特性,选择性丢弃副本以瞬时释放内存,使所有请求可在更大批次下无队列执行。为保证正确与高效,系统采用流水线并行在拥有完整参数副本的GPU上协同执行,并设计无需额外协作的合理丢弃策略。同时优化了请求执行模式下的流水线性能开销。实验表明,相比Llumnix、vLLM和InferCept等先进系统,该方案在负载压力下将请求尾部首次令牌传输时间(TTFT)降低最多达72.2倍。

原文摘要 · Abstract (English)

Serving LLMs with a cluster of GPUs is common nowadays, where the serving system must meet strict latency SLOs required by applications. However, the stateful nature of LLM serving requires maintaining huge states (i.e., KVCache) in limited GPU memory. Under spikes in real-world workloads, GPU memory can be easily throttled, leading to orders of magnitude higher response latency due to queuing introduced by waiting for KVCache to be reclaimed. Prior KVCache-centric approaches handle load throttling by dropping, migrating, or swapping KVCache. These methods fail to release sufficient memory quickly with requests still queued. This paper proposes the first parameter-centric approach to handling throttling by selectively dropping replicated parameters to instantly free memory for requests, based on an unnoticed observation that model parameters are commonly replicated across GPUs for serving LLMs. With additional memory, all requests can be served with a larger batch without queuing. To make the parameter-centric approach correct and efficient, we cooperatively execute requests on GPUs with a complete copy of parameters using pipeline parallelism, and derive an appropriate drop plan without unnecessary cooperation. We also design techniques to minimize the performance overhead due to pipeline parallelism with the execution patterns of requests under drop. Evaluations show that {\sys} reduces the tail TTFT of requests under throttling by up to 72.2 times compared to the state-of-the-art systems including Llumnix, vLLM and InferCept.

大模型服务显存管理参数冗余低延迟

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。