arXiv:2508.04950cs.LGmath.OC2025-08中稿 · TMLR被引 3

提出两种压缩的分布式优化算法,加速训练且降低通信开销。

Compressed Decentralized Momentum Stochastic Gradient Methods for Nonconvex Optimization

  • 结合动量与压缩通信,解决非凸优化中的收敛难题
  • 在梯度有界和数据异构场景下均达到最优收敛速度
  • 适合大规模分布式深度学习,尤其适合异构数据训练

本文针对非凸随机优化问题,设计了两种压缩的去中心化算法。第一种在梯度有界场景下采用自适应机制,是首个具备压缩通信的去中心化自适应方法;第二种在数据异构且梯度无界场景下,引入梯度追踪技术,结合重球动量,有效应对异构性挑战。两种方法均实现最优收敛速率,支持线性加速,且在特定误差容忍范围内参数不依赖网络拓扑。实验表明,在训练深度神经网络和Transformer模型时,性能优于现有先进方法。

原文摘要 · Abstract (English)

In this paper, we design two compressed decentralized algorithms for solving nonconvex stochastic optimization under two different scenarios. Both algorithms adopt a momentum technique to achieve fast convergence and a message-compression technique to save communication costs. Though momentum acceleration and compressed communication have been used in literature, it is highly nontrivial to theoretically prove the effectiveness of their composition in a decentralized algorithm that can maintain the benefits of both sides, because of the need to simultaneously control the consensus error, the compression error, and the bias from the momentum gradient. For the scenario where gradients are bounded, our proposal is a compressed decentralized adaptive method. To the best of our knowledge, this is the first decentralized adaptive stochastic gradient method with compressed communication. For the scenario of data heterogeneity without bounded gradients, our proposal is a compressed decentralized heavy-ball method, which applies a gradient tracking technique to address the challenge of data heterogeneity. Notably, both methods achieve an optimal convergence rate, and they can achieve linear speed up and adopt topology-independent algorithmic parameters within a certain regime of the user-specified error tolerance. Superior empirical performance is observed over state-of-the-art methods on training deep neural networks (DNNs) and Transformers.

分布式优化非凸优化压缩通信深度学习

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