arXiv:2606.07881cs.LG2026-06

提出无气泡异步流水线训练方法,有效控制权重不一致

Breaking the Bubble: Asynchronous Pipeline Parallel Training with Bounded Weight Inconsistency

论文配图:Breaking the Bubble: Asynchronous Pipeline Parallel Training with Bounded Weight Inconsistency
图 1 · 摘自论文原文
  • 用局部梯度累积控制权重版本漂移
  • 训练速度提升1.69倍,内存与同步方法相当
  • 适合大规模语言模型高效训练

流水线并行是训练大模型的关键技术,但现有方法在吞吐量、内存和优化一致性间权衡困难。同步流水线虽保证权重一致性但存在气泡;异步流水线消除气泡却引入版本不一致,通常需权重暂存、预测或修正机制。本文提出PACI(带可控不一致的异步流水线训练),在无需权重暂存、预测、额外参数副本或全局同步的前提下,有效限制前向/反向权重版本漂移。核心思想是利用局部梯度累积作为版本控制机制:通过减缓参数版本更新速度以匹配流水线延迟,使任意微批次跨越的优化器更新次数受限,同时保持稳定利用率。在GPT风格语言模型预训练中,PACI在稳定性与最终困惑度上媲美同步1F1B-flush方法,峰值内存不变,实现完全流水线吞吐利用率,并相较最快刷新基线提升训练时间-精度效率达1.69倍。结果表明,只要显式控制前向/反向不一致,其可被安全用于换取显著效率增益。

原文摘要 · Abstract (English)

Pipeline parallelism is essential for training large neural networks, but existing schedules trade off throughput, memory, and optimization consistency. Synchronous pipelines preserve forward/backward weight consistency but suffer from bubbles; asynchronous pipelines remove bubbles but introduce weight-version mismatch, typically requiring weight stashing, prediction, or correction mechanisms. We introduce PACI (Pipeline Asynchronous training with Controlled Inconsistency), a bubble-free asynchronous pipeline method that bounds forward/backward version drift without weight stashing, prediction, additional parameter copies, or global synchronization. The key idea is to use local gradient accumulation as a version-control mechanism: by slowing parameter-version evolution relative to pipeline delay, PACI limits the number of optimizer updates crossed by any micro-batch while preserving steady-state utilization. In GPT-style language-model pretraining, PACI matches the stability and final perplexity of synchronous 1F1B-flush, retains the same peak memory footprint, achieves fully utilized pipeline throughput, and improves training time-to-accuracy by up to $1.69\times$ over the fastest flush baseline. These results show that forward/backward inconsistency need not be eliminated: when explicitly bounded, it can be safely traded for substantial efficiency gains.

流水线并行训练效率大模型

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