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
related reading
- [2608.13335] Neural Quadratic Forms: A Unified Minimal Model for Sudden Learning and Scaling Lawsarxiv.org
- Why Momentum Really Worksdistill.pub
- Deriving Muonjeremybernste.in
- Understanding the Neural Tangent Kernel – EigenTaleseigentales.com
- Machine Learning Research Blog – Francis Bachfrancisbach.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
- [2604.06366] Stochastic Gradient Descent in the Saddle-to-Saddle Regime of Deep Linear Networksarxiv.org
- Pen and Paper Exercises in Machine Learningarxiv.org
- NL.pdfabehrouz.github.io
- nn-notes.pdfboris-hanin.github.io