mlsanityData to 7 Oct 2026

Reevaluating Theoretical Analysis Methods for Optimization in Deep Learning

arXiv 2407.01825 ↗First posted 1 Jul 2024cs.LG math.OCcited by 3

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

  1. 2015Deep Residual Learning for Image Recognitioncited by 51,809
  2. 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
  3. 2014Adam: A Method for Stochastic Optimizationcited by 36,960
  4. 2014Very Deep Convolutional Networks for Large-Scale Image Recognitioncited by 15,267
  5. 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,491
  6. 2020The Pile: An 800GB Dataset of Diverse Text for Language Modelingcited by 2,346
  7. 2017Visualizing the Loss Landscape of Neural Netscited by 1,126
  8. 2019On the Convergence of Adam and Beyondcited by 834
  9. 2019Introduction to Online Convex Optimizationcited by 793
  10. 2013Stochastic First- and Zeroth-order Methods for Nonconvex Stochastic Programmingcited by 716
  11. 2022Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasetscited by 599
  12. 2018Loss Surfaces, Mode Connectivity, and Fast Ensembling of DNNscited by 585
38 more
  1. 2014Convex Optimization: Algorithms and Complexitycited by 569
  2. 2014OpenML: networked science in machine learningcited by 523
  3. 2018Shampoo: Preconditioned Stochastic Tensor Optimizationcited by 413
  4. 2021Gradient Descent on Neural Networks Typically Occurs at the Edge of Stabilitycited by 413
  5. 2019Online Learning: A Modern Introduction Using Convex Optimizationcited by 407
  6. 2014SAGA: A Fast Incremental Gradient Method With Support for Non-Strongly Convex Composite Objectivescited by 401
  7. 2018How Does Batch Normalization Help Optimization?cited by 355
  8. 2019Why gradient clipping accelerates training: A theoretical justification for adaptivitycited by 335
  9. 2019Momentum-Based Variance Reduction in Non-Convex SGDcited by 302
  10. 2018SPIDER: Near-Optimal Non-Convex Optimization via Stochastic Path Integrated Differential Estimatorcited by 283
  11. 2014The Loss Surfaces of Multilayer Networkscited by 278
  12. 2023Sophia: A Scalable Stochastic Second-order Optimizer for Language Model Pre-trainingcited by 260
  13. 2020ADAHESSIAN: An Adaptive Second Order Optimizer for Machine Learningcited by 201
  14. 2014Qualitatively characterizing neural network optimization problemscited by 186
  15. 2018AdaGrad stepsizes: Sharp convergence over nonconvex landscapescited by 182
  16. 2018Stochastic model-based minimization of weakly convex functionscited by 157
  17. 2020Better Theory for SGD in the Nonconvex Worldcited by 156
  18. 2018On the Convergence of Stochastic Gradient Descent with Adaptive Stepsizescited by 155
  19. 2017Non-convex Optimization for Machine Learningcited by 132
  20. 2020Dynamic Regret of Convex and Smooth Functionscited by 90
  21. 2022The Slingshot Mechanism: An Empirical Study of Adaptive Optimizers and the Grokking Phenomenoncited by 90
  22. 2022Adaptive Gradient Methods at the Edge of Stabilitycited by 88
  23. 2018A Walk with SGDcited by 74
  24. 2016Variance Reduction for Faster Non-Convex Optimizationcited by 72
  25. 2022The Power of Adaptivity in SGD: Self-Tuning Step Sizes with Unbounded Gradients and Affine Variancecited by 60
  26. 2023Convergence of AdaGrad for Non-convex Objectives: Simple Proofs and Relaxed Assumptionscited by 55
  27. 2017"Convex Until Proven Guilty": Dimension-Free Acceleration of Gradient Descent on Non-Convex Functionscited by 46
  28. 2023Convex and Non-convex Optimization Under Generalized Smoothnesscited by 44
  29. 2023Optimal Stochastic Non-smooth Non-convex Optimization through Online-to-Non-convex Conversioncited by 42
  30. 2020Second-Order Information in Non-Convex Stochastic Optimization: Power and Limitationscited by 39
  31. 2020Convergence of a Stochastic Gradient Method with Momentum for Non-Smooth Non-Convex Optimizationcited by 32
  32. 2024Understanding Adam Optimizer via Online Learning of Updates: Adam is FTRL in Disguisecited by 23
  33. 2016An empirical analysis of the optimization of deep network loss surfacescited by 21
  34. 2024Random Scaling and Momentum for Non-smooth Non-convex Optimizationcited by 18
  35. 2021Better SGD using Second-order Momentumcited by 14
  36. 2020Complexity of Finding Stationary Points of Nonsmooth Nonconvex Functionscited by 13
  37. 2022On the Convergence of AdaGrad(Norm) on Rd: Beyond Convexity, Non-Asymptotic Rate and Accelerationcited by 13
  38. 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

Cited by year (2026 to 7 Oct)

Citing papers per year: 2024: 1; 2025: 0; 2026: 2.

3 papers.