An Investigation into Neural Net Optimization via Hessian Eigenvalue Density
Not on the map. Why?
Abstract (arXiv)
To understand the dynamics of optimization in deep neural networks, we develop a tool to study the evolution of the entire Hessian spectrum throughout the optimization process. Using this, we study a number of hypotheses concerning smoothness, curvature, and sharpness in the deep learning literature. We then thoroughly analyze a crucial structural feature of the spectra: in non-batch normalized networks, we observe the rapid appearance of large isolated eigenvalues in the spectrum, along with a surprising concentration of the gradient in the corresponding eigenspaces. In batch normalized networks, these two effects are almost absent. We characterize these effects, and explain how they affect optimization speed through both theory and experiments. As part of this work, we adapt advanced tools from numerical linear algebra that allow scalable and accurate estimation of the entire Hessian spectrum of ImageNet-scale neural networks; this technique may be of independent interest in other applications.
Key facts. Cited by 289 papers in the map (data through 7 Oct 2026). First posted on arXiv 29 Jan 2019.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 28; 2022: 31; 2023: 40; 2024: 44; 2025: 48; 2026: 98.
289 papers.
Sort:
By area:
- 2021Gradient Descent on Neural Networks Typically Occurs at the Edge of Stabilitycited by 413
- 2023Fine-Tuning Language Models with Just Forward Passescited by 337
- 2022How Do Vision Transformers Work?cited by 305
- 2023Sophia: A Scalable Stochastic Second-order Optimizer for Language Model Pre-trainingcited by 260
- 2021Fishr: Invariant Gradient Variances for Out-of-Distribution Generalizationcited by 186
- 2023Understanding plasticity in neural networkscited by 170
- 2023Stabilizing Transformer Training by Preventing Attention Entropy Collapsecited by 154
- 2021Deep Learning Through the Lens of Example Difficultycited by 139
- 2024Adam-mini: Use Fewer Learning Rates To Gain Morecited by 134
- 2021Scaling Up Influence Functionscited by 130
- 2024Why Transformers Need Adam: A Hessian Perspectivecited by 123
- 2024Challenges in Training PINNs: A Loss Landscape Perspectivecited by 109
38 more
- 2021Reasonable Effectiveness of Random Weighting: A Litmus Test for Multi-Task Learningcited by 100
- 2022Neural Collapse: A Review on Modelling Principles and Generalizationcited by 95
- 2022Understanding the Generalization Benefit of Normalization Layers: Sharpness Reductioncited by 86
- 2022A Closer Look at Smoothness in Domain Adversarial Trainingcited by 85
- 2022When Do Flat Minima Optimizers Work?cited by 78
- 2022Make Sharpness-Aware Minimization Stronger: A Sparsified Perturbation Approachcited by 70
- 2021Analytic Insights into Structure and Rank of Neural Network Hessian Mapscited by 52
- 2024Friendly Sharpness-Aware Minimizationcited by 52
- 2024Second-Order Fine-Tuning without Pain for LLMs:A Hessian Informed Zeroth-Order Optimizercited by 51
- 2022When Does Differentially Private Learning Not Suffer in High Dimensions?cited by 48
- 2021A study on the plasticity of neural networkscited by 44
- 2022Second-order regression models exhibit progressive sharpening to the edge of stabilitycited by 40
- 2021A Loss Curvature Perspective on Training Instability in Deep Learningcited by 39
- 2022Low-Pass Filtering SGD for Recovering Flat Optima in the Deep Learning Optimization Landscapecited by 39
- 2023Enhancing Sharpness-Aware Optimization Through Variance Suppressioncited by 37
- 2024Does SGD really happen in tiny subspaces?cited by 35
- 2022Robust Fine-Tuning of Deep Neural Networks with Hessian-based Generalization Guaranteescited by 34
- 2021SGD in the Large: Average-case Analysis, Asymptotics, and Stepsize Criticalitycited by 33
- 2021Hessian Eigenspectra of More Realistic Nonlinear Modelscited by 33
- 2024Unveiling the optimization process of Physics Informed Neural Networks: How accurate and competitive can PINNs be?cited by 31
- 2021How many degrees of freedom do we need to train deep networks: a loss landscape perspectivecited by 28
- 2022Escaping Saddle Points for Effective Generalization on Class-Imbalanced Datacited by 28
- 2023Theoretical and Practical Perspectives on what Influence Functions Docited by 28
- 2023An operator preconditioning perspective on training in physics-informed machine learningcited by 27
- 2022Stochastic Weight Averaging Revisitedcited by 26
- 2023DPZero: Private Fine-Tuning of Language Models without Backpropagationcited by 26
- 2023Directions of Curvature as an Explanation for Loss of Plasticitycited by 26
- 2023Symmetry Induces Structure and Constraint of Learningcited by 25
- 2022An Empirical Study of Implicit Regularization in Deep Offline RLcited by 24
- 2023SAM operates far from home: eigenvalue regularization as a dynamical phenomenoncited by 24
- 2023Memorization Through the Lens of Curvature of Loss Function Around Samplescited by 24
- 2023Improving Pixel-based MIM by Reducing Wasted Modeling Capabilitycited by 24
- 2023ZO-AdaMU Optimizer: Adapting Perturbation by the Momentum and Uncertainty in Zeroth-order Optimizationcited by 24
- 2023Improving Generalization of Adversarial Training via Robust Critical Fine-Tuningcited by 23
- 2024Super Consistency of Neural Network Landscapes and Learning Rate Transfercited by 23
- 2025Stable Gradients for Stable Learning at Scale in Deep Reinforcement Learningcited by 22
- 2021On the Periodic Behavior of Neural Network Training with Batch Normalization and Weight Decaycited by 20
- 2023Training trajectories, mini-batch losses and the curious role of the learning ratecited by 20
50 most cited of 289 All 289 (JSON)
- 2026Probe-Space Preconditioning for Fast and Stable Zero-Order Training29 Sep 2026
- 2026Background Gradients Shape Memorization in Flow Matching27 Sep 2026
- 2026Balancing Early Performance Sacrifices with Long-Term Gains: Scaling Learning-Rate Warmup Duration Across Training Horizons27 Sep 2026
- 2026The Multiple Timescales of Gradient Descent on the Edge of Stability: A Perturbative Derivation of the Central Flow1 Sep 2026
- 2026Curvature-Conditioned Multiscale Momentum with Sphere Constraints for LLM Pretraining28 Aug 2026
- 2026Across the Loss Landscape with Progressive Growth25 Aug 2026
- 2026A Data-dependent Early Stopping Rule using Rademacher Complexity with L1-norm25 Aug 2026
- 2026Loss Landscape Features That Make Adam Stall: Definitions, Estimators, and the Preconditioned Hessian View23 Aug 2026
- 2026Blockwise Stabilized Adaptive Cubic Regularization with Subsolvers via Recurrence22 Aug 2026
- 2026Loss-Parameterized Fisher Width Along Learning Trajectories21 Aug 2026
- 2026Activation-Keyed Momentum: An Anisotropic Momentum Update via the Delta Rule19 Aug 2026
- 2026Layers Matter: Why Continual Learning Regularization Should Be Layer-Adaptive16 Aug 2026
38 more
- 2026MALT: Lightweight Curvature-Aware Muon via Diagonal Preconditioning5 Aug 2026
- 2026The Grokked Illusion: True Equilibrium Mitigates Catastrophic Forgetting31 Jul 2026
- 2026SPRKD: Effective Knowledge Distillation for Deep Neural Networks via Saddle Region Approximation25 Jul 2026
- 2026Mini-batch Noise Lowers Sharpness via Dominant-Subspace Fluctuations25 Jul 2026
- 2026A Defense of the Quadratic Model23 Jul 2026
- 2026A Geometric Perspective on Stabilizing Value Conflict Resolution20 Jul 2026
- 2026First-Order Predictable but Pairwise Fragile: Local Task Adaptation in Trained Transformers18 Jul 2026
- 2026Explaining Near-Zero Hessian Eigenvalues Through Approximate Symmetries in Neural Networks8 Jul 2026
- 2026Leveraging Extragradient for Effective Sharpness-Aware Minimization in Deep Learning7 Jul 2026
- 2026Beyond Activation Alignment:The Alignment-Diversity Tradeoff in Task-Aware LLM Quantization1 Jul 2026
- 2026Curvature-Weighted Gradient Diversity: A Noise Measure for Geometry-Adaptive SGD Schedules29 Jun 2026
- 2026Scalar Representations of Neural Network Training Dynamics29 Jun 2026
- 2026Characterizing Optimizer-Dependent Training Dynamics Through Hessian Eigenvector Displacement and Localization29 Jun 2026
- 2026Not All Objectives Are Born Equal: Priority-Constrained Descent for Hierarchical Multi-Objective Optimization28 Jun 2026
- 2026Closed-Form Steepest Descent Direction toward Flat Minima: Reducing Upper Bounds on the Loss Hessian Eigenspectrum in Neural Networks27 Jun 2026
- 2026Evaluation-Strategy Gap in Fault Diagnosis of Deep Learning Programs25 Jun 2026
- 2026Dynamic estimation of slowly varying sequences22 Jun 2026
- 2026Exposing the Illusion of Erasure in Knowledge Editing for LLMs22 Jun 2026
- 2026Dead-Direction Signatures: A Cheap Spectral Reading of Singular Complexity19 Jun 2026
- 2026Algebraic Dead Directions in LayerNorm Transformers: A Forward-Pass-Only Diagnostic at LLM Scale17 Jun 2026
- 2026The Geometry of Saturation: Effective Rank Predicts When Labels Stop Helping in Few-Shot Classification12 Jun 2026
- 2026Accelerating SAV-based optimization via randomized low-rank Hessian approximation9 Jun 2026
- 2026Second-Order Path Kernel Interpolation Formulas in Machine Learning5 Jun 2026
- 2026The Fine-Tuning Trap: Evaluating Negative Transfer and the Role of PEFT in Sub-1B Mathematical Reasoning5 Jun 2026
- 2026Curvature-aware dynamic precision approach for physics-informed neural networks3 Jun 2026
- 2026Beyond ℓ2-norm and ℓ_∞-norm: A Curvature-Inspired ℓp-Norm Scheme for Deep Neural Networks1 Jun 2026
- 2026A2SG:Adaptive and Asymmetric Surrogate Gradients for Training Deep Spiking Neural Networks29 May 2026
- 2026Worker Disagreement Reveals Sharp Directions in Local SGD26 May 2026
- 2026How the Optimizer Shapes Learned Solutions in Equivariant Neural Networks26 May 2026
- 2026Stochastic Estimation of the Layer-wise Hessian Trace for Monitoring Neural-network Training25 May 2026
- 2026Non-normal spectral signatures of instability in neural network training dynamics22 May 2026
- 2026Anytime Training with Schedule-Free Spectral Optimization21 May 2026
- 2026AMUSE: Anytime Muon with Stable Gradient Evaluation21 May 2026
- 2026SMA-DP: Spectral Memory-Aware Differential Privacy for Deep Learning19 May 2026
- 2026LLM Pretraining Shapes a Generalizable Manifold: Insights into Cross-Modal Transfer to Time Series19 May 2026
- 2026Position: Weight Space Should Be a First-Class Generative AI Modality18 May 2026
- 2026WinQ: Accelerating Quantization-Aware Training of Language Models Around Saddle Points17 May 2026
- 2026Radial-Angular Geometry for Reliable Update Diagnosis in Noisy-Label Learning17 May 2026
50 newest of 289 All 289 (JSON)
- 2026Revisiting Robustness for LLM Safety Alignment via Selective Geometry Controlcited by 3
- 2026CrispEdit: Low-Curvature Projections for Scalable Non-Destructive LLM Editingcited by 3
- 2025Leveraging Machine Unlearning for Cost-Efficient Preference Alignmentcited by 2
- 2025Data Curation Through the Lens of Spectral Dynamics: Static Limits, Dynamic Acceleration, and Practical Oraclescited by 0
- 2026A Geometric Perspective on Stabilizing Value Conflict Resolutioncited by 0
- 2026A Geometric Perspective on Stabilizing Value Conflict Resolution20 Jul 2026
- 2026CrispEdit: Low-Curvature Projections for Scalable Non-Destructive LLM Editing17 Feb 2026
- 2026Revisiting Robustness for LLM Safety Alignment via Selective Geometry Control7 Feb 2026
- 2025Data Curation Through the Lens of Spectral Dynamics: Static Limits, Dynamic Acceleration, and Practical Oracles2 Dec 2025
- 2025Leveraging Machine Unlearning for Cost-Efficient Preference Alignment9 Apr 2025
- 2024On Newton's Method to Unlearn Neural Networkscited by 5
- 2024Dataset Condensation Driven Machine Unlearningcited by 1
- 2025WARP: Weight Teleportation for Attack-Resilient Unlearning Protocolscited by 1
- 2025A mean teacher algorithm for unlearning of language modelscited by 0
- 2026Why Some Models Resist Unlearning: A Linear Stability Perspectivecited by 0
- 2026Why Some Models Resist Unlearning: A Linear Stability Perspective3 Feb 2026
- 2025WARP: Weight Teleportation for Attack-Resilient Unlearning Protocols29 Nov 2025
- 2025A mean teacher algorithm for unlearning of language models18 Apr 2025
- 2024On Newton's Method to Unlearn Neural Networks20 Jun 2024
- 2024Dataset Condensation Driven Machine Unlearning31 Jan 2024
- 2025How Instruction and Reasoning Data shape Post-Training: Data Quality through the Lens of Layer-wise Gradientscited by 8
- 2025From Memorization to Reasoning in the Spectrum of Loss Curvaturecited by 1
- 2026The Fine-Tuning Trap: Evaluating Negative Transfer and the Role of PEFT in Sub-1B Mathematical Reasoningcited by 0
- 2026The Fine-Tuning Trap: Evaluating Negative Transfer and the Role of PEFT in Sub-1B Mathematical Reasoning5 Jun 2026
- 2025From Memorization to Reasoning in the Spectrum of Loss Curvature28 Oct 2025
- 2025How Instruction and Reasoning Data shape Post-Training: Data Quality through the Lens of Layer-wise Gradients14 Apr 2025
- 2021Gradient Descent on Neural Networks Typically Occurs at the Edge of Stabilitycited by 413
- 2023Fine-Tuning Language Models with Just Forward Passescited by 337
- 2022How Do Vision Transformers Work?cited by 305
- 2023Sophia: A Scalable Stochastic Second-order Optimizer for Language Model Pre-trainingcited by 260
- 2021Fishr: Invariant Gradient Variances for Out-of-Distribution Generalizationcited by 186
- 2023Understanding plasticity in neural networkscited by 170
- 2023Stabilizing Transformer Training by Preventing Attention Entropy Collapsecited by 154
- 2021Deep Learning Through the Lens of Example Difficultycited by 139
- 2024Adam-mini: Use Fewer Learning Rates To Gain Morecited by 134
- 2021Scaling Up Influence Functionscited by 130
- 2024Why Transformers Need Adam: A Hessian Perspectivecited by 123
- 2024Challenges in Training PINNs: A Loss Landscape Perspectivecited by 109
12 most cited of 271 All 271 (JSON)
- 2026Probe-Space Preconditioning for Fast and Stable Zero-Order Training29 Sep 2026
- 2026Balancing Early Performance Sacrifices with Long-Term Gains: Scaling Learning-Rate Warmup Duration Across Training Horizons27 Sep 2026
- 2026The Multiple Timescales of Gradient Descent on the Edge of Stability: A Perturbative Derivation of the Central Flow1 Sep 2026
- 2026Curvature-Conditioned Multiscale Momentum with Sphere Constraints for LLM Pretraining28 Aug 2026
- 2026Across the Loss Landscape with Progressive Growth25 Aug 2026
- 2026A Data-dependent Early Stopping Rule using Rademacher Complexity with L1-norm25 Aug 2026
- 2026Loss Landscape Features That Make Adam Stall: Definitions, Estimators, and the Preconditioned Hessian View23 Aug 2026
- 2026Blockwise Stabilized Adaptive Cubic Regularization with Subsolvers via Recurrence22 Aug 2026
- 2026Loss-Parameterized Fisher Width Along Learning Trajectories21 Aug 2026
- 2026Activation-Keyed Momentum: An Anisotropic Momentum Update via the Delta Rule19 Aug 2026
- 2026Layers Matter: Why Continual Learning Regularization Should Be Layer-Adaptive16 Aug 2026
- 2026MALT: Lightweight Curvature-Aware Muon via Diagonal Preconditioning5 Aug 2026
12 newest of 271 All 271 (JSON)