arXiv:2410.01228cs.DCcs.LG2024-10被引 9

让大模型在线与离线任务共享显卡,提升吞吐量并保证响应速度。

ConServe: Fine-Grained GPU Harvesting for LLM Online and Offline Co-Serving

  • 按令牌级别调度,精准控制离线任务占用时间。
  • 支持层间微秒级抢占,离线任务可快速让出资源。
  • 离线任务暂停恢复几乎零开销,适合高负载场景。

大语言模型(LLM)服务需低延迟和高吞吐,但负载波动大导致难以高效利用GPU。本文发现将对延迟敏感的在线请求与容忍延迟的离线任务(如模型评测)协同处理具有潜力,但现有系统因请求或迭代粒度粗,无法在不干扰在线延迟的前提下捕获毫秒级空闲周期。ConServe 是一种新的共服务系统,通过细粒度资源管理实现高吞吐和强在线延迟保障。其核心包含三项技术:(1)基于延迟感知的令牌级调度器,精确调节离线批大小与令牌数以满足在线延迟目标;(2)子迭代、逐层预占机制,使离线任务可在在线负载突增时快速让出资源;(3)增量式KV缓存管理,实现离线任务近乎零成本的暂停与恢复。在真实负载下使用 Llama-3.1 与 Qwen-2.5 模型评估表明,相比当前最优系统,ConServe 平均吞吐提升 2.2 倍,线上服务尾延迟平均降低 2.9 倍。

原文摘要 · Abstract (English)

Large language model (LLM) serving demands low latency and high throughput, but high load variability makes it challenging to achieve high GPU utilization. In this paper, we identify a synergetic but overlooked opportunity to co-serve latency-critical online requests alongside latency-tolerant offline tasks such as model benchmarking. While promising, existing serving systems fail to co-serve them efficiently, as their coarse-grained resource management at the request or iteration level cannot harvest millisecond-level GPU idle cycles without introducing interference that violates online latency objectives. ConServe is a new LLM co-serving system that achieves high throughput and strong online latency guarantees by managing resources at finer granularities. ConServe introduces three techniques: (1) a latency-aware token-level scheduler that precisely sizes offline batches and tokens to fit within online latency objectives; (2) sub-iteration, layer-wise preemption that allows offline tasks to yield to online load spikes; and (3) incremental KV cache management that enables preempting and resuming offline requests at near-zero cost. Evaluations with Llama-3.1 and Qwen-2.5 models on real-world workloads show that ConServe delivers an average of 2.2$\times$ higher throughput and reduces online serving tail latency by 2.9$\times$ on average compared to state-of-the-art systems.

大模型推理资源调度延迟优化

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