Permutation-Based SGD: Is Random Optimal?
Shashank Rajput, Kangwook Lee, Dimitris S. Papailiopoulos
摘要
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.
问问这篇 Paper
智能体会读完全文。
Lune 把这篇 Paper 索引到了每一个公式,引用它的顶会 Paper 也一样。你提问,回答直接引用原文。
引用它的顶会 Paper9
- Tighter Lower Bounds for Shuffling SGD: Random Permutations and BeyondJaeyoung Cha, Jaewook Lee, Chulhee YunICML 2023 · 被引用 26 次
- On the Last-Iterate Convergence of Shuffling Gradient MethodsZijian Liu, Zhengyuan ZhouICML 2024 · 被引用 11 次
- CD-GraB: Coordinating Distributed Example Orders for Provably Accelerated TrainingA. Feder Cooper, Wentao Guo, Khiem Pham, Tiancheng Yuan 等NeurIPS 2023 · 被引用 9 次
- Are Greedy Task Orderings Better Than Random in Continual Linear Regression?Matan Tsipory, Ran Levinstein, Itay Evron, Mark Kong 等NeurIPS 2025 · 被引用 5 次
- Stochastic Extragradient with Flip-Flop Shuffling & Anchoring: Provable ImprovementsJiseok Chae, Chulhee Yun, Donghwan KimNeurIPS 2024 · 被引用 2 次
它引用的顶会 Paper4
- Random Reshuffling: Simple Analysis with Vast ImprovementsKonstantin Mishchenko, Ahmed Khaled, Peter RichtárikNeurIPS 2020 · 被引用 172 次
- SGD with shuffling: optimal rates without component convexity and large epoch requirementsKwangjun Ahn, Chulhee Yun, Suvrit SraNeurIPS 2020 · 被引用 83 次
- Closing the convergence gap of SGD without replacementShashank Rajput, Anant Gupta, Dimitris S. PapailiopoulosICML 2020 · 被引用 73 次
- Random Reshuffling is Not Always BetterChristopher De SaNeurIPS 2020 · 被引用 27 次
相关 Paper
- 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 次
- Tighter Convergence Bounds for Shuffled SGD via Primal-Dual PerspectiveXufeng Cai, Cheuk Yin Lin, Jelena DiakonikolasNeurIPS 2024 · 被引用 9 次
- 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 等ICML 2026
