arXiv:2501.15259cs.LGmath.OC2025-01ICLR被引 3

通过选择性激活节点,提升大规模去中心化学习的收敛速度与稳定性。

Scalable Decentralized Learning with Teleportation

  • 仅激活部分节点,由其从历史节点获取参数并进行局部通信。
  • 在小规模活跃节点拓扑上执行SGD更新与平均,显著改善收敛率。
  • 适合大规模分布式训练场景,尤其适用于通信受限的系统。

去中心化随机梯度下降(Decentralized SGD)虽通信成本低,但稀疏通信导致收敛速度下降,尤其在节点数量庞大时更为明显。尽管已有研究通过设计谱间隙大的拓扑缓解该问题,但在大规模场景下仍存在显著性能损失。本文提出TELEPORTATION:仅激活部分节点,这些节点从先前活跃节点获取参数,随后在仅包含活跃节点的小型拓扑上执行SGD更新与对等平均。我们证明,合理激活节点数量可完全消除收敛速度退化。此外,提出一种高效的超参数调优方法,自动搜索最优激活节点数。实验表明,TELEPORTATION能更稳定地训练神经网络,并获得比Decentralized SGD更高的准确率。

原文摘要 · Abstract (English)

Decentralized SGD can run with low communication costs, but its sparse communication characteristics deteriorate the convergence rate, especially when the number of nodes is large. In decentralized learning settings, communication is assumed to occur on only a given topology, while in many practical cases, the topology merely represents a preferred communication pattern, and connecting to arbitrary nodes is still possible. Previous studies have tried to alleviate the convergence rate degradation in these cases by designing topologies with large spectral gaps. However, the degradation is still significant when the number of nodes is substantial. In this work, we propose TELEPORTATION. TELEPORTATION activates only a subset of nodes, and the active nodes fetch the parameters from previous active nodes. Then, the active nodes update their parameters by SGD and perform gossip averaging on a relatively small topology comprising only the active nodes. We show that by activating only a proper number of nodes, TELEPORTATION can completely alleviate the convergence rate degradation. Furthermore, we propose an efficient hyperparameter-tuning method to search for the appropriate number of nodes to be activated. Experimentally, we showed that TELEPORTATION can train neural networks more stably and achieve higher accuracy than Decentralized SGD.

去中心化学习通信优化分布式训练

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