Shuffling the Data, Extrapolating the Step: Sharper Bias In Constant Step-Size SGD
Konstantinos Emmanouilidis, Emmanouil-Vasileios Vlatakis-Gkaragkounis, Rene Vidal
摘要
From adversarial robustness to multi-agent learning, many machine learning tasks can be cast as finite-sum min–max optimization or, more generally, as variational inequality problems (VIPs). Owing to their simplicity and scalability, stochastic gradient methods with constant step size are widely used, despite the fact that they converge only up to a constant term. Among the many heuristics adopted in practice, two classical techniques have recently attracted attention to mitigate this issue: Random Reshuffling of data and Richardson–Romberg extrapolation across iterates. Random Reshuffling sharpens the mean-squared error (MSE) of the estimated solution, while Richardson-Romberg extrapolation acts orthogonally, providing a second order reduction in its bias. In this work, we show that their composition is strictly better than both, not only maintaining the enhanced MSE guarantees but also yielding an even greater cubic refinement in the bias. To the best of our knowledge, our work provides the first theoretical guarantees for such a synergy in structured non-monotone VIPs. Our analysis proceeds in two steps: (i) we smooth the discrete noise induced by reshuffling and leverage tools from continuous-state Markov chain theory to establish a novel law of large numbers and a central limit theorem for its iterates; and (ii) we employ spectral tensor techniques to prove that extrapolation debiases and sharpens the asymptotic behavior even under the biased gradient oracle induced by reshuffling. Finally, extensive experiments validate our theory, consistently demonstrating substantial speedups in practice.
问问这篇 Paper
智能体会读完全文。
Lune 把这篇 Paper 索引到了每一个公式,引用它的顶会 Paper 也一样。你提问,回答直接引用原文。
它引用的顶会 Paper27
- What is Local Optimality in Nonconvex-Nonconcave Minimax Optimization?Chi Jin, Praneeth Netrapalli, Michael I. JordanICML 2020 · 被引用 381 次
- Random Reshuffling: Simple Analysis with Vast ImprovementsKonstantin Mishchenko, Ahmed Khaled, Peter RichtárikNeurIPS 2020 · 被引用 172 次
- On the Almost Sure Convergence of Stochastic Gradient Descent in Non-Convex ProblemsPanayotis Mertikopoulos, Nadav Hallak, Ali Kavis, Volkan CevherNeurIPS 2020 · 被引用 120 次
- The Limits of Min-Max Optimization Algorithms: Convergence to Spurious Non-Critical SetsYa-Ping Hsieh, Panayotis Mertikopoulos, Volkan CevherICML 2021 · 被引用 96 次
- SGD with shuffling: optimal rates without component convexity and large epoch requirementsKwangjun Ahn, Chulhee Yun, Suvrit SraNeurIPS 2020 · 被引用 83 次
相关 Paper
- High-Order Error Bounds for Markovian LSA with Richardson-Romberg ExtrapolationIlya Levin, Alexey Naumov, Sergey SamsonovAAAI 2026
- Nonasymptotic Analysis of Stochastic Gradient Descent with the Richardson-Romberg ExtrapolationMarina Sheshukova, Denis Belomestny, Alain Oliviero Durmus, Eric Moulines 等ICLR 2025
- Effectiveness of Constant Stepsize in Markovian LSA and Statistical InferenceDongyan Lucy Huo, Yudong Chen, Qiaomin XieAAAI 2024 · 被引用 5 次
- Single-Call Stochastic Extragradient Methods for Structured Non-monotone Variational Inequalities: Improved Analysis under Weaker ConditionsSayantan Choudhury, Eduard Gorbunov, Nicolas LoizouNeurIPS 2023 · 被引用 24 次
- Proximal and Federated Random ReshufflingKonstantin Mishchenko, Ahmed Khaled, Peter RichtárikICML 2022 · 被引用 39 次
