Shuffling the Data, Stretching the Step-size: Sharper Bias in constant step-size SGD
Not on the map. Why?
Abstract (arXiv)
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.
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 77 of its 138 references that are in the map. First posted on arXiv 11 Apr 2026.
Builds on
77 of 138 references.
References · 77
- 2016Optimization Methods for Large-Scale Machine Learningcited by 1,150
- 2018Gradient Descent Finds Global Minima of Deep Neural Networkscited by 574
- 2019Multi-Agent Reinforcement Learning: A Selective Overview of Theories and Algorithmscited by 567
- 2017Non-convex learning via Stochastic Gradient Langevin Dynamics: a nonasymptotic analysiscited by 243
- 2017Training GANs with Optimismcited by 217
- 2014Theoretical guarantees for approximate sampling from smooth and log-concave densitiescited by 208
- 2019SGD: General Analysis and Improved Ratescited by 207
- 2015Non-asymptotic convergence analysis for the Unadjusted Langevin Algorithmcited by 197
- 2015Fast Convergence of Regularized Learning in Gamescited by 150
- 2017The Power of Interpolation: Understanding the Effectiveness of SGD in Modern Over-parametrized Learningcited by 150
- 2018Fast and Faster Convergence of SGD for Over-Parameterized Models and an Accelerated Perceptroncited by 143
- 2013Stochastic Gradient Descent, Weighted Sampling, and the Randomized Kaczmarz algorithmcited by 133
38 more
- 2017Underdamped Langevin MCMC: A non-asymptotic analysiscited by 133
- 2018Log-concave sampling: Metropolis-Hastings algorithms are fastcited by 118
- 2017User-friendly guarantees for the Langevin Monte Carlo with inaccurate gradientcited by 116
- 2020Random Reshuffling: Simple Analysis with Vast Improvementscited by 115
- 2017The Numerics of GANscited by 105
- 2019On the convergence of single-call stochastic extra-gradient methodscited by 89
- 2017Bridging the Gap between Constant Step Size Stochastic Gradient Descent and Markov Chainscited by 88
- 2020The Complexity of Constrained Min-Max Optimizationcited by 82
- 2016Learning in games with continuous action sets and unknown payoff functionscited by 81
- 2020On the Almost Sure Convergence of Stochastic Gradient Descent in Non-Convex Problemscited by 80
- 2015Why Random Reshuffling Beats Stochastic Gradient Descentcited by 79
- 2019The Randomized Midpoint Method for Log-Concave Samplingcited by 78
- 2018Sharp convergence rates for Langevin dynamics in the nonconvex settingcited by 68
- 2018Random Shuffling Beats SGD after Finite Epochscited by 67
- 2020Global Convergence of Deep Networks with One Wide Layer Followed by Pyramidal Topologycited by 66
- 2019How Good is SGD with Random Shuffling?cited by 64
- 2018On sampling from a log-concave density using kinetic Langevin diffusionscited by 62
- 2019SGD without Replacement: Sharper Rates for General Smooth Convex Functionscited by 61
- 2020SGD with shuffling: optimal rates without component convexity and large epoch requirementscited by 58
- 2020Closing the convergence gap of SGD without replacementcited by 57
- 2020On the Convergence of Langevin Monte Carlo: The Interplay between Tail Growth and Smoothnesscited by 55
- 2013Fast Convergence of Stochastic Gradient Descent under a Strong Growth Conditioncited by 54
- 2015Sampling from a log-concave distribution with Projected Langevin Monte Carlocited by 52
- 2016Connecting Generative Adversarial Networks and Actor-Critic Methodscited by 50
- 2020The limits of min-max optimization algorithms: convergence to spurious non-critical setscited by 50
- 2020An Analysis of Constant Step Size SGD in the Non-convex Regime: Asymptotic Normality and Biascited by 48
- 2015Incremental Gradient, Subgradient, and Proximal Methods for Convex Optimization: A Surveycited by 45
- 2018Global Non-convex Optimization with Discretized Diffusionscited by 44
- 2017Sampling from a log-concave distribution with compact support with proximal Langevin Monte Carlocited by 43
- 2021Stochastic Gradient Descent-Ascent and Consensus Optimization for Smooth Games: Convergence Analysis under Expected Co-coercivitycited by 43
- 2018Averaging Stochastic Gradient Descent on Riemannian Manifoldscited by 39
- 2021Extragradient Method: O(1/K) Last-Iterate Convergence for Monotone Variational Inequalities and Connections With Cocoercivitycited by 39
- 2019Revisiting Stochastic Extragradientcited by 35
- 2019The Complexity of Finding Stationary Points with Stochastic Gradient Descentcited by 35
- 2019Escaping Saddle Points with Adaptive Gradient Methodscited by 34
- 2019Online Stochastic Gradient Descent with Arbitrary Initialization Solves Non-smooth, Non-convex Phase Retrievalcited by 34
- 2019Stochastic Runge-Kutta Accelerates Langevin Monte Carlo and Beyondcited by 33
- 2022Last-Iterate Convergence of Optimistic Gradient Method for Monotone Variational Inequalitiescited by 33
50 most cited of 77 All 77 (JSON)
Built on by
Citing papers per year: 2026: 1.
1 paper.