arXiv:2509.12213cs.LGcs.AI2025-09

提出自适应通信图方法,让去中心化训练在超大规模下稳定高效。

Scaling Up Data Parallelism in Decentralized Deep Learning

  • 设计动态调整通信图的自适应算法,应对参数方差波动。
  • 在1008张GPU上训练ResNet50,精度与集中式相当。
  • 发现通信图连接数和参数方差直接影响模型精度。

尽管去中心化学习在理论上被广泛研究,但尚未进入生产环境,主要因其在大规模深度神经网络训练中缺乏稳定性、可扩展性和通用性。本文通过引入基准框架DBench,对比分析集中式与去中心式DNN训练,揭示了模型精度与参数张量方差之间的关联。实验表明:(1) 去中心化训练在大规模下同样存在可扩展性与通用性问题;(2) 模型精度与通信图的连接数正相关;(3) 精度对模型副本间参数张量方差极为敏感。基于此,提出Ada算法,一种基于去中心化SGD并动态调整通信图的自适应方法。在大规模训练中,Ada始终实现最优收敛速度,在所有样本任务中达到与集中式相当的精度,包括在1008张GPU上训练ImageNet-1K的ResNet50。

原文摘要 · Abstract (English)

Although it has been extensively explored in theory, decentralized learning is not yet green-lighted for production use, largely due to a lack of stability, scalability, and generality in large scale DNN training. To shed light on the production use of decentralized learning, this work studies decentralized data parallel training at scale. To this end, we introduce a benchmarking framework, namely DBench, to host both centralized and decentralized DNN training. Building upon DBench, we introduce a benchmarking methodology to uncover the correlations between model accuracy and the variances of parameter tensors by varying communication graphs and training scales. Based on the benchmarking results, we observe that, (1) Similar to centralized learning, decentralized data parallel training also presents the issues of scalability and generality when the training scales up; (2) The model accuracy of decentralized learning is correlated to the number of connections in a communication graph; (3) The model accuracy of decentralized learning is surprisingly sensitive to the variance of parameter tensors across model replicas. Built upon the observations, we propose Ada, a decentralized adaptive approach that performs large scale DNN training following a decentralized SGD method and adapting the communication graph in use dynamically throughout training iterations. We apply Ada on large scale training and observe that Ada can obtain the best convergence rates consistently in decentralized DNN training, and delivers equally or comparably good model accuracy for all sample applications as centralized learning does, even when training ResNet50 for ImageNet-1K on the scale of 1008 GPUs.

去中心化训练分布式优化自适应通信大规模训练

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