arXiv:2604.21428cs.CL2026-04被引 5

提出解耦版DiLoCo,让大模型训练更抗故障、不卡死。

Decoupled DiLoCo for Resilient Distributed Pre-training

论文配图:Decoupled DiLoCo for Resilient Distributed Pre-training
图 1 · 摘自论文原文
  • 将训练拆成多个独立学习器,异步更新参数
  • 通过最小多数派聚合,实现零全局宕机
  • 适合大规模分布式训练,尤其硬件不稳定的场景

现代大规模语言模型预训练依赖于单程序多数据(SPMD)范式,需加速器间紧密耦合。由于这种耦合,瞬时延迟、硬件故障和同步开销会导致整个计算停滞,造成大规模算力浪费。尽管近期分布式方法如DiLoCo减少了通信带宽,但仍为同步机制,易受系统停顿影响。为此,我们提出解耦版DiLoCo,对DiLoCo框架的演进,旨在打破同步锁步障碍,超越SPMD以最大化训练吞吐量。该方法将计算划分为多个独立的“学习器”,各自执行本地内优化步骤,异步向中心同步器发送参数片段。同步器通过最小多数派、自适应宽限期及动态标记加权合并机制,绕过故障或慢速学习器。受“混沌工程”启发,在数百万模拟芯片环境中实现显著提升的训练效率,且严格保持零全局宕机,同时在文本与视觉任务上维持具有竞争力的模型性能,适用于密集与专家混合架构。

原文摘要 · Abstract (English)

Modern large-scale language model pre-training relies heavily on the single program multiple data (SPMD) paradigm, which requires tight coupling across accelerators. Due to this coupling, transient slowdowns, hardware failures, and synchronization overhead stall the entire computation, wasting significant compute time at scale. While recent distributed methods like DiLoCo reduced communication bandwidth, they remained fundamentally synchronous and vulnerable to these system stalls. To address this, we introduce Decoupled DiLoCo, an evolution of the DiLoCo framework designed to break the lock-step synchronization barrier and go beyond SPMD to maximize training goodput. Decoupled DiLoCo partitions compute across multiple independent ``learners'' that execute local inner optimization steps. These learners asynchronously communicate parameter fragments to a central synchronizer, which circumvents failed or straggling learners by aggregating updates using a minimum quorum, an adaptive grace window, and dynamic token-weighted merging. Inspired by ``chaos engineering'', we achieve significantly improved training efficiency in failure-prone environments with millions of simulated chips with strictly zero global downtime, while maintaining competitive model performance across text and vision tasks, for both dense and mixture-of-expert architectures.

分布式训练抗故障大模型预训练

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