Analyzing Sharpness along GD Trajectory: Progressive Sharpening and Edge of Stability
Not on the map. Why?
Abstract (arXiv)
Recent findings (e.g., arXiv:2103.00065) demonstrate that modern neural networks trained by full-batch gradient descent typically enter a regime called Edge of Stability (EOS). In this regime, the sharpness, i.e., the maximum Hessian eigenvalue, first increases to the value 2/(step size) (the progressive sharpening phase) and then oscillates around this value (the EOS phase). This paper aims to analyze the GD dynamics and the sharpness along the optimization trajectory. Our analysis naturally divides the GD trajectory into four phases depending on the change of the sharpness. We empirically identify the norm of output layer weight as an interesting indicator of sharpness dynamics. Based on this empirical observation, we attempt to theoretically and empirically explain the dynamics of various key quantities that lead to the change of sharpness in each phase of EOS. Moreover, based on certain assumptions, we provide a theoretical proof of the sharpness behavior in EOS regime in two-layer fully-connected linear neural networks. We also discuss some other empirical findings and the limitation of our theoretical results.
Key facts. Cited by 63 papers in the map (data through 14 Oct 2026). Builds on 26 of its 37 references that are in the map. First posted on arXiv 26 Jul 2022.
Builds on
26 of 37 references.
References · 26
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,305
- 2018Neural Tangent Kernel: Convergence and Generalization in Neural Networkscited by 2,621
- 2020Sharpness-Aware Minimization for Efficiently Improving Generalizationcited by 1,417
- 2016Optimization Methods for Large-Scale Machine Learningcited by 1,150
- 2019Wide Neural Networks of Any Depth Evolve as Linear Models Under Gradient Descentcited by 738
- 2018On Lazy Training in Differentiable Programmingcited by 675
- 2018Gradient Descent Provably Optimizes Over-parameterized Neural Networkscited by 659
- 2018Gradient Descent Finds Global Minima of Deep Neural Networkscited by 574
- 2019Fine-Grained Analysis of Optimization and Generalization for Overparameterized Two-Layer Neural Networkscited by 566
- 2021Gradient Descent on Neural Networks Typically Occurs at the Edge of Stabilitycited by 427
- 2017Empirical Analysis of the Hessian of Over-Parametrized Neural Networkscited by 290
14 more
- 2020The large learning rate phase of deep learning: the catapult mechanismcited by 216
- 2016Eigenvalues of the Hessian in Deep Learning: Singularity and Beyondcited by 159
- 2020The Break-Even Point on Optimization Trajectories of Deep Neural Networkscited by 150
- 2022Understanding Gradient Descent on Edge of Stability in Deep Learningcited by 128
- 2019Polylogarithmic width suffices for gradient descent to achieve arbitrarily small test error with shallow ReLU networkscited by 121
- 2021What Happens after SGD Reaches Zero Loss? --A Mathematical Frameworkcited by 107
- 2022Understanding the unstable convergence of gradient descentcited by 86
- 2022Understanding the Generalization Benefit of Normalization Layers: Sharpness Reductioncited by 86
- 2018A Walk with SGDcited by 78
- 2019Measurements of Three-Level Hierarchical Structure in the Outliers in the Spectrum of Deepnet Hessianscited by 68
- 2020The Surprising Simplicity of the Early-Time Learning Dynamics of Neural Networkscited by 58
- 2020Dissecting Hessian: Understanding Common Structure of Hessian in Neural Networkscited by 45
- 2019Pathological spectra of the Fisher information metric and its variants in deep neural networkscited by 32
- 2019The Normalization Method for Alleviating Pathological Sharpness in Wide Neural Networkscited by 20
Built on by
Citing papers per year: 2022: 6; 2023: 18; 2024: 9; 2025: 12; 2026: 18.
63 papers.
Sort:
By area:
- 2022Self-Stabilization: The Implicit Bias of Gradient Descent at the Edge of Stabilitycited by 155
- 2022Understanding Edge-of-Stability Training Dynamics with a Minimalist Examplecited by 56
- 2022Learning threshold neurons via the "edge of stability"cited by 51
- 2023Implicit Bias of Gradient Descent for Logistic Regression at the Edge of Stabilitycited by 42
- 2022Second-order regression models exhibit progressive sharpening to the edge of stabilitycited by 40
- 2024Large Stepsize Gradient Descent for Logistic Loss: Non-Monotonicity of the Loss Improves Optimization Efficiencycited by 35
- 2023Catapults in SGD: spikes in the training loss and their impact on generalization through feature learningcited by 31
- 2023The Crucial Role of Normalization in Sharpness-Aware Minimizationcited by 26
- 2023Trajectory Alignment: Understanding the Edge of Stability Phenomenon via Bifurcation Theorycited by 26
- 2023Good regularity creates large learning rate implicit biases: edge of stability, balancing, and catapultcited by 25
- 2023Universal Sharpness Dynamics in Neural Network Training: Fixed Point Analysis, Edge of Stability, and Route to Chaoscited by 22
- 2024Transformers Provably Learn Sparse Token Selection While Fully-Connected Nets Cannotcited by 22
38 more
- 2023Gradient Descent Monotonically Decreases the Sharpness of Gradient Flow Solutions in Scalar Networks and Beyondcited by 21
- 2023Phase diagram of early training dynamics in deep neural networks: effect of the learning rate, depth, and widthcited by 19
- 2023When and Why Momentum Accelerates SGD:An Empirical Studycited by 19
- 2024Deep linear networks for regression are implicitly regularized towards flat minimacited by 17
- 2024Large Stepsize Gradient Descent for Non-Homogeneous Two-Layer Networks: Margin Improvement and Fast Optimizationcited by 17
- 2023Loss Spike in Training Neural Networkscited by 16
- 2023Irreducible Curriculum for Language Model Pretrainingcited by 16
- 2022Beyond Losses Reweighting: Empowering Multi-Task Learning via the Generalization Perspectivecited by 15
- 2023Understanding Multi-phase Optimization Dynamics and Rich Nonlinear Behaviors of ReLU Networkscited by 15
- 2023Outliers with Opposing Signals Have an Outsized Effect on Neural Network Optimizationcited by 13
- 2025Towards Quantifying the Hessian Structure of Neural Networkscited by 13
- 2025Adaptive Preconditioners Trigger Loss Spikes in Adamcited by 13
- 2022Maximal Initial Learning Rates in Deep ReLU Networkscited by 11
- 2025Through the River: Understanding the Benefit of Schedule-Free Methods for Language Model Trainingcited by 9
- 2023On progressive sharpening, flat minima and generalisationcited by 7
- 2025Minimax Optimal Convergence of Gradient Descent in Logistic Regression via Large and Adaptive Stepsizescited by 7
- 2023The Inductive Bias of Flatness Regularization for Deep Matrix Factorizationcited by 6
- 2025Zeroth-Order Optimization Finds Flat Minimacited by 6
- 2025Understanding the Evolution of the Neural Tangent Kernel at the Edge of Stabilitycited by 6
- 2025Understanding Sharpness Dynamics in NN Training with a Minimalist Example: The Effects of Dataset Difficulty, Depth, Stochasticity, and Morecited by 5
- 2024Deconstructing the Goldilocks Zone of Neural Network Initializationcited by 4
- 2024Simplicity Bias via Global Convergence of Sharpness Minimizationcited 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 4
- 2023Gradient Descent with Polyak's Momentum Finds Flatter Minima via Large Catapultscited by 3
- 2025New Evidence of the Two-Phase Learning Dynamics of Neural Networkscited by 3
- 2026On the Stability of Nonlinear Dynamics in GD and SGD: Beyond Quadratic Potentialscited by 3
- 2024GD doesn't make the cut: Three ways that non-differentiability affects neural network trainingcited by 2
- 2025Large Learning Rates Simultaneously Achieve Robustness to Spurious Correlations and Compressibilitycited by 2
- 2026Generalization at the Edge of Stabilitycited by 2
- 2026Towards Understanding the Power and Limits of the Muon Optimizer: A River-Valley Perspectivecited by 2
- 2024Training on the Edge of Stability Is Caused by Layerwise Jacobian Alignmentcited by 1
- 2025A Physics-Inspired Optimizer: Velocity Regularized Adamcited by 1
- 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
- 2026Weight-norm Criticality: A Mechanism for Loss Spikes Induced by the Normalization and Weight Decaycited by 1
- 2026The Road Taken: The Role of Optimizers at the Edge of Stabilitycited by 1
50 most cited of 63 All 63 (JSON)
- 2026Can Representation Learning Decouple from Loss Minimization? Polar Updates Have an Answer28 Sep 2026
- 2026Muon Sublates the Edge of Stability in LLM Pretraining28 Sep 2026
- 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
- 2026Adam at the Edge of Stability: Adaptive Feedback, Provable Oscillation, and Gradient Reversal21 Aug 2026
- 2026The Road Taken: The Role of Optimizers at the Edge of Stability19 Aug 2026
- 2026Weight-norm Criticality: A Mechanism for Loss Spikes Induced by the Normalization and Weight Decay23 Jul 2026
- 2026Dynamics of Gradient Descent with Large Step Size Near a Manifold of Flat Minima9 Jul 2026
- 2026Towards Understanding the Power and Limits of the Muon Optimizer: A River-Valley Perspective19 Jun 2026
- 2026Taming Curvature: Architecture Warm-Up for Stable Transformer Training15 Jun 2026
- 2026Improving Adversarial Robustness of Attribution via Implicit Regularization28 May 2026
- 2026SGD at the Edge of Stability: The Stochastic Sharpness Gap22 Apr 2026
38 more
- 2026Generalization at the Edge of Stability21 Apr 2026
- 2026Centre manifold theorem for maps along manifolds of fixed points20 Apr 2026
- 2026A Mechanism Study of Delayed Loss Spikes in Batch-Normalized Linear Models18 Apr 2026
- 2026Zeroth-Order Optimization at the Edge of Stability16 Apr 2026
- 2026On the Stability of Nonlinear Dynamics in GD and SGD: Beyond Quadratic Potentials16 Feb 2026
- 2026Does SGD Seek Flatness or Sharpness? An Exactly Solvable Model4 Feb 2026
- 2025Convergence Rates for Gradient Descent on the Edge of Stability in Overparametrised Least Squares20 Oct 2025
- 2025Large Learning Rates Simultaneously Achieve Robustness to Spurious Correlations and Compressibility23 Jul 2025
- 2025Understanding the Evolution of the Neural Tangent Kernel at the Edge of Stability17 Jul 2025
- 2025Through the River: Understanding the Benefit of Schedule-Free Methods for Language Model Training14 Jul 2025
- 2025Understanding Sharpness Dynamics in NN Training with a Minimalist Example: The Effects of Dataset Difficulty, Depth, Stochasticity, and More7 Jun 2025
- 2025Zeroth-Order Optimization Finds Flat Minima5 Jun 2025
- 2025Adaptive Preconditioners Trigger Loss Spikes in Adam5 Jun 2025
- 2025New Evidence of the Two-Phase Learning Dynamics of Neural Networks20 May 2025
- 2025A Physics-Inspired Optimizer: Velocity Regularized Adam19 May 2025
- 2025Towards Quantifying the Hessian Structure of Neural Networks5 May 2025
- 2025Minimax Optimal Convergence of Gradient Descent in Logistic Regression via Large and Adaptive Stepsizes5 Apr 2025
- 2025A Minimalist Example of Edge-of-Stability and Progressive Sharpening4 Mar 2025
- 2024Criteria and Bias of Parameterized Linear Regression under Edge of Stability Regime11 Dec 2024
- 2024Simplicity Bias via Global Convergence of Sharpness Minimization21 Oct 2024
- 2024Large Stepsize Gradient Descent for Non-Homogeneous Two-Layer Networks: Margin Improvement and Fast Optimization12 Jun 2024
- 2024Transformers Provably Learn Sparse Token Selection While Fully-Connected Nets Cannot11 Jun 2024
- 2024Training on the Edge of Stability Is Caused by Layerwise Jacobian Alignment31 May 2024
- 2024Deep linear networks for regression are implicitly regularized towards flat minima22 May 2024
- 2024Large Stepsize Gradient Descent for Logistic Loss: Non-Monotonicity of the Loss Improves Optimization Efficiency24 Feb 2024
- 2024Deconstructing the Goldilocks Zone of Neural Network Initialization5 Feb 2024
- 2024GD doesn't make the cut: Three ways that non-differentiability affects neural network training16 Jan 2024
- 2023Gradient Descent with Polyak's Momentum Finds Flatter Minima via Large Catapults25 Nov 2023
- 2023Outliers with Opposing Signals Have an Outsized Effect on Neural Network Optimization7 Nov 2023
- 2023Universal Sharpness Dynamics in Neural Network Training: Fixed Point Analysis, Edge of Stability, and Route to Chaos3 Nov 2023
- 2023Good regularity creates large learning rate implicit biases: edge of stability, balancing, and catapult26 Oct 2023
- 2023Irreducible Curriculum for Language Model Pretraining23 Oct 2023
- 2023Learning Stochastic Dynamical Systems as an Implicit Regularization with Graph Neural Networks12 Jul 2023
- 2023Trajectory Alignment: Understanding the Edge of Stability Phenomenon via Bifurcation Theory9 Jul 2023
- 2023The Inductive Bias of Flatness Regularization for Deep Matrix Factorization22 Jun 2023
- 2023When and Why Momentum Accelerates SGD:An Empirical Study15 Jun 2023
- 2023Catapults in SGD: spikes in the training loss and their impact on generalization through feature learning7 Jun 2023
- 2023SANE: The phases of gradient descent through Sharpness Adjusted Number of Effective parameters29 May 2023
50 newest of 63 All 63 (JSON)
- 2025Zeroth-Order Optimization Finds Flat Minimacited by 6
- 2025Zeroth-Order Optimization Finds Flat Minima5 Jun 2025
- 2022Self-Stabilization: The Implicit Bias of Gradient Descent at the Edge of Stabilitycited by 155
- 2022Understanding Edge-of-Stability Training Dynamics with a Minimalist Examplecited by 56
- 2022Learning threshold neurons via the "edge of stability"cited by 51
- 2023Implicit Bias of Gradient Descent for Logistic Regression at the Edge of Stabilitycited by 42
- 2022Second-order regression models exhibit progressive sharpening to the edge of stabilitycited by 40
- 2024Large Stepsize Gradient Descent for Logistic Loss: Non-Monotonicity of the Loss Improves Optimization Efficiencycited by 35
- 2023Catapults in SGD: spikes in the training loss and their impact on generalization through feature learningcited by 31
- 2023The Crucial Role of Normalization in Sharpness-Aware Minimizationcited by 26
- 2023Trajectory Alignment: Understanding the Edge of Stability Phenomenon via Bifurcation Theorycited by 26
- 2023Good regularity creates large learning rate implicit biases: edge of stability, balancing, and catapultcited by 25
- 2023Universal Sharpness Dynamics in Neural Network Training: Fixed Point Analysis, Edge of Stability, and Route to Chaoscited by 22
- 2024Transformers Provably Learn Sparse Token Selection While Fully-Connected Nets Cannotcited by 22
12 most cited of 61 All 61 (JSON)
- 2026Muon Sublates the Edge of Stability in LLM Pretraining28 Sep 2026
- 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
- 2026Adam at the Edge of Stability: Adaptive Feedback, Provable Oscillation, and Gradient Reversal21 Aug 2026
- 2026The Road Taken: The Role of Optimizers at the Edge of Stability19 Aug 2026
- 2026Weight-norm Criticality: A Mechanism for Loss Spikes Induced by the Normalization and Weight Decay23 Jul 2026
- 2026Dynamics of Gradient Descent with Large Step Size Near a Manifold of Flat Minima9 Jul 2026
- 2026Towards Understanding the Power and Limits of the Muon Optimizer: A River-Valley Perspective19 Jun 2026
- 2026Taming Curvature: Architecture Warm-Up for Stable Transformer Training15 Jun 2026
- 2026Improving Adversarial Robustness of Attribution via Implicit Regularization28 May 2026
- 2026SGD at the Edge of Stability: The Stochastic Sharpness Gap22 Apr 2026
- 2026Generalization at the Edge of Stability21 Apr 2026
12 newest of 61 All 61 (JSON)