arXiv:2410.12918cs.DCcs.AI2024-10中稿 · appear in the Proc…被引 9

通过模型分片加速去中心化学习,有效缓解通信延迟问题。

Boosting Asynchronous Decentralized Learning with Model Fragmentation

  • 将模型拆分为小片段并行发送,提升带宽利用效率
  • 在CIFAR-10上比AD-PSGD快3.9倍,准确率提升19.4%
  • 首次证明异步通信延迟下的收敛性,适合高延迟场景

去中心化学习(DL)允许网络节点在不共享原始数据的情况下协同训练模型。处理计算或通信较慢的节点(即拖延者)是其关键挑战。本文提出DivShare,一种新型异步DL算法,在存在通信拖延者时实现快速模型收敛。该方法将模型参数分块,节点并行向随机选择的其他节点发送各子集,而非顺序传输完整模型。小片段传输更高效利用集体带宽,使网络慢的节点也能快速贡献部分参数。我们首次理论上证明了考虑异步通信延迟的DL算法收敛性。实验对比AD-PSGD和Swift两个先进基线,在CIFAR-10和MovieLens数据集上,DivShare在存在通信拖延者时,时间到精度降低最多达3.9倍;在CIFAR-10和MovieLens上分别实现最高19.4%的准确率提升和9.5%的测试损失降低。

原文摘要 · Abstract (English)

Decentralized learning (DL) is an emerging technique that allows nodes on the web to collaboratively train machine learning models without sharing raw data. Dealing with stragglers, i.e., nodes with slower compute or communication than others, is a key challenge in DL. We present DivShare, a novel asynchronous DL algorithm that achieves fast model convergence in the presence of communication stragglers. DivShare achieves this by having nodes fragment their models into parameter subsets and send, in parallel to computation, each subset to a random sample of other nodes instead of sequentially exchanging full models. The transfer of smaller fragments allows more efficient usage of the collective bandwidth and enables nodes with slow network links to quickly contribute with at least some of their model parameters. By theoretically proving the convergence of DivShare, we provide, to the best of our knowledge, the first formal proof of convergence for a DL algorithm that accounts for the effects of asynchronous communication with delays. We experimentally evaluate DivShare against two state-of-the-art DL baselines, AD-PSGD and Swift, and with two standard datasets, CIFAR-10 and MovieLens. We find that DivShare with communication stragglers lowers time-to-accuracy by up to 3.9x compared to AD-PSGD on the CIFAR-10 dataset. Compared to baselines, DivShare also achieves up to 19.4% better accuracy and 9.5% lower test loss on the CIFAR-10 and MovieLens datasets, respectively.

去中心化学习异步通信模型分片分布式优化

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