Benign Underfitting of Stochastic Gradient Descent
Tomer Koren, Roi Livni, Yishay Mansour, Uri Sherman
Abstract
We study to what extent may stochastic gradient descent (SGD) be understood as a"conventional"learning rule that achieves generalization performance by obtaining a good fit to training data. We consider the fundamental stochastic convex optimization framework, where (one pass, without-replacement) SGD is classically known to minimize the population risk at rate , and prove that, surprisingly, there exist problem instances where the SGD solution exhibits both empirical risk and generalization gap of . Consequently, it turns out that SGD is not algorithmically stable in any sense, and its generalization ability cannot be explained by uniform convergence or any other currently known generalization bound technique for that matter (other than that of its classical analysis). We then continue to analyze the closely related with-replacement SGD, for which we show that an analogous phenomenon does not occur and prove that its population risk does in fact converge at the optimal rate. Finally, we interpret our main results in the context of without-replacement SGD for finite-sum convex optimization problems, and derive upper and lower bounds for the multi-epoch regime that significantly improve upon previously known results.
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 9e3ffe8f-9007-4ac1-9bf0-0503c2b52ba5Cited by top-tier papers8
- Stability and Generalization Analysis of Gradient Methods for Shallow Neural NetworksYunwen Lei, Rong Jin, Yiming YingNeurIPS 2022 · 30 citations
- Information Complexity of Stochastic Convex Optimization: Applications to Generalization, Memorization, and TracingIdan Attias, Gintare Karolina Dziugaite, Mahdi Haghifam, Roi Livni et al.ICML 2024 · 6 citations
- Toward Better PAC-Bayes Bounds for Uniformly Stable AlgorithmsSijia Zhou, Yunwen Lei, Ata KabánNeurIPS 2023 · 4 citations
- Flat Minima and Generalization: Insights from Stochastic Convex OptimizationMatan Schliserman, Shira Vansover-Hager, Tomer KorenICML 2026 · 2 citations
- Improved Last-Iterate Convergence of Shuffling Gradient Methods for Nonsmooth Convex OptimizationZijian Liu, Zhengyuan ZhouICML 2025
Builds on6
- Stability of Stochastic Gradient Descent on Nonsmooth Convex LossesRaef Bassily, Vitaly Feldman, Cristóbal Guzmán, Kunal TalwarNeurIPS 2020 · 240 citations
- Random Reshuffling: Simple Analysis with Vast ImprovementsKonstantin Mishchenko, Ahmed Khaled, Peter RichtárikNeurIPS 2020 · 172 citations
- Closing the convergence gap of SGD without replacementShashank Rajput, Anant Gupta, Dimitris S. PapailiopoulosICML 2020 · 73 citations
- Can Implicit Bias Explain Generalization? Stochastic Convex Optimization as a Case StudyAssaf Dauber, Meir Feder, Tomer Koren, Roi LivniNeurIPS 2020 · 26 citations
- Random Shuffling Beats SGD Only After Many Epochs on Ill-Conditioned ProblemsItay Safran, Ohad ShamirNeurIPS 2021 · 24 citations
Related papers
- Rapid Overfitting of Multi-Pass SGD in Stochastic Convex OptimizationShira Vansover-Hager, Tomer Koren, Roi LivniICML 2025
- SGD with shuffling: optimal rates without component convexity and large epoch requirementsKwangjun Ahn, Chulhee Yun, Suvrit SraNeurIPS 2020 · 83 citations
- Risk Bounds of Multi-Pass SGD for Least Squares in the Interpolation RegimeDifan Zou, Jingfeng Wu, Vladimir Braverman, Quanquan Gu et al.NeurIPS 2022 · 9 citations
- SGD: The Role of Implicit Regularization, Batch-size and Multiple-epochsAyush Sekhari, Karthik Sridharan, Satyen KaleNeurIPS 2021 · 36 citations
- Stability and Generalization for Markov Chain Stochastic Gradient MethodsPuyu Wang, Yunwen Lei, Yiming Ying, Ding-Xuan ZhouNeurIPS 2022 · 26 citations
