Stability & Generalisation of Gradient Descent for Shallow Neural Networks without the Neural Tangent Kernel
Not on the map. Why?
Abstract (arXiv)
We revisit on-average algorithmic stability of GD for training overparameterised shallow neural networks and prove new generalisation and excess risk bounds without the NTK or PL assumptions. In particular, we show oracle type bounds which reveal that the generalisation and excess risk of GD is controlled by an interpolating network with the shortest GD path from initialisation (in a sense, an interpolating network with the smallest relative norm). While this was known for kernelised interpolants, our proof applies directly to networks trained by GD without intermediate kernelisation. At the same time, by relaxing oracle inequalities developed here we recover existing NTK-based risk bounds in a straightforward way, which demonstrates that our analysis is tighter. Finally, unlike most of the NTK-based analyses we focus on regression with label noise and show that GD with early stopping is consistent.
Key facts. Cited by 26 papers in the map (data through 14 Oct 2026). Builds on 29 of its 41 references that are in the map. First posted on arXiv 27 Jul 2021.
Builds on
29 of 41 references.
References · 29
- 2018Neural Tangent Kernel: Convergence and Generalization in Neural Networkscited by 2,621
- 2018A Convergence Theory for Deep Learning via Over-Parameterizationcited by 787
- 2019Wide Neural Networks of Any Depth Evolve as Linear Models Under Gradient Descentcited by 738
- 2018Gradient Descent Provably Optimizes Over-parameterized Neural Networkscited by 659
- 2019Fine-Grained Analysis of Optimization and Generalization for Overparameterized Two-Layer Neural Networkscited by 566
- 2015Train faster, generalize better: Stability of stochastic gradient descentcited by 552
- 2017Size-Independent Sample Complexity of Neural Networkscited by 290
- 2021Deep learning: a statistical viewpointcited by 240
- 2020Loss landscapes and optimization in over-parameterized non-linear systems and neural networkscited by 233
- 2019Towards moderate overparameterization: global convergence guarantees for training shallow neural networkscited by 176
- 2016Eigenvalues of the Hessian in Deep Learning: Singularity and Beyondcited by 159
- 2019Gradient Descent with Early Stopping is Provably Robust to Label Noise for Overparameterized Neural Networkscited by 159
17 more
- 2018Towards Understanding the Role of Over-Parametrization in Generalization of Neural Networkscited by 142
- 2020On the linearity of large non-linear models: when and why the tangent kernel is constantcited by 129
- 2020Fine-Grained Analysis of Stability and Generalization for Stochastic Gradient Descentcited by 119
- 2017Stability and Generalization of Learning Algorithms that Converge to Global Optimacited by 100
- 2019High probability generalization bounds for uniformly stable algorithms with nearly optimal ratecited by 98
- 2019Sharper bounds for uniformly stable algorithmscited by 93
- 2017Data-Dependent Stability of Stochastic Gradient Descentcited by 92
- 2019Beyond Linearization: On Quadratic and Higher-Order Approximation of Wide Neural Networkscited by 79
- 2018Stability and Convergence Trade-off of Iterative Optimization Algorithmscited by 53
- 2020Regularization Matters: A Nonparametric Perspective on Overparametrized Neural Networkcited by 38
- 2021Early-stopped neural networks are consistentcited by 34
- 2020Analyzing Finite Neural Networks: Can We Trust Neural Tangent Kernel Theory?cited by 33
- 2021Learning with Gradient Descent and Weakly Convex Lossescited by 16
- 2020Benefit of deep learning with non-convex noisy gradient descent: Provable excess risk bound and superiority to kernel methodscited by 9
- 2017Fast Rates for Empirical Risk Minimization of Strict Saddle Problemscited by 7
- 2019An Exponential Efron-Stein Inequality for Lq Stable Learning Rulescited by 6
- 2021Nonparametric Regression with Shallow Overparameterized Neural Networks Trained by GD with Early Stoppingcited by 5
Built on by
Citing papers per year: 2022: 3; 2023: 7; 2024: 4; 2025: 4; 2026: 8.
26 papers.
Sort:
By area:
- 2023On the Optimization and Generalization of Multi-head Attentioncited by 37
- 2023Three-Way Trade-Off in Multi-Objective Learning: Optimization, Generalization and Conflict-Avoidancecited by 36
- 2022Stability and Generalization of Stochastic Optimization with Nonconvex and Nonsmooth Problemscited by 25
- 2022Stability and Generalization Analysis of Gradient Methods for Shallow Neural Networkscited by 21
- 2023Generalization and Stability of Interpolating Neural Networks with Minimal Widthcited by 21
- 2024Stable Minima Cannot Overfit in Univariate ReLU Networks: Generalization by Large Step Sizescited by 12
- 2023Generalization Guarantees of Gradient Descent for Multi-Layer Neural Networkscited by 9
- 2023Norm-based Generalization Bounds for Compositionally Sparse Neural Networkscited by 7
- 2024NTK-Guided Few-Shot Class Incremental Learningcited by 7
- 2024Sharper Guarantees for Learning Neural Network Classifiers with Gradient Methodscited by 7
- 2025Stability and Generalization of Quantum Neural Networkscited by 3
- 2025Optimal Rates for Generalization of Gradient Descent for Deep ReLU Classificationcited by 3
14 more
- 2025On the Theory of Continual Learning with Gradient Descent for Neural Networkscited by 3
- 2022Black-Box Generalization: Stability of Zeroth-Order Learningcited by 1
- 2026Optimization, Generalization and Differential Privacy Bounds for Gradient Descent on Kolmogorov-Arnold Networkscited by 1
- 2026Sufficient Conditions for Stability of Minimum-Norm Interpolating Deep ReLU Networkscited by 1
- 2026Stochastic Gradient Descent with Momentum is Algorithmically Stablecited by 1
- 2023A Linearly Convergent GAN Inversion-based Algorithm for Reverse Engineering of Deceptionscited by 0
- 2023Towards Understanding the Generalizability of Delayed Stochastic Gradient Descentcited by 0
- 2024Bootstrap SGD: Algorithmic Stability and Robustnesscited by 0
- 2025Generalization and Optimization of SGD with Lookaheadcited by 0
- 2026Sequential Subspace Noise Injection Prevents Accuracy Collapse in Certified Unlearningcited by 0
- 2026Towards Initialization-dependent and Non-vacuous Generalization Bounds for Overparameterized Shallow Neural Networkscited by 0
- 2026Learning Theory of the SVRG: Generalization and Convergence Analysiscited by 0
- 2026Minimax-Optimal Generalization Bounds for Smooth Deep Neural Networks Trained by (Stochastic) Gradient Descentcited by 0
- 2026Generalization Analysis of Biased Stochastic Gradient Methods for Minimax Problemscited by 0
- 2026Generalization Analysis of Biased Stochastic Gradient Methods for Minimax Problems14 Oct 2026
- 2026Minimax-Optimal Generalization Bounds for Smooth Deep Neural Networks Trained by (Stochastic) Gradient Descent4 Jun 2026
- 2026Stochastic Gradient Descent with Momentum is Algorithmically Stable27 May 2026
- 2026Learning Theory of the SVRG: Generalization and Convergence Analysis27 May 2026
- 2026Towards Initialization-dependent and Non-vacuous Generalization Bounds for Overparameterized Shallow Neural Networks1 Apr 2026
- 2026Sufficient Conditions for Stability of Minimum-Norm Interpolating Deep ReLU Networks14 Feb 2026
- 2026Optimization, Generalization and Differential Privacy Bounds for Gradient Descent on Kolmogorov-Arnold Networks29 Jan 2026
- 2026Sequential Subspace Noise Injection Prevents Accuracy Collapse in Certified Unlearning8 Jan 2026
- 2025On the Theory of Continual Learning with Gradient Descent for Neural Networks7 Oct 2025
- 2025Optimal Rates for Generalization of Gradient Descent for Deep ReLU Classification3 Oct 2025
- 2025Generalization and Optimization of SGD with Lookahead19 Sep 2025
- 2025Stability and Generalization of Quantum Neural Networks22 Jan 2025
14 more
- 2024Sharper Guarantees for Learning Neural Network Classifiers with Gradient Methods13 Oct 2024
- 2024Bootstrap SGD: Algorithmic Stability and Robustness2 Sep 2024
- 2024Stable Minima Cannot Overfit in Univariate ReLU Networks: Generalization by Large Step Sizes10 Jun 2024
- 2024NTK-Guided Few-Shot Class Incremental Learning19 Mar 2024
- 2023On the Optimization and Generalization of Multi-head Attention19 Oct 2023
- 2023Towards Understanding the Generalizability of Delayed Stochastic Gradient Descent18 Aug 2023
- 2023A Linearly Convergent GAN Inversion-based Algorithm for Reverse Engineering of Deceptions7 Jun 2023
- 2023Three-Way Trade-Off in Multi-Objective Learning: Optimization, Generalization and Conflict-Avoidance31 May 2023
- 2023Generalization Guarantees of Gradient Descent for Multi-Layer Neural Networks26 May 2023
- 2023Generalization and Stability of Interpolating Neural Networks with Minimal Width18 Feb 2023
- 2023Norm-based Generalization Bounds for Compositionally Sparse Neural Networks28 Jan 2023
- 2022Stability and Generalization Analysis of Gradient Methods for Shallow Neural Networks19 Sep 2022
- 2022Stability and Generalization of Stochastic Optimization with Nonconvex and Nonsmooth Problems14 Jun 2022
- 2022Black-Box Generalization: Stability of Zeroth-Order Learning14 Feb 2022
- 2023On the Optimization and Generalization of Multi-head Attentioncited by 37
- 2023Three-Way Trade-Off in Multi-Objective Learning: Optimization, Generalization and Conflict-Avoidancecited by 36
- 2022Stability and Generalization of Stochastic Optimization with Nonconvex and Nonsmooth Problemscited by 25
- 2022Stability and Generalization Analysis of Gradient Methods for Shallow Neural Networkscited by 21
- 2023Generalization and Stability of Interpolating Neural Networks with Minimal Widthcited by 21
- 2024Stable Minima Cannot Overfit in Univariate ReLU Networks: Generalization by Large Step Sizescited by 12
- 2023Generalization Guarantees of Gradient Descent for Multi-Layer Neural Networkscited by 9
- 2023Norm-based Generalization Bounds for Compositionally Sparse Neural Networkscited by 7
- 2024NTK-Guided Few-Shot Class Incremental Learningcited by 7
- 2024Sharper Guarantees for Learning Neural Network Classifiers with Gradient Methodscited by 7
- 2025Stability and Generalization of Quantum Neural Networkscited by 3
- 2025Optimal Rates for Generalization of Gradient Descent for Deep ReLU Classificationcited by 3
12 most cited of 25 All 25 (JSON)
- 2026Generalization Analysis of Biased Stochastic Gradient Methods for Minimax Problems14 Oct 2026
- 2026Minimax-Optimal Generalization Bounds for Smooth Deep Neural Networks Trained by (Stochastic) Gradient Descent4 Jun 2026
- 2026Stochastic Gradient Descent with Momentum is Algorithmically Stable27 May 2026
- 2026Learning Theory of the SVRG: Generalization and Convergence Analysis27 May 2026
- 2026Towards Initialization-dependent and Non-vacuous Generalization Bounds for Overparameterized Shallow Neural Networks1 Apr 2026
- 2026Sufficient Conditions for Stability of Minimum-Norm Interpolating Deep ReLU Networks14 Feb 2026
- 2026Optimization, Generalization and Differential Privacy Bounds for Gradient Descent on Kolmogorov-Arnold Networks29 Jan 2026
- 2025On the Theory of Continual Learning with Gradient Descent for Neural Networks7 Oct 2025
- 2025Optimal Rates for Generalization of Gradient Descent for Deep ReLU Classification3 Oct 2025
- 2025Generalization and Optimization of SGD with Lookahead19 Sep 2025
- 2025Stability and Generalization of Quantum Neural Networks22 Jan 2025
- 2024Sharper Guarantees for Learning Neural Network Classifiers with Gradient Methods13 Oct 2024
12 newest of 25 All 25 (JSON)