arXiv:2509.10439cs.LGmath.OC2025-09NeurIPS被引 6

调外学习率可降误差、补内学习率失误,还能加速收敛。

Understanding Outer Optimizers in Local SGD: Learning Rates, Momentum, and Acceleration

  • 用外优化器调节学习率,平衡误差与噪声
  • 外学习率可超1,动量调整后仍有效
  • 适合分布式训练场景,尤其边缘设备

现代机器学习常需大批次、分布式数据和大规模并行计算(如移动设备或分布式数据中心)。通信成为主要瓶颈,而局部随机梯度下降(Local SGD)能有效降低通信开销。该方法包含三部分:本地优化、聚合机制及使用节点聚合更新的外优化器。尽管对本地优化超参数已有大量研究,外优化器及其超参数选择仍不明确。本文研究外优化器的作用,证明了新的收敛性保证。我们发现,调节外学习率可(a)在优化误差与随机梯度噪声方差间权衡,(b)弥补内学习率不当带来的影响。理论表明,外学习率有时应设为大于1。我们还将结果扩展至外优化器使用动量的情形,并揭示动量调整后的外学习率有类似作用。此外,研究了外优化器中的加速机制,其显著提升通信轮次相关的收敛速度,优于此前仅在本地应用加速的算法。最后,提出一种新颖的数据依赖分析,进一步揭示外学习率调优规律。通过标准语言模型和多种外优化器的全面实验验证了理论结论。

原文摘要 · Abstract (English)

Modern machine learning often requires training with large batch size, distributed data, and massively parallel compute hardware (like mobile and other edge devices or distributed data centers). Communication becomes a major bottleneck in such settings but methods like Local Stochastic Gradient Descent (Local SGD) show great promise in reducing this additional communication overhead. Local SGD consists of three parts: a local optimization process, an aggregation mechanism, and an outer optimizer that uses the aggregated updates from the nodes to produce a new model. While there exists an extensive literature on understanding the impact of hyperparameters in the local optimization process, the choice of outer optimizer and its hyperparameters is less clear. We study the role of the outer optimizer in Local SGD, and prove new convergence guarantees for the algorithm. In particular, we show that tuning the outer learning rate allows us to (a) trade off between optimization error and stochastic gradient noise variance, and (b) make up for ill-tuning of the inner learning rate. Our theory suggests that the outer learning rate should sometimes be set to values greater than $1$. We extend our results to settings where we use momentum in the outer optimizer, and we show a similar role for the momentum-adjusted outer learning rate. We also study acceleration in the outer optimizer and show that it improves the convergence rate as a function of the number of communication rounds, improving upon the convergence rate of prior algorithms that apply acceleration locally. Finally, we also introduce a novel data-dependent analysis of Local SGD that yields further insights on outer learning rate tuning. We conduct comprehensive experiments with standard language models and various outer optimizers to validate our theory.

分布式训练优化器通信效率

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