arXiv:2510.07980cs.LGcs.AI2025-10NeurIPS被引 2

揭示多轮广播通信如何提升去中心化训练的泛化能力

Unveiling the Power of Multiple Gossip Steps: A Stability-Based Generalization Analysis in Decentralized Training

  • 用稳定性分析证明多轮广播可指数级降低优化误差
  • 即使无限轮次广播,去中心化仍存在不可消除的泛化差距
  • 首次在非凸条件下统一分析学习率、数据异构等影响因素

去中心化训练通过移除中心服务器实现高效通信,但性能常逊于集中式训练。多轮广播(MGS)作为连接两者的关键机制,显著缩小了性能差距,但其理论依据及是否能完全消除差距仍不明确。本文通过稳定性分析,推导出MGS的泛化误差与过拟合误差上界,系统回答这两个核心问题:1)优化误差以指数速率下降,从而指数级收紧泛化误差上界,使收敛至更优解;2)即便MGS趋于无穷,与集中式小批量SGD相比仍存在显著泛化误差差距(集中式为$/mathcal{O}(T^{ rac{cβ}{cβ+1}}/{n m})$,去中心化为$/mathcal{O}(T^{ rac{2cβ}{2cβ+2}}/{n m^{ rac{1}{2cβ+2}}})$)。此外,本文首次在非凸设定下,无需有界梯度假设,统一分析学习率、数据异构性、节点数、单节点样本量及通信拓扑对泛化的影响,填补了去中心化训练的理论空白。CIFAR数据集上的实验验证了理论结论。

原文摘要 · Abstract (English)

Decentralized training removes the centralized server, making it a communication-efficient approach that can significantly improve training efficiency, but it often suffers from degraded performance compared to centralized training. Multi-Gossip Steps (MGS) serve as a simple yet effective bridge between decentralized and centralized training, significantly reducing experiment performance gaps. However, the theoretical reasons for its effectiveness and whether this gap can be fully eliminated by MGS remain open questions. In this paper, we derive upper bounds on the generalization error and excess error of MGS using stability analysis, systematically answering these two key questions. 1). Optimization Error Reduction: MGS reduces the optimization error bound at an exponential rate, thereby exponentially tightening the generalization error bound and enabling convergence to better solutions. 2). Gap to Centralization: Even as MGS approaches infinity, a non-negligible gap in generalization error remains compared to centralized mini-batch SGD ($\mathcal{O}(T^{\frac{cβ}{cβ+1}}/{n m})$ in centralized and $\mathcal{O}(T^{\frac{2cβ}{2cβ+2}}/{n m^{\frac{1}{2cβ+2}}})$ in decentralized). Furthermore, we provide the first unified analysis of how factors like learning rate, data heterogeneity, node count, per-node sample size, and communication topology impact the generalization of MGS under non-convex settings without the bounded gradients assumption, filling a critical theoretical gap in decentralized training. Finally, promising experiments on CIFAR datasets support our theoretical findings.

去中心化训练泛化分析多轮广播稳定性理论

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