arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2602.14516cs.DC
收藏

高效多轮LLM推理的解耦服务

Efficient Multi-round LLM Inference over Disaggregated Serving

Shanghai Jiao Tong University(上海交通大学);Southeast University(东南大学);University of Cambridge(剑桥大学);Peking University(北京大学);OceanBase, Ant Group(蚂蚁集团 OceanBase);Computational Social Science, Peking University (Qingdao)(北京大学(青岛)计算社会科学)

Wenhao He, Youhe Jiang, Penghao Zhao, Quanqing Xu, Eiko Yoneki, Bin Cui, Fangcheng Fu

AI总结:

AMPD通过自适应协调预填充和解码工作负载,提升多轮LLM推理的服务级别目标达成率。

摘要:

随着大型语言模型(LLMs)的快速发展,多轮工作流(如自主代理和迭代检索)变得越来越普遍。然而,这在预填充-解码(PD)解耦服务下服务LLMs时提出了挑战,这是一种广泛采用的范式,将计算密集型预填充阶段和内存密集型解码阶段分别分配到独立资源上。具体来说,现有系统忽略了多轮推理中交织的预填充-解码工作负载模式,导致对增量预填充工作负载和两个阶段的模型部署处理不 optimal。在本工作中,我们提出了AMPD,一种新的解耦服务框架,用于多轮LLM推理。AMPD的核心是根据实时工作负载协调预填充工作负载,通过自适应确定这些工作负载在哪里执行以及如何调度,以最大化服务级别目标(SLO)的达成。此外,我们为我们的场景定制了一个规划算法,有助于推导出两个阶段的最佳资源分配和并行策略。实证结果表明,AMPD相比最先进的基线显著提高了SLO的达成。

英文摘要:

With the rapid evolution of Large Language Models (LLMs), multi-round workflows, such as autonomous agents and iterative retrieval, have become increasingly prevalent. However, this raises hurdles for serving LLMs under prefill-decode (PD) disaggregation, a widely adopted paradigm that separates the compute-bound prefill phase and memory-bound decode phase onto individual resources. Specifically, existing systems overlook the interleaved prefill-decode workload pattern in multi-round inference, leading to sub-optimal handling of the incremental prefill workloads and model deployment for the two phases. In this work, we present AMPD, a brand new disaggregated serving framework for multi-round LLM inference. The core of AMPD is to coordinate the prefill workloads based on real-time workloads by adaptively determining where to carry out these workloads and how they are scheduled, in order to maximize service level objective (SLO) attainment. In addition, we tailor a planning algorithm for our scenario, facilitating the deduction of optimal resource allocation and parallel strategies for the two phases. Empirical results demonstrate that AMPD substantially improves SLO attainment compared to state-of-the-art baselines.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑