arXiv:2410.05985cs.LGcs.AI2024-10被引 1

LayUp通过分层异步更新,显著提升分布式训练速度与鲁棒性。

LAYUP: Asynchronous decentralized gradient descent with LAYer-wise UPdates

  • 分层异步更新,边反向传播边通信,无需缓冲区
  • 比同步数据并行快32%,比同类高效算法快27%
  • 对慢节点不敏感,适合实际分布式训练场景

深度学习模型规模扩大使得跨多设备分布式训练成为必要。同步集中式方法存在较大通信与同步开销。通信高效算法虽可降低开销,但常需额外缓冲、易受慢节点或参数漂移影响。本文提出LayUp,一种分层异步的去中心化随机梯度下降方法。它在反向传播过程中异步交换分层增量更新,采用随机播送通信,使更新可立即应用而无需缓冲。该设计减少参数漂移,提升对慢节点的鲁棒性。我们建立了分层更新引入梯度偏差的理论上限,并证明了LayUp的收敛性。在视觉与语言建模任务上实验验证,其收敛速度比同步数据并行快约32%(墙钟时间),比同类通信高效算法快约27%,且任务性能更优。部分提速源于更高的模型浮点运算利用率。通过人为注入通信延迟,表明LayUp对慢节点保持鲁棒,而DDP等方法性能下降。总体而言,LayUp为分布式训练提供了一种新颖、实用、抗慢节点的替代方案,且不牺牲精度。

原文摘要 · Abstract (English)

The increasing size of deep learning models has made distributed training across multiple devices essential. Synchronous, centralized methods incur large communication and synchronization overheads. Communication efficient algorithms can reduce these overheads, but often require extra buffers, remain sensitive to stragglers or parameter drift. We present LayUp, an asynchronous decentralized SGD method with layer-wise updates. LayUp asynchronously exchanges incremental layer-wise updates during backpropagation. It uses randomized gossip communication, enabling updates to be applied as soon as they are available without buffering. These design choices reduce parameter drift and improve robustness to stragglers. We establish a theoretical upper bound for the gradient bias introduced by layer-wise updates and prove convergence of LayUp. We empirically validate LayUp on vision and language modeling tasks, showing convergence up to ~32% faster in terms of wall-clock time compared to synchronous data parallel training and up to ~27% faster than comparable communication efficient algorithms while maintaining better task performance. This speed-up is partly due to higher model FLOPs utilization, as we demonstrate. By injecting delays into the communication between workers, we show that LayUp remains robust to stragglers while DDP and other methods degrade in performance. Overall, LayUp provides a novel practical, straggler-robust alternative for distributed training without sacrificing accuracy.

分布式训练异步更新去中心化

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