On the Last-Iterate Convergence of Shuffling Gradient Methods
Zijian Liu, Zhengyuan Zhou
Abstract
Shuffling gradient methods are widely used in modern machine learning tasks and include three popular implementations: Random Reshuffle (RR), Shuffle Once (SO), and Incremental Gradient (IG). Compared to the empirical success, the theoretical guarantee of shuffling gradient methods was not well-understood for a long time. Until recently, the convergence rates had just been established for the average iterate for convex functions and the last iterate for strongly convex problems (using squared distance as the metric). However, when using the function value gap as the convergence criterion, existing theories cannot interpret the good performance of the last iterate in different settings (e.g., constrained optimization). To bridge this gap between practice and theory, we prove the first last-iterate convergence rates for shuffling gradient methods with respect to the objective value even without strong convexity. Our new results either (nearly) match the existing last-iterate lower bounds or are as fast as the previous best upper bounds for the average iterate.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 643f8df0-4cf7-48f9-b347-7cf29fd374eeCited by top-tier papers7
- Fast Last-Iterate Convergence of SGD in the Smooth Interpolation RegimeAmit Attia, Matan Schliserman, Uri Sherman, Tomer KorenNeurIPS 2025 · 18 citations
- New Hybrid Fine-Tuning Paradigm for LLMs: Algorithm Design and Convergence Analysis FrameworkShaocong Ma, Peiran Yu, Heng HuangICLR 2026 · 1 citation
- Provable Benefit of Random Permutations over Uniform Sampling in Stochastic Coordinate DescentDonghwa Kim, Jaewook Lee, Chulhee YunICML 2025
- Improved Last-Iterate Convergence of Shuffling Gradient Methods for Nonsmooth Convex OptimizationZijian Liu, Zhengyuan ZhouICML 2025
- A Unified Analysis of Stochastic Gradient Descent with Arbitrary Data Permutations and BeyondYipeng Li, Xinchen Lyu, Zhenyu LiuNeurIPS 2025
Builds on15
- Random Reshuffling: Simple Analysis with Vast ImprovementsKonstantin Mishchenko, Ahmed Khaled, Peter RichtárikNeurIPS 2020 · 172 citations
- DoG is SGD's Best Friend: A Parameter-Free Dynamic Step Size ScheduleMaor Ivgi, Oliver Hinder, Yair CarmonICML 2023 · 98 citations
- SGD with shuffling: optimal rates without component convexity and large epoch requirementsKwangjun Ahn, Chulhee Yun, Suvrit SraNeurIPS 2020 · 83 citations
- Closing the convergence gap of SGD without replacementShashank Rajput, Anant Gupta, Dimitris S. PapailiopoulosICML 2020 · 73 citations
- Minibatch vs Local SGD with Shuffling: Tight Convergence Bounds and BeyondChulhee Yun, Shashank Rajput, Suvrit SraICLR 2022 · 47 citations
Related papers
- Revisiting Convergence: Shuffling Complexity Beyond Lipschitz SmoothnessQi He, Peiran Yu, Ziyi Chen, Heng HuangICML 2025
- SMG: A Shuffling Gradient-Based Method with MomentumTrang H. Tran, Lam M. Nguyen, Quoc Tran-DinhICML 2021 · 25 citations
- Tighter Lower Bounds for Shuffling SGD: Random Permutations and BeyondJaeyoung Cha, Jaewook Lee, Chulhee YunICML 2023 · 26 citations
- Incremental Gradient Descent with Small Epoch Counts is Surprisingly Slow on Ill-Conditioned ProblemsYujun Kim, Jaeyoung Cha, Chulhee YunICML 2025
- Sampling without Replacement Leads to Faster Rates in Finite-Sum Minimax OptimizationAniket Das, Bernhard Schölkopf, Michael MuehlebachNeurIPS 2022 · 11 citations
