Reevaluating Theoretical Analysis Methods for Optimization in Deep Learning
Not on the map. Why?
Abstract (arXiv)
There is a significant gap between our theoretical understanding of optimization algorithms used in deep learning and their practical performance. Theoretical development usually focuses on proving convergence guarantees under a variety of different assumptions, which are themselves often chosen based on a rough combination of intuitive match to practice and analytical convenience. In this paper, we carefully measure the degree to which the standard optimization analyses are capable of explaining modern algorithms. To do this, we develop new empirical metrics that compare real optimization behavior with analytically predicted behavior. Our investigation is notable for its tight integration with modern optimization analysis: rather than simply checking high-level assumptions made in the analysis (e.g. smoothness), we also verify key low-level identities used by the analysis to explain optimization behavior that might hold even if the high-level motivating assumptions do not. Notably, we find that smoothness-based analyses fail in practice under most scenarios, but the key identities commonly used in convex-optimization analyses often hold in practice despite the objective's global non-convexity.
Key facts. Cited by 3 papers in the map (data through 7 Oct 2026). Builds on 52 of its 68 references that are in the map. First posted on arXiv 1 Jul 2024.
Builds on
52 of 68 references.
References · 52
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 2014Very Deep Convolutional Networks for Large-Scale Image Recognitioncited by 15,267
- 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,491
- 2020The Pile: An 800GB Dataset of Diverse Text for Language Modelingcited by 2,346
- 2017Visualizing the Loss Landscape of Neural Netscited by 1,126
- 2019On the Convergence of Adam and Beyondcited by 834
- 2019Introduction to Online Convex Optimizationcited by 793
- 2013Stochastic First- and Zeroth-order Methods for Nonconvex Stochastic Programmingcited by 716
- 2022Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasetscited by 599
- 2018Loss Surfaces, Mode Connectivity, and Fast Ensembling of DNNscited by 585
38 more
- 2014Convex Optimization: Algorithms and Complexitycited by 569
- 2014OpenML: networked science in machine learningcited by 523
- 2018Shampoo: Preconditioned Stochastic Tensor Optimizationcited by 413
- 2021Gradient Descent on Neural Networks Typically Occurs at the Edge of Stabilitycited by 413
- 2019Online Learning: A Modern Introduction Using Convex Optimizationcited by 407
- 2014SAGA: A Fast Incremental Gradient Method With Support for Non-Strongly Convex Composite Objectivescited by 401
- 2018How Does Batch Normalization Help Optimization?cited by 355
- 2019Why gradient clipping accelerates training: A theoretical justification for adaptivitycited by 335
- 2019Momentum-Based Variance Reduction in Non-Convex SGDcited by 302
- 2018SPIDER: Near-Optimal Non-Convex Optimization via Stochastic Path Integrated Differential Estimatorcited by 283
- 2014The Loss Surfaces of Multilayer Networkscited by 278
- 2023Sophia: A Scalable Stochastic Second-order Optimizer for Language Model Pre-trainingcited by 260
- 2020ADAHESSIAN: An Adaptive Second Order Optimizer for Machine Learningcited by 201
- 2014Qualitatively characterizing neural network optimization problemscited by 186
- 2018AdaGrad stepsizes: Sharp convergence over nonconvex landscapescited by 182
- 2018Stochastic model-based minimization of weakly convex functionscited by 157
- 2020Better Theory for SGD in the Nonconvex Worldcited by 156
- 2018On the Convergence of Stochastic Gradient Descent with Adaptive Stepsizescited by 155
- 2017Non-convex Optimization for Machine Learningcited by 132
- 2020Dynamic Regret of Convex and Smooth Functionscited by 90
- 2022The Slingshot Mechanism: An Empirical Study of Adaptive Optimizers and the Grokking Phenomenoncited by 90
- 2022Adaptive Gradient Methods at the Edge of Stabilitycited by 88
- 2018A Walk with SGDcited by 74
- 2016Variance Reduction for Faster Non-Convex Optimizationcited by 72
- 2022The Power of Adaptivity in SGD: Self-Tuning Step Sizes with Unbounded Gradients and Affine Variancecited by 60
- 2023Convergence of AdaGrad for Non-convex Objectives: Simple Proofs and Relaxed Assumptionscited by 55
- 2017"Convex Until Proven Guilty": Dimension-Free Acceleration of Gradient Descent on Non-Convex Functionscited by 46
- 2023Convex and Non-convex Optimization Under Generalized Smoothnesscited by 44
- 2023Optimal Stochastic Non-smooth Non-convex Optimization through Online-to-Non-convex Conversioncited by 42
- 2020Second-Order Information in Non-Convex Stochastic Optimization: Power and Limitationscited by 39
- 2020Convergence of a Stochastic Gradient Method with Momentum for Non-Smooth Non-Convex Optimizationcited by 32
- 2024Understanding Adam Optimizer via Online Learning of Updates: Adam is FTRL in Disguisecited by 23
- 2016An empirical analysis of the optimization of deep network loss surfacescited by 21
- 2024Random Scaling and Momentum for Non-smooth Non-convex Optimizationcited by 18
- 2021Better SGD using Second-order Momentumcited by 14
- 2020Complexity of Finding Stationary Points of Nonsmooth Nonconvex Functionscited by 13
- 2022On the Convergence of AdaGrad(Norm) on Rd: Beyond Convexity, Non-Asymptotic Rate and Accelerationcited by 13
- 2023Outliers with Opposing Signals Have an Outsized Effect on Neural Network Optimizationcited by 13
50 most cited of 52 All 52 (JSON)
Built on by
Citing papers per year: 2024: 1; 2025: 0; 2026: 2.
3 papers.