Closed-form dynamics beyond quadratics – Machine Learning Research Blog
Quadratic functions are the workhorse of the analysis of iterative algorithms such as gradient-based optimization. They lead, in discrete and continuous time, to closed-form dynamics that treat all eigensubspaces of the Hessian matrix independently. This leads to simple math for understanding convergence behaviors (maximal step-size, condition number, acceleration, scaling laws for gradient descent or its stochastic version, etc.). An important success of the optimization field is to have shown that the same behaviors hold for convex functions, where the analysis now relies on Lyapunov functions rather than closed-form computations (see, e.g., this post on automated proofs), with some extensions to non-convex functions, but only regarding convergence to stationary points. Going (globally) non-convex is of major interest in different fields, including machine learning for neural networks or matrix factorizations. In this post, I try to answer the following question: What are dynamics th
Quadratic functions are the workhorse of the analysis of iterative algorithms such as gradient-based optimization. They lead, in discrete and continuous time, to closed-form dynamics that treat all eigensubspaces of the Hessian matrix independently. This leads to simple math for understanding convergence behaviors (maximal step-size, condition number, acceleration , scaling laws for gradient descent or its stochastic version , etc.). An important success of the optimization field is to have shown that the same behaviors hold for convex functions, where the analysis now relies on Lyapunov funct
Explore this link on the map →related reading
- Why Momentum Really Worksdistill.pub
- Deriving Muonjeremybernste.in
- Understanding the Neural Tangent Kernel – EigenTaleseigentales.com
- Modular Manifolds - Thinking Machines Labthinkingmachines.ai
- [2604.21691] There Will Be a Scientific Theory of Deep Learningarxiv.org
- https://www.deeplearningbook.org/contents/linear_algebra.htmldeeplearningbook.org
- NL.pdfabehrouz.github.io
- Pen and Paper Exercises in Machine Learningarxiv.org
- Some Math behind Neural Tangent Kernel | Lil'Loglilianweng.github.io
- Scaling laws of optimization – Machine Learning Research Blogfrancisbach.com
- [2605.01172] A Theory of Generalization in Deep Learningarxiv.org
- Statistical Mechanics of Deep Learningganguli-gang.stanford.edu