Incremental Gradient Descent with Small Epoch Counts is Surprisingly Slow on Ill-Conditioned Problems
Yujun Kim, Jaeyoung Cha, Chulhee Yun
Abstract
Recent theoretical results demonstrate that the convergence rates of permutation-based SGD (e.g., random reshuffling SGD) are faster than uniform-sampling SGD; however, these studies focus mainly on the large epoch regime, where the number of epochs K exceeds the condition number κ. In contrast, little is known when K is smaller than κ, and it is still a challenging open question whether permutation-based SGD can converge faster in this small epoch regime (Safran & Shamir, 2021) . As a step toward understanding this gap, we study the naive deterministic variant, Incremental Gradient Descent (IGD), on smooth and strongly convex functions. Our lower bounds reveal that for the small epoch regime, IGD can exhibit surprisingly slow convergence even when all component functions are strongly convex. Furthermore, when some component functions are allowed to be nonconvex, we prove that the optimality gap of IGD can be significantly worse throughout the small epoch regime. Our analyses reveal that the convergence properties of permutationbased SGD in the small epoch regime may vary drastically depending on the assumptions on component functions. Lastly, we supplement the paper with tight upper and lower bounds for IGD in the large epoch regime.
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 papers1
Ask how each one uses itBuilds on11
- 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
- Revisiting the Last-Iterate Convergence of Stochastic Gradient MethodsZijian Liu, Zhengyuan ZhouICLR 2024 · 32 citations
- Tighter Lower Bounds for Shuffling SGD: Random Permutations and BeyondJaeyoung Cha, Jaewook Lee, Chulhee YunICML 2023 · 26 citations
Related papers
- Random Reshuffling: Simple Analysis with Vast ImprovementsKonstantin Mishchenko, Ahmed Khaled, Peter RichtárikNeurIPS 2020 · 172 citations
- Permutation-Based SGD: Is Random Optimal?Shashank Rajput, Kangwook Lee, Dimitris S. PapailiopoulosICLR 2022 · 15 citations
- On the Last-Iterate Convergence of Shuffling Gradient MethodsZijian Liu, Zhengyuan ZhouICML 2024 · 11 citations
- Tighter Convergence Bounds for Shuffled SGD via Primal-Dual PerspectiveXufeng Cai, Cheuk Yin Lin, Jelena DiakonikolasNeurIPS 2024 · 9 citations
- On Convergence of Incremental Gradient for Non-convex Smooth FunctionsAnastasia Koloskova, Nikita Doikov, Sebastian U. Stich, Martin JaggiICML 2024 · 6 citations
