Second-order regression models exhibit progressive sharpening to the edge of stability
Not on the map. Why?
Abstract (arXiv)
Recent studies of gradient descent with large step sizes have shown that there is often a regime with an initial increase in the largest eigenvalue of the loss Hessian (progressive sharpening), followed by a stabilization of the eigenvalue near the maximum value which allows convergence (edge of stability). These phenomena are intrinsically non-linear and do not happen for models in the constant Neural Tangent Kernel (NTK) regime, for which the predictive function is approximately linear in the parameters. As such, we consider the next simplest class of predictive models, namely those that are quadratic in the parameters, which we call second-order regression models. For quadratic objectives in two dimensions, we prove that this second-order regression model exhibits progressive sharpening of the NTK eigenvalue towards a value that differs slightly from the edge of stability, which we explicitly compute. In higher dimensions, the model generically shows similar behavior, even without the specific structure of a neural network, suggesting that progressive sharpening and edge-of-stability behavior aren't unique features of neural networks, and could be a more general property of discrete learning algorithms in high-dimensional non-linear models.
Key facts. Cited by 40 papers in the map (data through 7 Oct 2026). Builds on 24 of its 28 references that are in the map. First posted on arXiv 10 Oct 2022.
Builds on
24 of 28 references.
References · 24
- 2018Neural Tangent Kernel: Convergence and Generalization in Neural Networkscited by 2,597
- 2020Sharpness-Aware Minimization for Efficiently Improving Generalizationcited by 1,404
- 2018Reconciling modern machine learning practice and the bias-variance trade-offcited by 919
- 2019Wide Neural Networks of Any Depth Evolve as Linear Models Under Gradient Descentcited by 732
- 2017Exploring Generalization in Deep Learningcited by 595
- 2021Gradient Descent on Neural Networks Typically Occurs at the Edge of Stabilitycited by 413
- 2019An Investigation into Neural Net Optimization via Hessian Eigenvalue Densitycited by 289
- 2022Tensor Programs V: Tuning Large Neural Networks via Zero-Shot Hyperparameter Transfercited by 272
- 2020The large learning rate phase of deep learning: the catapult mechanismcited by 206
- 2021The Principles of Deep Learning Theorycited by 169
- 2022Self-Stabilization: The Implicit Bias of Gradient Descent at the Edge of Stabilitycited by 151
- 2019Neural Tangents: Fast and Easy Infinite Neural Networks in Pythoncited by 149
12 more
- 2019Tensor Programs I: Wide Feedforward or Recurrent Neural Networks of Any Architecture are Gaussian Processescited by 148
- 2022Self-Consistent Dynamical Field Theory of Kernel Evolution in Wide Neural Networkscited by 104
- 2020The Neural Tangent Kernel in High Dimensions: Triple Descent and a Multi-Scale Theory of Generalizationcited by 101
- 2019Dynamics of Deep Neural Networks and Neural Tangent Hierarchycited by 96
- 2022Adaptive Gradient Methods at the Edge of Stabilitycited by 88
- 2019Beyond Linearization: On Quadratic and Higher-Order Approximation of Wide Neural Networkscited by 79
- 2022Analyzing Sharpness along GD Trajectory: Progressive Sharpening and Edge of Stabilitycited by 61
- 2022Understanding Edge-of-Stability Training Dynamics with a Minimalist Examplecited by 54
- 2020Multiple Descent: Design Your Own Generalization Curvecited by 44
- 2019The asymptotic spectrum of the Hessian of DNN throughout trainingcited by 19
- 2022Quadratic models for understanding catapult dynamics of neural networkscited by 17
- 2019At Stability's Edge: How to Adjust Hyperparameters to Preserve Minima Selection in Asynchronous Training of Neural Networks?cited by 16
Built on by
Citing papers per year: 2023: 13; 2024: 11; 2025: 7; 2026: 9.
40 papers.
Sort:
By area:
- 2024Normalization and effective learning rates in reinforcement learningcited by 55
- 2023Dynamics of Finite Width Kernel and Prediction Fluctuations in Mean Field Neural Networkscited by 42
- 2023Catapults in SGD: spikes in the training loss and their impact on generalization through feature learningcited by 29
- 2023Hitting the High-Dimensional Notes: An ODE for SGD learning dynamics on GLMs and multi-index modelscited by 25
- 2023SAM operates far from home: eigenvalue regularization as a dynamical phenomenoncited by 24
- 2024Super Consistency of Neural Network Landscapes and Learning Rate Transfercited by 23
- 2024Rethinking Conventional Wisdom in Machine Learning: From Generalization to Scalingcited by 21
- 2023Gradient Descent Monotonically Decreases the Sharpness of Gradient Flow Solutions in Scalar Networks and Beyondcited by 20
- 2023Phase diagram of early training dynamics in deep neural networks: effect of the learning rate, depth, and widthcited by 19
- 2024Deep linear networks for regression are implicitly regularized towards flat minimacited by 17
- 2023Sharpness-Aware Minimization and the Edge of Stabilitycited by 16
- 2023From Stability to Chaos: Analyzing Gradient Descent Dynamics in Quadratic Regressioncited by 16
28 more
- 2023Loss Spike in Training Neural Networkscited by 15
- 2023How connectivity structure shapes rich and lazy learning in neural circuitscited by 15
- 2024High dimensional analysis reveals conservative sharpening and a stochastic edge of stabilitycited by 15
- 2024Learning Associative Memories with Gradient Descentcited by 14
- 2024The Optimization Landscape of SGD Across the Feature Learning Strengthcited by 13
- 2024Neglected Hessian component explains mysteries in Sharpness regularizationcited by 12
- 2024Feature learning as alignment: a structural property of gradient descent in non-linear neural networkscited by 9
- 2023Catapult Dynamics and Phase Transitions in Quadratic Netscited by 8
- 2024Stepping on the Edge: Curvature Aware Learning Rate Tunerscited by 8
- 2025What Can Grokking Teach Us About Learning Under Nonstationarity?cited by 7
- 2026Rod Flow: A Continuous-Time Model for Gradient Descent at the Edge of Stabilitycited by 6
- 2026The Origin of Edge of Stabilitycited by 6
- 2024To Clip or not to Clip: the Dynamics of SGD with Gradient Clipping in High-Dimensionscited by 5
- 2025Understanding Sharpness Dynamics in NN Training with a Minimalist Example: The Effects of Dataset Difficulty, Depth, Stochasticity, and Morecited by 5
- 2025Gradient Descent with Large Step Sizes: Chaos and Fractal Convergence Regioncited by 5
- 2023On the Interplay Between Stepsize Tuning and Progressive Sharpeningcited by 4
- 2025A Minimalist Example of Edge-of-Stability and Progressive Sharpeningcited by 4
- 2025Convergence Rates for Gradient Descent on the Edge of Stability in Overparametrised Least Squarescited by 3
- 2023Gradient Descent with Polyak's Momentum Finds Flatter Minima via Large Catapultscited by 2
- 2025Variational Learning Finds Flatter Solutions at the Edge of Stabilitycited by 2
- 2026Does SGD Seek Flatness or Sharpness? An Exactly Solvable Modelcited by 1
- 2026SGD at the Edge of Stability: The Stochastic Sharpness Gapcited by 1
- 2026Dynamics of Gradient Descent with Large Step Size Near a Manifold of Flat Minimacited by 1
- 2026The Road Taken: The Role of Optimizers at the Edge of Stabilitycited by 1
- 2025A Special Case of Quadratic Extrapolation Under the Neural Tangent Kernelcited by 0
- 2026Centre manifold theorem for maps along manifolds of fixed pointscited by 0
- 2026The Multiple Timescales of Gradient Descent on the Edge of Stability: A Perturbative Derivation of the Central Flowcited by 0
- 2026A Journey to the Edge of Stabilitycited by 0
- 2026A Journey to the Edge of Stability26 Sep 2026
- 2026The Multiple Timescales of Gradient Descent on the Edge of Stability: A Perturbative Derivation of the Central Flow1 Sep 2026
- 2026The Road Taken: The Role of Optimizers at the Edge of Stability19 Aug 2026
- 2026Dynamics of Gradient Descent with Large Step Size Near a Manifold of Flat Minima9 Jul 2026
- 2026SGD at the Edge of Stability: The Stochastic Sharpness Gap22 Apr 2026
- 2026The Origin of Edge of Stability22 Apr 2026
- 2026Centre manifold theorem for maps along manifolds of fixed points20 Apr 2026
- 2026Does SGD Seek Flatness or Sharpness? An Exactly Solvable Model4 Feb 2026
- 2026Rod Flow: A Continuous-Time Model for Gradient Descent at the Edge of Stability1 Feb 2026
- 2025A Special Case of Quadratic Extrapolation Under the Neural Tangent Kernel11 Dec 2025
- 2025Convergence Rates for Gradient Descent on the Edge of Stability in Overparametrised Least Squares20 Oct 2025
- 2025Gradient Descent with Large Step Sizes: Chaos and Fractal Convergence Region29 Sep 2025
28 more
- 2025What Can Grokking Teach Us About Learning Under Nonstationarity?26 Jul 2025
- 2025Variational Learning Finds Flatter Solutions at the Edge of Stability15 Jun 2025
- 2025Understanding Sharpness Dynamics in NN Training with a Minimalist Example: The Effects of Dataset Difficulty, Depth, Stochasticity, and More7 Jun 2025
- 2025A Minimalist Example of Edge-of-Stability and Progressive Sharpening4 Mar 2025
- 2024The Optimization Landscape of SGD Across the Feature Learning Strength6 Oct 2024
- 2024Rethinking Conventional Wisdom in Machine Learning: From Generalization to Scaling23 Sep 2024
- 2024Stepping on the Edge: Curvature Aware Learning Rate Tuners8 Jul 2024
- 2024Normalization and effective learning rates in reinforcement learning1 Jul 2024
- 2024To Clip or not to Clip: the Dynamics of SGD with Gradient Clipping in High-Dimensions17 Jun 2024
- 2024Deep linear networks for regression are implicitly regularized towards flat minima22 May 2024
- 2024High dimensional analysis reveals conservative sharpening and a stochastic edge of stability30 Apr 2024
- 2024Learning Associative Memories with Gradient Descent28 Feb 2024
- 2024Super Consistency of Neural Network Landscapes and Learning Rate Transfer27 Feb 2024
- 2024Feature learning as alignment: a structural property of gradient descent in non-linear neural networks7 Feb 2024
- 2024Neglected Hessian component explains mysteries in Sharpness regularization19 Jan 2024
- 2023On the Interplay Between Stepsize Tuning and Progressive Sharpening30 Nov 2023
- 2023Gradient Descent with Polyak's Momentum Finds Flatter Minima via Large Catapults25 Nov 2023
- 2023How connectivity structure shapes rich and lazy learning in neural circuits12 Oct 2023
- 2023From Stability to Chaos: Analyzing Gradient Descent Dynamics in Quadratic Regression2 Oct 2023
- 2023Sharpness-Aware Minimization and the Edge of Stability21 Sep 2023
- 2023Hitting the High-Dimensional Notes: An ODE for SGD learning dynamics on GLMs and multi-index models17 Aug 2023
- 2023Catapults in SGD: spikes in the training loss and their impact on generalization through feature learning7 Jun 2023
- 2023Gradient Descent Monotonically Decreases the Sharpness of Gradient Flow Solutions in Scalar Networks and Beyond22 May 2023
- 2023Loss Spike in Training Neural Networks20 May 2023
- 2023Dynamics of Finite Width Kernel and Prediction Fluctuations in Mean Field Neural Networks6 Apr 2023
- 2023Phase diagram of early training dynamics in deep neural networks: effect of the learning rate, depth, and width23 Feb 2023
- 2023SAM operates far from home: eigenvalue regularization as a dynamical phenomenon17 Feb 2023
- 2023Catapult Dynamics and Phase Transitions in Quadratic Nets18 Jan 2023
- 2024Normalization and effective learning rates in reinforcement learningcited by 55
- 2023Dynamics of Finite Width Kernel and Prediction Fluctuations in Mean Field Neural Networkscited by 42
- 2023Catapults in SGD: spikes in the training loss and their impact on generalization through feature learningcited by 29
- 2023Hitting the High-Dimensional Notes: An ODE for SGD learning dynamics on GLMs and multi-index modelscited by 25
- 2023SAM operates far from home: eigenvalue regularization as a dynamical phenomenoncited by 24
- 2024Super Consistency of Neural Network Landscapes and Learning Rate Transfercited by 23
- 2023Gradient Descent Monotonically Decreases the Sharpness of Gradient Flow Solutions in Scalar Networks and Beyondcited by 20
- 2023Phase diagram of early training dynamics in deep neural networks: effect of the learning rate, depth, and widthcited by 19
- 2024Deep linear networks for regression are implicitly regularized towards flat minimacited by 17
- 2023Sharpness-Aware Minimization and the Edge of Stabilitycited by 16
- 2023From Stability to Chaos: Analyzing Gradient Descent Dynamics in Quadratic Regressioncited by 16
- 2023Loss Spike in Training Neural Networkscited by 15
12 most cited of 39 All 39 (JSON)
- 2026A Journey to the Edge of Stability26 Sep 2026
- 2026The Multiple Timescales of Gradient Descent on the Edge of Stability: A Perturbative Derivation of the Central Flow1 Sep 2026
- 2026The Road Taken: The Role of Optimizers at the Edge of Stability19 Aug 2026
- 2026Dynamics of Gradient Descent with Large Step Size Near a Manifold of Flat Minima9 Jul 2026
- 2026SGD at the Edge of Stability: The Stochastic Sharpness Gap22 Apr 2026
- 2026The Origin of Edge of Stability22 Apr 2026
- 2026Centre manifold theorem for maps along manifolds of fixed points20 Apr 2026
- 2026Does SGD Seek Flatness or Sharpness? An Exactly Solvable Model4 Feb 2026
- 2026Rod Flow: A Continuous-Time Model for Gradient Descent at the Edge of Stability1 Feb 2026
- 2025A Special Case of Quadratic Extrapolation Under the Neural Tangent Kernel11 Dec 2025
- 2025Convergence Rates for Gradient Descent on the Edge of Stability in Overparametrised Least Squares20 Oct 2025
- 2025Gradient Descent with Large Step Sizes: Chaos and Fractal Convergence Region29 Sep 2025
12 newest of 39 All 39 (JSON)