Random Shuffling Beats SGD Only After Many Epochs on Ill-Conditioned Problems
Itay Safran, Ohad Shamir
Abstract
Recently, there has been much interest in studying the convergence rates of without-replacement SGD, and proving that it is faster than with-replacement SGD in the worst case. However, known lower bounds ignore the problem's geometry, including its condition number, whereas the upper bounds explicitly depend on it. Perhaps surprisingly, we prove that when the condition number is taken into account, without-replacement SGD does not significantly improve on with-replacement SGD in terms of worst-case bounds, unless the number of epochs (passes over the data) is larger than the condition number. Since many problems in machine learning and other areas are both ill-conditioned and involve large datasets, this indicates that without-replacement does not necessarily improve over with-replacement sampling for realistic iteration budgets. We show this by providing new lower and upper bounds which are tight (up to log factors), for quadratic problems with commuting quadratic terms, precisely quantifying the dependence on the problem parameters.
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 600533a8-c672-4849-892e-dea1ff05706bCited by top-tier papers14
- Convergence Analysis of Sequential Federated Learning on Heterogeneous DataYipeng Li, Xinchen LyuNeurIPS 2023 · 53 citations
- On the Convergence of Federated Averaging with Cyclic Client ParticipationYae Jee Cho, Pranay Sharma, Gauri Joshi, Zheng Xu et al.ICML 2023 · 47 citations
- Minibatch vs Local SGD with Shuffling: Tight Convergence Bounds and BeyondChulhee Yun, Shashank Rajput, Suvrit SraICLR 2022 · 47 citations
- Tighter Lower Bounds for Shuffling SGD: Random Permutations and BeyondJaeyoung Cha, Jaewook Lee, Chulhee YunICML 2023 · 26 citations
- Benign Underfitting of Stochastic Gradient DescentTomer Koren, Roi Livni, Yishay Mansour, Uri ShermanNeurIPS 2022 · 26 citations
Builds on4
- Random Reshuffling: Simple Analysis with Vast ImprovementsKonstantin Mishchenko, Ahmed Khaled, Peter RichtárikNeurIPS 2020 · 172 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
- Random Reshuffling is Not Always BetterChristopher De SaNeurIPS 2020 · 27 citations
Related papers
- Tighter Convergence Bounds for Shuffled SGD via Primal-Dual PerspectiveXufeng Cai, Cheuk Yin Lin, Jelena DiakonikolasNeurIPS 2024 · 9 citations
- Recht-Re Noncommutative Arithmetic-Geometric Mean Conjecture is FalseZehua Lai, Lek-Heng LimICML 2020 · 22 citations
- Permutation-Based SGD: Is Random Optimal?Shashank Rajput, Kangwook Lee, Dimitris S. PapailiopoulosICLR 2022 · 15 citations
- Provably Faster Algorithms for Bilevel Optimization via Without-Replacement SamplingJunyi Li, Heng HuangNeurIPS 2024 · 1 citation
- Incremental Gradient Descent with Small Epoch Counts is Surprisingly Slow on Ill-Conditioned ProblemsYujun Kim, Jaeyoung Cha, Chulhee YunICML 2025
