arXiv:2604.24013cs.LGcs.AI2026-04

提出CommFuse方法,消除分布式大模型训练中的通信尾延迟。

CommFuse: Hiding Tail Latency via Communication Decomposition and Fusion for Distributed LLM Training

论文配图:CommFuse: Hiding Tail Latency via Communication Decomposition and Fusion for Distributed LLM Training
图 1 · 摘自论文原文
  • 将集体通信拆分为点对点通信,实现细粒度计算-通信重叠。
  • 在多种并行策略下,尾延迟归零,模型浮点利用率提升显著。
  • 适合大规模分布式训练,尤其对数据并行和张量并行场景有效。

大型语言模型规模的快速扩张要求将计算任务分布在GPU、TPU等加速器上。然而,这种并行化策略带来显著的数据通信开销,严重制约计算效率。尽管计算-通信重叠是潜在解决方案,但现有基于数据切分的方法仍存在尾延迟问题。为此,本文提出一种新方法CommFuse,通过用分解后的点对点(P2P)通信替代传统的reduce-scatter和all-gather操作,并调度分块计算,实现细粒度重叠,从而精确消除主流重叠方法中的尾延迟。该方法可有效缓解张量并行与数据并行的通信瓶颈,适用于数据并行及多种张量级并行策略(如TPSP和UP)。实验表明,该技术在各类场景下均实现更低延迟、更高模型浮点运算利用率(MFU)和更高吞吐量。

原文摘要 · Abstract (English)

The rapid growth in the size of large language models has necessitated the partitioning of computational workloads across accelerators such as GPUs, TPUs, and NPUs. However, these parallelization strategies incur substantial data communication overhead significantly hindering computational efficiency. While communication-computation overlap presents a promising direction, existing data slicing based solutions suffer from tail latency. To overcome this limitation, this research introduces a novel communication-computation overlap technique to eliminate this tail latency in state of the art overlap methods for distributed LLM training. The aim of this technique is to effectively mitigate communication bottleneck of tensor parallelism and data parallelism for distributed training and inference. In particular, we propose a novel method termed CommFuse that replaces conventional collective operations of reduce-scatter and all-gather with decomposed peer-to-peer (P2P) communication and schedules partitioned computations to enable fine-grained overlap. Our method provides an exact algorithm for reducing communication overhead that eliminates tail latency. Moreover, it presents a versatile solution compatible with data-parallel training and various tensor-level parallelism strategies, including TPSP and UP. Experimental evaluations demonstrate that our technique consistently achieves lower latency, superior Model FLOPS Utilization (MFU), and high throughput.

分布式训练通信优化大模型

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。