English

A Universal Load Balancing Principle and Its Application to Large Language Model Serving

Distributed, Parallel, and Cluster Computing 2026-02-03 v2 Machine Learning

Abstract

Over 40% of computational power in Large Language Model (LLM) serving systems can be systematically wasted - not from hardware limits, but from load imbalance in barrier-synchronized parallel processing. When progress is gated by the slowest worker at each step, heterogeneous and evolving workloads create persistent stragglers; faster workers idle while drawing power, producing nothing. In large language model inference alone, this translates to gigawatt-hours of wasted electricity daily. Here we develop a universal load-balancing principle for barrier-synchronized systems with non-migratable state. We prove worst-case theoretical guarantees: imbalance reduction grows with system scale, and the resulting energy savings can exceed 52% for modern hardware at fleet scale. Experiments corroborate the theory, demonstrating 28% energy reduction alongside substantial throughput and latency improvements. Formulated as an online integer optimization with provable guarantees, the principle extends beyond LLM serving to broad classes of barrier-synchronized parallel systems, establishing a theoretical foundation for sustainable high-performance computing.

Keywords

Cite

@article{arxiv.2601.17855,
  title  = {A Universal Load Balancing Principle and Its Application to Large Language Model Serving},
  author = {Zixi Chen and Tianci Bu and Chendong Song and Xin Lu and Yinyu Ye and Zijie Zhou},
  journal= {arXiv preprint arXiv:2601.17855},
  year   = {2026}
}
R2 v1 2026-07-01T09:19:12.621Z