Permutation-Based SGD: Is Random Optimal?
Shashank Rajput, Kangwook Lee, Dimitris S. Papailiopoulos
Abstract
A recent line of ground-breaking results for permutation-based SGD has corroborated a widely observed phenomenon: random permutations offer faster convergence than with-replacement sampling. However, is random optimal? We show that this depends heavily on what functions we are optimizing, and the convergence gap between optimal and random permutations can vary from exponential to nonexistent. We first show that for 1-dimensional strongly convex functions, with smooth second derivatives, there exist optimal permutations that offer exponentially faster convergence compared to random. However, for general strongly convex functions, random permutations are optimal. Finally, we show that for quadratic, strongly-convex functions, there are easy-to-construct permutations that lead to accelerated convergence compared to random. Our results suggest that a general convergence characterization of optimal permutations cannot capture the nuances of individual function classes, and can mistakenly indicate that one cannot do much better than random.
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.
Cited by top-tier papers9
- Tighter Lower Bounds for Shuffling SGD: Random Permutations and BeyondJaeyoung Cha, Jaewook Lee, Chulhee YunICML 2023 · 26 citations
- On the Last-Iterate Convergence of Shuffling Gradient MethodsZijian Liu, Zhengyuan ZhouICML 2024 · 11 citations
- CD-GraB: Coordinating Distributed Example Orders for Provably Accelerated TrainingA. Feder Cooper, Wentao Guo, Khiem Pham, Tiancheng Yuan et al.NeurIPS 2023 · 9 citations
- Are Greedy Task Orderings Better Than Random in Continual Linear Regression?Matan Tsipory, Ran Levinstein, Itay Evron, Mark Kong et al.NeurIPS 2025 · 5 citations
- Stochastic Extragradient with Flip-Flop Shuffling & Anchoring: Provable ImprovementsJiseok Chae, Chulhee Yun, Donghwan KimNeurIPS 2024 · 2 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
- Incremental Gradient Descent with Small Epoch Counts is Surprisingly Slow on Ill-Conditioned ProblemsYujun Kim, Jaeyoung Cha, Chulhee YunICML 2025
- Provable Benefit of Random Permutations over Uniform Sampling in Stochastic Coordinate DescentDonghwa Kim, Jaewook Lee, Chulhee YunICML 2025
- A General Analysis of Example-Selection for Stochastic Gradient DescentYucheng Lu, Si Yi Meng, Christopher De SaICLR 2022 · 23 citations
- Tighter Convergence Bounds for Shuffled SGD via Primal-Dual PerspectiveXufeng Cai, Cheuk Yin Lin, Jelena DiakonikolasNeurIPS 2024 · 9 citations
- The benefits of full data shuffle, now with optimal I/O cost: -wise independence and matrix transposition to the rescuePeyman Afshani, Rezaul Chowdhury, Mayank Goswami, Jens Kristian R Schou et al.ICML 2026
