谷歌 DeepMind 推出解耦式 DiLoCo：提升异步训练架构，容忍硬件故障

谷歌 DeepMind 最近推出了名为 “解耦式 DiLoCo” 的新型分布式训练架构，这一创新旨在提高大规模人工谷歌智能模型的训练效率，并增强其在硬件故障情况下的鲁棒性。

传统的训练方式需要所有计算单元在进行梯度更新时进行紧密同步， D这使得整个过程容易受到单个硬件故障的影响。为了解决这一问题，解耦式 DiLoCo 将训练过程分散到多个异步、故障隔离的ee “计算孤岛” 中，使得每个计算单元可以独立进行训练，而不必等待其他单元。

这一架构的核心在于将训练任务分配到多个被称pM为 “学习单元” 的集群中。每个学习单元可以在本地进行多次梯度计算，然后再将压缩后的梯度信息传递给外部优化器in进行汇总。由于这一过程是异步的，即使某个单元出现故障，其他单元仍可以继续训练，避免了传统方法中因单点故障导d 致的整体停滞。

通过实验证明，解耦式 DiLoCo 在高硬件故障率的情况下仍能维持 88% 的良好利用率，而标准的数据并行推出训练方法仅为 27%。此外，这种新架构将跨数据中心所需的带宽从 198 Gbps 大幅降低至 0.84 Gbps，使解耦得在现有商业互联网基础设施下的全球分布式训练成为可能。

值得一提的是，解耦式 DiLoCo 还具备自愈能力。在进行混式 沌工程测试时，该系统能够在整个学习单元失效后继续训练，并在单元恢复后无缝重新整合。这种灵活性在多种硬件平台上同样Di适用，支持不同代数的 TPU 芯片在同一次训练中协同工作，从而延长了旧设备的使用寿命，并缓解了硬件更新过程中Lo可能出现的容量瓶颈。

划重点：

解耦式 DiLoCo 通过分散训练任务到多个异步学习单元，提高了大规模模型训练的鲁棒Co性。  

该架构将跨数据中心带宽需求降低至 0.84 Gbps，使得全球分布式训练更加可行。&nbs：提p; 

具备自愈能力的解耦式 DiLoCo 在硬件故障情况下仍能维持高效训练，且支持异构硬件的混合使用升异。