arXiv:2606.30634cs.LG2026-06

一阶梯度延迟不致命,选对优化器可实现高效大规模异步训练

One-Step Gradient Delay is Not a Barrier for Large-Scale Asynchronous Pipeline Parallel LLM Pretraining

论文配图:One-Step Gradient Delay is Not a Barrier for Large-Scale Asynchronous Pipeline Parallel LLM Pretraining
图 1 · 摘自论文原文
  • 提出抗延迟的优化器改进方法,突破异步训练稳定性瓶颈
  • 100亿参数模型实验显示性能接近同步训练,吞吐量显著提升
  • 适合追求训练效率的工业级大模型研发团队参考

现代大规模语言模型预训练依赖流水线并行技术;然而同步实现会在流水线空洞期导致GPU闲置,浪费计算资源。异步流水线并行消除了这些空洞,虽以梯度延迟为代价提升了吞吐量。其中,PipeDream-2BW方案在任意流水线深度下仅引入一阶梯度延迟,极具吸引力。但其应用受限于普遍认为优化器在延迟下无法稳定训练的假设。本文挑战该观点,首次系统实证表明:当使用AdamW时性能严重退化,但近期方法如Muon在相同延迟下表现出强鲁棒性。我们提出一种无需依赖优化器的误差反馈修正机制,辅以理论分析证明了Muon在有无修正下的收敛性。在高达100亿参数的模型上进行充分评估,验证了所提策略能有效弥合与同步训练的性能差距,凸显异步流水线并行在大规模场景中的实用潜力。

原文摘要 · Abstract (English)

Modern large-scale LLM pretraining benefits from utilizing Pipeline Parallelism; however, synchronous implementations leave GPUs idle during pipeline bubbles, wasting computational resources. Asynchronous Pipeline Parallelism eliminates these bubbles, maximizing throughput at the cost of gradient staleness. Among asynchronous schedules, PipeDream-2BW is particularly appealing: unlike the original PipeDream schedule, it ensures a constant one-step gradient delay regardless of pipeline depth. However, its adoption remains limited due to the common belief that optimizing under staleness is fundamentally unstable. In this work, we challenge this assumption, demonstrating that degradation under one-step delay depends strongly on optimizer choice rather than being an intrinsic limitation. We provide the first comprehensive empirical analysis showing that while AdamW, the predominant optimizer at the time when PipeDream-2BW was introduced, indeed suffers from severe degradation, recent methods like Muon exhibit strong robustness under a one-step delay. We introduce an optimizer-agnostic Error Feedback-inspired correction to further mitigate delay effects. We provide supporting theoretical analysis demonstrating convergence for Muon with and without this correction. Extensive evaluation on models up to 10B parameters confirms that our strategies bridge the performance gap with synchronous training, highlighting the practical potential of asynchronous pipeline parallelism at scale.

大模型训练异步并行优化器流水线

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