arXiv:2511.09741cs.LGcs.AI2025-11中稿 · AAAI被引 1

TawPipe通过感知设备拓扑优化权重流水并行,显著提升长上下文大模型训练效率。

TawPipe: Topology-Aware Weight Pipeline Parallelism for Accelerating Long-Context Large Models Training

  • 按设备拓扑分组,减少跨节点通信流量
  • 固定权重分片避免重复传输,提升带宽利用率
  • 通信与计算重叠,隐藏延迟,适合大规模长序列训练

大语言模型训练受限于设备内存不足和跨设备通信开销。虽然流水线并行通过跨设备划分模型缓解内存压力,但其激活值通信开销随序列长度线性增长,制约长上下文训练效率。现有权重传递方法(如WeiPipe)虽传输权重而非激活值以降低开销,但仍存在冗余点对点传输和节点内带宽未充分利用问题。本文提出TawPipe——拓扑感知权重流水并行,利用分布式集群的层次化带宽优势提升通信效率:(i) 根据设备拓扑分组,优化节点内集体通信与节点间点对点通信;(ii) 为每个设备分配固定的模型权重与梯度分片,避免冗余传输;(iii) 将通信与计算重叠,隐藏延迟。与全分片数据并行(FSDP)使用的全局集体操作不同,TawPipe将大部分通信限制在节点内部,显著降低跨节点流量。在最多24块GPU上使用LLaMA风格模型的大量实验表明,TawPipe相比当前最优基线实现更高吞吐量和可扩展性。

原文摘要 · Abstract (English)

Training large language models (LLMs) is fundamentally constrained by limited device memory and costly inter-device communication. Although pipeline parallelism alleviates memory pressure by partitioning models across devices, it incurs activation communication overhead that scales linearly with sequence length, limiting efficiency in long-context training. Recent weight-passing approaches (e.g., WeiPipe) mitigate this by transmitting model weights instead of activations, but suffer from redundant peer-to-peer (P2P) transfers and underutilized intra-node bandwidth. We propose TawPipe--topology-aware weight pipeline parallelism, which exploits hierarchical bandwidth in distributed clusters for improved communication efficiency. TawPipe: (i) groups devices based on topology to optimize intra-node collective and inter-node P2P communication; (ii) assigns each device a fixed shard of model weights and gradients, avoiding redundant transfers; and (iii) overlaps communication with computation to hide latency. Unlike global collective operations used in fully sharded data parallelism (FSDP), TawPipe confines most communication within node boundaries, significantly reducing cross-node traffic. Extensive experiments on up to 24 GPUs with LLaMA-style models show that TawPipe achieves superior throughput and scalability compared to state-of-the-art baselines.

大模型训练并行计算通信优化长序列

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