English

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

Artificial Intelligence 2026-03-04 v1 Machine Learning

Abstract

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.

Keywords

Cite

@article{arxiv.2603.02599,
  title  = {SUN: Shared Use of Next-token Prediction for Efficient Multi-LLM Disaggregated Serving},
  author = {Sunghyeon Woo and Ahreum Seo and Jaegwang Lee and Jaeeun Kil and Hanbae Seo and Joonghoon Kim and Baeseong Park and Se Jung Kwon and Dongsoo Lee},
  journal= {arXiv preprint arXiv:2603.02599},
  year   = {2026}
}

Comments

Preprint, 15 pages, 5 figures

R2 v1 2026-07-01T11:00:25.587Z