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
- Why Momentum Really Worksdistill.pub
- Machine Learning Research Blog – Francis Bachfrancisbach.com
- [2608.13335] Neural Quadratic Forms: A Unified Minimal Model for Sudden Learning and Scaling Lawsarxiv.org
- Scaling laws of optimization – Machine Learning Research Blogfrancisbach.com
- Modular Manifolds - Thinking Machines Labthinkingmachines.ai
- Understanding the Neural Tangent Kernel – EigenTaleseigentales.com
- Some Math behind Neural Tangent Kernel | Lil'Loglilianweng.github.io
- [2604.04891] Muon Dynamics as a Spectral Wasserstein Flowarxiv.org
- [2402.01258] Transformers Learn Nonlinear Features In Context: Nonconvex Mean-field Dynamics on the Attention Landscapearxiv.org
- Effortless optimization through gradient flows – Machine Learning Research Blogfrancisbach.com
- [2605.15522] Stochastic Non-Smooth Convex Optimization with Unbounded Gradientsarxiv.org
- [1605.08101] Global rates of convergence for nonconvex optimization on manifoldsarxiv.org