arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2603.02599cs.AIcs.LG
收藏

SUN: 为高效多LLM解耦服务实现下一个token预测的共享使用

SUN: Shared Use of Next-token Prediction for Efficient Multi-LLM Disaggregated Serving

Department of XXX, University of YYY, Location, Country(XXX系,YYY大学,地点,国家);School of ZZZ, Institute of WWW, Location, Country(ZZZ学院,WWW研究所,地点,国家)

Sunghyeon Woo, Ahreum Seo, Jaegwang Lee, Jaeeun Kil, Hanbae Seo, Joonghoon Kim, Baeseong Park, Se Jung Kwon, Dongsoo Lee

AI总结:

SUN通过共享解码模块提升多LLM服务效率,实现高吞吐量与低TPOT,支持低精度解码。

摘要:

在多模型LLM服务中,解码执行仍然效率低下,因为模型特定的资源划分:由于跨模型批处理不可行,内存受限的解码常遭受严重的GPU利用率不足,尤其是在负载倾斜的情况下。我们提出了共享使用下一个token预测(SUN),这是首个能够在解耦多LLM服务中实现跨模型解码共享的方法。SUN将解码器-only Transformer分解为prefill模块和decode模块,并仅微调任务特定的prefill模块,使冻结的decode模块能够在模型间共享。这种设计使解码路由策略能够平衡共享工人之间的解码请求以最大化利用率。在多样化的任务和模型家族中,SUN在保持系统吞吐量的同时,其准确性与全微调相当,且使用更少的解码工人。特别是,SUN在传统解耦方案上,每GPU吞吐量提高了2.0倍,同时保持时间每输出token(TPOT)在5%以内。SUN本质上使低精度解码成为可能;使用量化SUN(QSUN)时,其准确度与SUN相当,同时实现45%的速度提升,同时保留共享解码的优势。

英文摘要:

In multi-model LLM serving, decode execution remains inefficient due to model-specific resource partitioning: since cross-model batching is not possible, memory-bound decoding often suffers from severe GPU underutilization, especially under skewed workloads. We propose Shared Use of Next-token Prediction (SUN), the first approach that enables cross-model sharing of decode execution in disaggregated multi-LLM serving. SUN decomposes a decoder-only Transformer into a prefill module and a decode module, and fine-tunes only the task-specific prefill module, enabling a frozen decode module to be shared across models. This design enables a model-agnostic decode routing policy that balances decode requests across shared workers to maximize utilization. Across diverse tasks and model families, SUN achieves accuracy comparable to full fine-tuning while maintaining system throughput with fewer decode workers. In particular, SUN improves throughput per GPU by up to 2.0x over conventional disaggregation while keeping time-per-output-token (TPOT) within 5%. SUN inherently enables and facilitates low-bit decoding; with Quantized SUN (QSUN), it achieves a 45% speedup with comparable accuracy to SUN while preserving the benefits of shared decoding.

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

↑