cs.CLApr 23, 2026

Decoupled DiLoCo for Resilient Distributed Pre-training

Authors: Arthur DouillardKeith RushYani DonchevZachary CharlesNova FallenAyush DubeyIonel GogJosef Dean+9 more

Organizations: 1Google DeepMind · 2Google Research

Abstract

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.

Explore similar work

CardsList
  1. Factored Gossip DiLoCo: Reducing Blocking Communication in DiLoCo

    Jun 22, 2026Chamin Hewa Koneputugodage, Thalaiyasingam Ajanthan, Sameera Ramasinghe +7Distributed OptimizationDistributional Learning

  2. NoLoCo: No-all-reduce Low Communication Training Method for Large Models

    Jun 12, 2025Jari Kolehmainen, Nikolay Blagoev, Semih Kara +3Large ModelsParallel