arXiv:2605.08871math.OCcs.DC2026-05

提出改进版优化算法,显著降低异构系统中训练时间。

Rennala MVR: Improved Time Complexity for Parallel Stochastic Optimization via Momentum-Based Variance Reduction

论文配图:Rennala MVR: Improved Time Complexity for Parallel Stochastic Optimization via Momentum-Based Variance Reduction
图 1 · 摘自论文原文
  • 基于动量的方差缩减技术提升并行优化效率
  • 理论证明在特定条件下时间复杂度可进一步降低
  • 适合大规模分布式训练场景,尤其硬件异构环境

大规模机器学习模型常在存在硬件差异、网络延迟和系统不稳定的集群上训练,此时时间复杂度比迭代复杂度更关键。近期工作(Tyurin & Richtárik, 2023)首次建立并行一阶随机优化的时间复杂度分析,提出Rennala SGD作为平滑非凸优化中的时间最优方法。但该方法本质上是SGD的变体,而方差缩减技术已被证明能改善SGD的迭代复杂度。本文研究方差缩减是否也能提升异构系统下的时间复杂度。在均方光滑性假设下,我们证明方差缩减可在相关参数区间内改善时间复杂度。为此,提出Rennala MVR——基于动量的方差缩减对Rennala SGD的扩展,并分析其查询与时间复杂度。建立该假设下的时间复杂度下界。在随机二次基准上,精确方法的实验结果支持理论;神经网络实验中使用近似实用版本也显示出相对于Rennala SGD的相似性能提升。

原文摘要 · Abstract (English)

Large-scale machine learning models are trained on clusters of machines that exhibit heterogeneous performance due to hardware variability, network delays, and system-level instabilities. In such environments, time complexity rather than iteration complexity becomes the relevant performance metric for optimization algorithms. Recent work by Tyurin and Richtárik (2023) established the first time complexity analysis for parallel first-order stochastic optimization, proposing Rennala SGD as a time-optimal method for smooth nonconvex optimization. However, Rennala SGD is fundamentally a modification of SGD, and variance reduction techniques are known to improve the iteration complexity of SGD. In this work, we investigate whether variance reduction can also improve time complexity in heterogeneous systems. We show that, under a mean-squared smoothness assumption, variance reduction can improve time complexity in relevant parameter regimes. To this end, we propose Rennala MVR, a variance-reduced extension of Rennala SGD based on momentum-based variance reduction, and analyze its oracle and time complexity. We establish lower bounds for time complexity under these assumptions. On a stochastic quadratic benchmark, experiments with the exact method support the theory, while neural-network experiments with a practical inexact variant show similar empirical gains over Rennala SGD.

优化算法异构训练方差缩减时间复杂度

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