arXiv:2504.18454cs.LG2025-04中稿 · TMLR被引 2

通过伪异步机制减少通信频率,提升大规模分布式训练效率。

Pseudo-Asynchronous Local SGD: Robust and Efficient Data-Parallel Training

  • 引入伪同步机制,延长同步间隔以降低通信开销。
  • 在ImageNet-1K和TinyStories上比DDP快18.4%~24.4%。
  • 理论证明收敛性,适合超大规模模型训练场景。

随着人工智能模型规模持续扩大,训练前沿模型需依赖海量计算资源,推动分布式深度学习方法发展。数据并行是加速训练的关键手段,但频繁的全局通信成为大规模训练瓶颈。本文提出伪异步局部SGD(PALSGD),基于局部SGD与DiLoCo思想,通过伪同步机制进一步降低通信频率。该方法允许比标准局部SGD更长的同步间隔,同时保持模型一致性,性能与高频同步相当。我们提供了理论分析,证明其收敛性并推导收敛速率。在图像分类与语言建模任务上的实验表明,PALSGD在相同时间内表现优于DDP与DiLoCo:在ImageNet-1K上使用ResNet-50时,训练速度比DDP快18.4%;在TinyStories上使用GPT-Neo-125M时快24.4%,使用GPT-Neo-8M时快21.1%。

原文摘要 · Abstract (English)

Following AI scaling trends, frontier models continue to grow in size and continue to be trained on larger datasets. Training these models requires huge investments in exascale computational resources, which has in turn driven developtment of distributed deep learning methods. Data parallelism is an essential approach to speed up training, but it requires frequent global communication between workers, which can bottleneck training at the largest scales. In this work, we propose a method called Pseudo-Asynchronous Local SGD (PALSGD) to improve the efficiency of data-parallel training. PALSGD is an extension of Local SGD (Stich, 2018) and DiLoCo (Douillard et al., 2023), designed to further reduce communication frequency by introducing a pseudo-synchronization mechanism. PALSGD allows the use of longer synchronization intervals compared to standard Local SGD. Despite the reduced communication frequency, the pseudo-synchronization approach ensures that model consistency is maintained, leading to performance results comparable to those achieved with more frequent synchronization. Furthermore, we provide a theoretical analysis of PALSGD, establishing its convergence and deriving its convergence rate. This analysis offers insights into the algorithm's behavior and performance guarantees. We evaluated PALSGD on image classification and language modeling tasks. Our results show that PALSGD achieves better performance in less time compared to existing methods like Distributed Data Parallel (DDP), and DiLoCo. Notably, PALSGD trains 18.4% faster than DDP on ImageNet-1K with ResNet-50, 24.4% faster than DDP on TinyStories with GPT-Neo-125M, and 21.1% faster than DDP on TinyStories with GPT-Neo-8M.

分布式训练通信优化模型加速

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