Generalization Bounds of Stochastic Gradient Descent for Wide and Deep Neural Networks
Not on the map. Why?
Abstract (arXiv)
We study the training and generalization of deep neural networks (DNNs) in the over-parameterized regime, where the network width (i.e., number of hidden nodes per layer) is much larger than the number of training data points. We show that, the expected 0-1 loss of a wide enough ReLU network trained with stochastic gradient descent (SGD) and random initialization can be bounded by the training loss of a random feature model induced by the network gradient at initialization, which we call a neural tangent random feature (NTRF) model. For data distributions that can be classified by NTRF model with sufficiently small error, our result yields a generalization error bound in the order of tildeO(n-1/2) that is independent of the network width. Our result is more general and sharper than many existing generalization error bounds for over-parameterized neural networks. In addition, we establish a strong connection between our generalization error bound and the neural tangent kernel (NTK) proposed in recent work.
Key facts. Cited by 222 papers in the map (data through 14 Oct 2026). First posted on arXiv 30 May 2019.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 50; 2022: 53; 2023: 49; 2024: 35; 2025: 19; 2026: 16.
222 papers.
Sort:
By area:
- 2021The Values Encoded in Machine Learning Researchcited by 172
- 2022A Kernel-Based View of Language Model Fine-Tuningcited by 121
- 2022Benign Overfitting in Two-layer Convolutional Neural Networkscited by 91
- 2022Understanding the Generalization Benefit of Normalization Layers: Sharpness Reductioncited by 86
- 2023Dichotomy of Early and Late Phase Implicit Biases Can Provably Induce Grokkingcited by 80
- 2023A Theoretical Understanding of Shallow Vision Transformers: Learning, Generalization, and Sample Complexitycited by 77
- 2021Classifying high-dimensional Gaussian mixtures: Where kernel methods fail and neural networks succeedcited by 69
- 2021Training Multi-Layer Over-Parametrized Neural Network in Subquadratic Timecited by 62
- 2021Exploring Memorization in Adversarial Trainingcited by 55
- 2022Towards Understanding Mixture of Experts in Deep Learningcited by 52
- 2021Understanding the Generalization of Adam in Learning Neural Networks with Proper Regularizationcited by 49
- 2021A Local Convergence Theory for Mildly Over-Parameterized Two-Layer Neural Networkcited by 47
38 more
- 2022Deep Nonparametric Estimation of Operators between Infinite Dimensional Spacescited by 45
- 2021EE-Net: Exploitation-Exploration Neural Networks in Contextual Banditscited by 44
- 2021Towards General Function Approximation in Zero-Sum Markov Gamescited by 43
- 2022Graph Neural Networks are Inherently Good Generalizers: Insights by Bridging GNNs and MLPscited by 42
- 2021Neural Contextual Bandits without Regretcited by 40
- 2023On the Optimization and Generalization of Multi-head Attentioncited by 37
- 2024How Do Nonlinear Transformers Learn and Generalize in In-Context Learning?cited by 35
- 2021Early-stopped neural networks are consistentcited by 34
- 2021Scaling Neural Tangent Kernels via Sketching and Random Featurescited by 32
- 2021Does Preprocessing Help Training Over-parameterized Neural Networks?cited by 32
- 2021Deep Active Learning by Leveraging Training Dynamicscited by 32
- 2022Understanding Gradual Domain Adaptation: Improved Analysis, Optimal Path and Beyondcited by 32
- 2024PINNACLE: PINN Adaptive ColLocation and Experimental points selectioncited by 32
- 2024Implicit Bias and Fast Convergence Rates for Self-attentioncited by 31
- 2022Bypass Exponential Time Preprocessing: Fast Neural Network Training via Weight-Data Correlation Preprocessingcited by 29
- 2022Improving Multi-Task Generalization via Regularizing Spurious Correlationcited by 28
- 2021Properties of the After Kernelcited by 27
- 2021Rethinking Influence Functions of Neural Networks in the Over-parameterized Regimecited by 26
- 2023Mind the spikes: Benign overfitting of kernels and neural networks in fixed dimensioncited by 26
- 2023How to Protect Copyright Data in Optimization of Large Language Models?cited by 26
- 2021Measuring Generalization with Optimal Transportcited by 25
- 2021Offline Neural Contextual Bandits: Pessimism, Optimization and Generalizationcited by 25
- 2022Sample-Then-Optimize Batch Neural Thompson Samplingcited by 25
- 2023Over-Parameterization Exponentially Slows Down Gradient Descent for Learning a Single Neuroncited by 25
- 2022Convergence of gradient descent for deep neural networkscited by 24
- 2021Neural Active Learning with Performance Guaranteescited by 22
- 2022Global Convergence of MAML and Theory-Inspired Neural Architecture Search for Few-Shot Learningcited by 22
- 2021The Limitations of Large Width in Neural Networks: A Deep Gaussian Process Perspectivecited by 21
- 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
- 2024Operator Learning Using Random Features: A Tool for Scientific Computingcited by 21
- 2021Multi-facet Contextual Bandits: A Neural Network Perspectivecited by 20
- 2023Graph Neural Banditscited by 20
- 2023Six Lectures on Linearized Neural Networkscited by 20
- 2024A Survey on Statistical Theory of Deep Learning: Approximation, Training Dynamics, and Generative Modelscited by 20
- 2021Towards Understanding Learning in Neural Networks with Linear Teacherscited by 19
- 2023Provable Guarantees for Nonlinear Feature Learning in Three-Layer Neural Networkscited by 19
- 2022Fast Neural Kernel Embeddings for General Activationscited by 18
50 most cited of 222 All 222 (JSON)
- 2026Training-Loss Guarantees for Muon with Finite-Step Newton--Schulz Orthogonalization2 Oct 2026
- 2026Generalist-Specialist Mixture-of-Experts for Rare Pathology Detection in Multimodal Imaging16 Sep 2026
- 2026Generalization in Nonlinear Least Squares via Learned Feature Geometry7 Jun 2026
- 2026Optimal Rates for Generalization of Gradient Descent Methods with Deep Neural Networks4 Jun 2026
- 2026The Neural Tangent Kernel for Classification17 May 2026
- 2026A Layer Separation Optimization Framework for Cross-Entropy Training in Deep Learning25 Apr 2026
- 2026Generalization error bounds for two-layer neural networks with Lipschitz loss function7 Apr 2026
- 2026Gradient Descent with Projection Finds Over-Parameterized Neural Networks for Learning Low-Degree Polynomials with Nearly Minimax Optimal Rate22 Mar 2026
- 2026A Gaussian Comparison Theorem for Training Dynamics in Machine Learning10 Mar 2026
- 2026On the Equivalence of Random Network Distillation, Deep Ensembles, and Bayesian Inference23 Feb 2026
- 2026Data-Driven Deep MIMO Detection:Network Architectures and Generalization Analysis13 Feb 2026
- 2026Pruning at Initialisation through the lens of Graphon Limit: Convergence, Expressivity, and Generalisation6 Feb 2026
38 more
- 2026Optimization, Generalization and Differential Privacy Bounds for Gradient Descent on Kolmogorov-Arnold Networks29 Jan 2026
- 2026Quantum-Enhanced Neural Contextual Bandit Algorithms6 Jan 2026
- 2026Leveraging Flatness to Improve Information-Theoretic Generalization Bounds for SGD4 Jan 2026
- 2026Deep Networks Learn Deep Hierarchical Models1 Jan 2026
- 2025Shallow Neural Networks Learn Low-Degree Spherical Polynomials with Feature Learning by Learnable Channel Attention23 Dec 2025
- 2025Data Value in the Age of Scaling: Understanding LLM Scaling Dynamics Under Real-Synthetic Data Mixtures17 Nov 2025
- 2025Mining Useful General Data for Low-Resource Domain Adaptation10 Nov 2025
- 2025On the Generalization Properties of Learning the Random Feature Models with Learnable Activation Functions17 Oct 2025
- 2025Optimal Rates for Generalization of Gradient Descent for Deep ReLU Classification3 Oct 2025
- 2025How Can Mamba Learn In Context with Outliers and Generalize Provably?1 Oct 2025
- 2025Dextr: Zero-Shot Neural Architecture Search with Singular Value Decomposition and Extrinsic Curvature18 Aug 2025
- 2025Compact Vision Transformer by Reduction of Kernel Complexity17 Jul 2025
- 2025Constructive Universal Approximation and Sure Convergence for Multi-Layer Neural Networks7 Jul 2025
- 2025Generalization Bound of Gradient Flow through Training Trajectory and Data-dependent Kernel12 Jun 2025
- 2025Models of Heavy-Tailed Mechanistic Universality4 Jun 2025
- 2025Neural Variance-aware Dueling Bandits with Deep Representation and Shallow Exploration2 Jun 2025
- 2025Overfitting has a limitation: a model-independent generalization gap bound based on R'enyi entropy30 May 2025
- 2025Unsupervised anomaly detection in MeV ultrafast electron diffraction19 May 2025
- 2025NeuroSep-CP-LCB: A Deep Learning-based Contextual Multi-armed Bandit Algorithm with Uncertainty Quantification for Early Sepsis Prediction20 Mar 2025
- 2025Global Convergence and Rich Feature Learning in L-Layer Infinite-Width Neural Networks under μP Parametrization12 Mar 2025
- 2025Generalizability of Neural Networks Minimizing Empirical Risk Based on Expressive Ability6 Mar 2025
- 2025DiffoRA: Enabling Parameter-Efficient Fine-Tuning via Differential Module Selection13 Feb 2025
- 2025Linearly Separable Features in Shallow Nonlinear Networks: Width Scales Polynomially with Intrinsic Data Dimension4 Jan 2025
- 2024Learning Expressive Random Feature Models via Parametrized Activations29 Nov 2024
- 2024Variance-Aware Linear UCB with Deep Representation for Neural Contextual Bandits8 Nov 2024
- 2024Gradient Descent Finds Over-Parameterized Neural Networks with Sharp Generalization for Nonparametric Regression5 Nov 2024
- 2024PageRank Bandits for Link Prediction3 Nov 2024
- 2024Understanding Transfer Learning via Mean-field Analysis22 Oct 2024
- 2024Neural Combinatorial Clustered Bandits for Recommendation Systems18 Oct 2024
- 2024Sharper Guarantees for Learning Neural Network Classifiers with Gradient Methods13 Oct 2024
- 2024On the Convergence Analysis of Over-Parameterized Variational Autoencoders: A Neural Tangent Kernel Perspective9 Sep 2024
- 2024A practical generalization metric for deep networks benchmarking2 Sep 2024
- 2024Operator Learning Using Random Features: A Tool for Scientific Computing12 Aug 2024
- 2024Meta Clustering of Neural Bandits10 Aug 2024
- 2024Differential Privacy Mechanisms in Neural Tangent Kernel Regression18 Jul 2024
- 2024Convergence of Implicit Gradient Descent for Training Two-Layer Physics-Informed Neural Networks3 Jul 2024
- 2024Stable Minima Cannot Overfit in Univariate ReLU Networks: Generalization by Large Step Sizes10 Jun 2024
- 2024What Improves the Generalization of Graph Transformers? A Theoretical Dive into the Self-attention and Positional Encoding4 Jun 2024
50 newest of 222 All 222 (JSON)
- 2021The Values Encoded in Machine Learning Researchcited by 172
- 2022A Kernel-Based View of Language Model Fine-Tuningcited by 121
- 2022Benign Overfitting in Two-layer Convolutional Neural Networkscited by 91
- 2022Understanding the Generalization Benefit of Normalization Layers: Sharpness Reductioncited by 86
- 2023Dichotomy of Early and Late Phase Implicit Biases Can Provably Induce Grokkingcited by 80
- 2023A Theoretical Understanding of Shallow Vision Transformers: Learning, Generalization, and Sample Complexitycited by 77
- 2021Classifying high-dimensional Gaussian mixtures: Where kernel methods fail and neural networks succeedcited by 69
- 2021Training Multi-Layer Over-Parametrized Neural Network in Subquadratic Timecited by 62
- 2021Exploring Memorization in Adversarial Trainingcited by 55
- 2021Understanding the Generalization of Adam in Learning Neural Networks with Proper Regularizationcited by 49
- 2021A Local Convergence Theory for Mildly Over-Parameterized Two-Layer Neural Networkcited by 47
- 2022Deep Nonparametric Estimation of Operators between Infinite Dimensional Spacescited by 45
12 most cited of 212 All 212 (JSON)
- 2026Training-Loss Guarantees for Muon with Finite-Step Newton--Schulz Orthogonalization2 Oct 2026
- 2026Generalization in Nonlinear Least Squares via Learned Feature Geometry7 Jun 2026
- 2026Optimal Rates for Generalization of Gradient Descent Methods with Deep Neural Networks4 Jun 2026
- 2026The Neural Tangent Kernel for Classification17 May 2026
- 2026A Layer Separation Optimization Framework for Cross-Entropy Training in Deep Learning25 Apr 2026
- 2026Generalization error bounds for two-layer neural networks with Lipschitz loss function7 Apr 2026
- 2026Gradient Descent with Projection Finds Over-Parameterized Neural Networks for Learning Low-Degree Polynomials with Nearly Minimax Optimal Rate22 Mar 2026
- 2026A Gaussian Comparison Theorem for Training Dynamics in Machine Learning10 Mar 2026
- 2026On the Equivalence of Random Network Distillation, Deep Ensembles, and Bayesian Inference23 Feb 2026
- 2026Data-Driven Deep MIMO Detection:Network Architectures and Generalization Analysis13 Feb 2026
- 2026Pruning at Initialisation through the lens of Graphon Limit: Convergence, Expressivity, and Generalisation6 Feb 2026
- 2026Optimization, Generalization and Differential Privacy Bounds for Gradient Descent on Kolmogorov-Arnold Networks29 Jan 2026
12 newest of 212 All 212 (JSON)