arXiv:2604.12171cs.DCcs.LG2026-04被引 1

实现大模型推理中无需停机的动态流水线并行重构

PipeLive: Efficient Live In-place Pipeline Parallelism Reconfiguration for Dynamic LLM Serving

论文配图:PipeLive: Efficient Live In-place Pipeline Parallelism Reconfiguration for Dynamic LLM Serving
图 1 · 摘自论文原文
  • 设计新缓存布局与增量补丁机制,支持运行时重配置
  • 重构耗时从秒级降至10ms内,首字延迟降低54.7%
  • 适合云服务、异构硬件等动态部署场景

流水线并行广泛用于将大语言模型分层部署于多GPU上,以实现大规模模型的可扩展推理。然而,现有系统依赖静态配置,在无服务器平台和异构GPU环境下难以适应。重新配置需停止并重启服务,造成高昂停机成本,因此必须实现不停机、原地重配置。但这一过程面临根本挑战:GPU已满载模型权重与键值缓存,难以容纳新层分配,且需调整缓存大小,与vLLM等预分配机制冲突。同时,运行中保持键值一致性困难:全量复制导致长暂停,后台同步又可能因状态变化引发不一致。本文提出PipeLive,实现低干扰的实时原地流水线并行重构。其引入新型键值缓存布局,并结合改进的PageAttention,形成统一的运行时缓存重缩放机制;进一步采用受虚拟机迁移启发的增量键值补丁技术,同步源与目标配置间的状态,并识别安全切换点。实验表明,相比禁用缓存重缩放,管道重建时间减少2.5倍,首字延迟下降54.7%;相比无补丁方案,重构开销由秒级降至10ms以下,首字延迟与每输出字延迟分别提升54.7%和14.7%。

原文摘要 · Abstract (English)

Pipeline parallelism (PP) is widely used to partition layers of large language models (LLMs) across GPUs, enabling scalable inference for large models. However, existing systems rely on static PP configurations that fail to adapt to dynamic settings, such as serverless platforms and heterogeneous GPU environments. Reconfiguring PP by stopping and redeploying service incurs prohibitive downtime, so reconfiguration must instead proceed live and in place, without interrupting inference. However, live in-place PP reconfiguration is fundamentally challenging. GPUs are already saturated with model weights and KV cache, leaving little room for new layer placements and necessitating KV cache resizing, at odds with systems like vLLM that preallocate for throughput. Moreover, maintaining KV consistency during execution is difficult: stop-and-copy introduces large pauses, while background synchronization risks inconsistency as states evolve. We present PipeLive, which enables live in-place PP reconfiguration with minimal disruption. PipeLive introduces a redesigned KV cache layout together with a co-designed extension to PageAttention, forming a unified mechanism for live KV resizing. It further adopts an incremental KV patching mechanism, inspired by live virtual machine migration, to synchronize KV states between source and target configurations and identify a safe switch point. PipeLive achieves a 2.5X reduction in time-to-first-token (TTFT) without KV cache overflow compared to disabling KV resizing. Furthermore, compared to a variant without KV patching, it reduces reconfiguration overhead from seconds to under 10ms, and improves TTFT and time-per-output-token (TPOT) by up to 54.7% and 14.7%, respectively.

大模型推理流水线并行动态调度缓存优化

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