中文

Lagom:解放通信与计算重叠的力量用于分布式 LLM 训练

分布式、并行与集群计算 2026-02-25 v1

摘要

在分布式大模型训练中,计算与通信的重叠至关重要, yet optimizing it——尤其是当计算成为瓶颈时——仍然具有挑战性。我们提出了 Lagom—a一种在计算和通信之间协调资源使用的系统。通过引入统一的成本模型和基于优先级的搜索算法,Lagom 将优化复杂度从指数级降低到线性级。在高和低带宽 GPU 集群上的评估表明,Lagom 在各种模型和并行化方式下,相较于 NCCL 和 AutoCCL 实现了 1.07-1.33 倍和 1.03-1.27 倍的加速。

关键词

引用

@article{arxiv.2602.20656,
  title  = {Lagom: Unleashing the Power of Communication and Computation Overlapping for Distributed LLM Training},
  author = {Guanbin Xu and ZhenGuo Xu and Yuzhe Li and Youhui Bai and Ping Gong and Chaoyi Ruan and Cheng Li},
  journal= {arXiv preprint arXiv:2602.20656},
  year   = {2026}
}

备注

6 pages, 8 figures