Learning and Generalization in Overparameterized Neural Networks, Going Beyond Two Layers
Not on the map. Why?
Abstract (arXiv)
The fundamental learning theory behind neural networks remains largely open. What classes of functions can neural networks actually learn? Why doesn't the trained network overfit when it is overparameterized? In this work, we prove that overparameterized neural networks can learn some notable concept classes, including two and three-layer networks with fewer parameters and smooth activations. Moreover, the learning can be simply done by SGD (stochastic gradient descent) or its variants in polynomial time using polynomially many samples. The sample complexity can also be almost independent of the number of parameters in the network. On the technique side, our analysis goes beyond the so-called NTK (neural tangent kernel) linearization of neural networks in prior works. We establish a new notion of quadratic approximation of the neural network (that can be viewed as a second-order variant of NTK), and connect it to the SGD theory of escaping saddle points.
Key facts. Cited by 340 papers in the map (data through 14 Oct 2026). First posted on arXiv 12 Nov 2018.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 92; 2022: 91; 2023: 53; 2024: 51; 2025: 36; 2026: 17.
340 papers.
Sort:
By area:
- 2022Not All Patches are What You Need: Expediting Vision Transformers via Token Reorganizationscited by 363
- 2021Training Neural Networks with Fixed Sparse Maskscited by 214
- 2021The Values Encoded in Machine Learning Researchcited by 172
- 2022Neural Networks can Learn Representations with Gradient Descentcited by 161
- 2022High-dimensional Asymptotics of Feature Learning: How One Gradient Step Improves the Representationcited by 150
- 2022A Kernel-Based View of Language Model Fine-Tuningcited by 121
- 2021Toward Understanding the Feature Learning Process of Self-supervised Contrastive Learningcited by 119
- 2021Generalization error of random features and kernel methods: hypercontractivity and kernel matrix concentrationcited by 110
- 2021What Happens after SGD Reaches Zero Loss? --A Mathematical Frameworkcited by 107
- 2023Scan and Snap: Understanding Training Dynamics and Token Composition in 1-layer Transformercited by 102
- 2022The Design Space of E(3)-Equivariant Atom-Centered Interatomic Potentialscited by 101
- 2021Theory of overparametrization in quantum neural networkscited by 100
38 more
- 2022Neural Collapse: A Review on Modelling Principles and Generalizationcited by 95
- 2022Benign Overfitting in Two-layer Convolutional Neural Networkscited by 91
- 2021Learning with invariances in random features and kernel modelscited by 89
- 2022Understanding the Generalization Benefit of Normalization Layers: Sharpness Reductioncited by 86
- 2021Gradient Descent on Two-layer Nets: Margin Maximization and Simplicity Biascited by 83
- 2023A Theoretical Understanding of Shallow Vision Transformers: Learning, Generalization, and Sample Complexitycited by 77
- 2021Pixelated Butterfly: Simple and Efficient Sparse training for Neural Network Modelscited by 76
- 2022Gradient flow dynamics of shallow ReLU networks for square loss and orthogonal inputscited by 76
- 2021On the Validity of Modeling SGD with Stochastic Differential Equations (SDEs)cited by 73
- 2023FedExP: Speeding Up Federated Averaging via Extrapolationcited by 69
- 2021Model Complexity of Deep Learning: A Surveycited by 65
- 2024Can Looped Transformers Learn to Implement Multi-step Gradient Descent for In-context Learning?cited by 65
- 2021Wide Neural Networks Forget Less Catastrophicallycited by 64
- 2021Lost in Pruning: The Effects of Pruning Neural Networks beyond Test Accuracycited by 62
- 2022Towards Understanding Mixture of Experts in Deep Learningcited by 52
- 2022Theoretical Guarantees for Permutation-Equivariant Quantum Neural Networkscited by 52
- 2022Predicting Out-of-Distribution Error with the Projection Normcited by 50
- 2021Understanding the Generalization of Adam in Learning Neural Networks with Proper Regularizationcited by 49
- 2024From Lazy to Rich: Exact Learning Dynamics in Deep Linear Networkscited by 49
- 2022Beyond Explaining: Opportunities and Challenges of XAI-Based Model Improvementcited by 48
- 2021Heavy Tails in SGD and Compressibility of Overparametrized Neural Networkscited by 44
- 2022Graph Neural Networks are Inherently Good Generalizers: Insights by Bridging GNNs and MLPscited by 42
- 2024Neural Network-Based Score Estimation in Diffusion Models: Optimization and Generalizationcited by 42
- 2024Get rich quick: exact solutions reveal how unbalanced initializations promote rapid feature learningcited by 42
- 2021Quantifying the Benefit of Using Differentiable Learning over Tangent Kernelscited by 41
- 2021Recent Advances on Neural Network Pruning at Initializationcited by 41
- 2023What can a Single Attention Layer Learn? A Study Through the Random Features Lenscited by 40
- 2022Implicit Bias of Gradient Descent on Reparametrized Models: On Equivalence to Mirror Descentcited by 39
- 2023A Survey on Efficient Training of Transformerscited by 39
- 2022The Mechanism of Prediction Head in Non-contrastive Self-supervised Learningcited by 38
- 2022Pareto Invariant Risk Minimization: Towards Mitigating the Optimization Dilemma in Out-of-Distribution Generalizationcited by 38
- 2022On the Power of Foundation Modelscited by 37
- 2023Scaling MLPs: A Tale of Inductive Biascited by 37
- 2023On the Stepwise Nature of Self-Supervised Learningcited by 36
- 2021On the Theory of Implicit Deep Learning: Global Convergence with Implicit Layerscited by 35
- 2022Understanding Deep Contrastive Learning via Coordinate-wise Optimizationcited by 35
- 2022Limitations of the NTK for Understanding Generalization in Deep Learningcited by 35
- 2023Patch-level Routing in Mixture-of-Experts is Provably Sample-efficient for Convolutional Neural Networkscited by 35
50 most cited of 340 All 340 (JSON)
- 2026Relative Generalization Invariance of LLM Pretraining26 Sep 2026
- 2026A Data-dependent Early Stopping Rule using Rademacher Complexity with L1-norm25 Aug 2026
- 2026FlexPooling with Simple Auxiliary Classifiers in Deep Networks12 Jun 2026
- 2026Activation Functions, Statistics and Learning of Higher-Order Interactions in Restricted Boltzmann Machines18 May 2026
- 2026Strategic Over-Parameterization for Generalizable Low-Rank Adaptation15 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
- 2026Towards Initialization-dependent and Non-vacuous Generalization Bounds for Overparameterized Shallow Neural Networks1 Apr 2026
- 2026Curvature-Guided LoRA: Matching Full Fine-Tuning in Function Space31 Mar 2026
- 2026Beyond Consistency: Inference for the Relative risk functional in Deep Nonparametric Cox Models25 Mar 2026
- 2026An Optimal Control Approach To Transformer Training10 Mar 2026
- 2026Convex Dominance in Deep Learning I: A Scaling Law of Loss and Learning Rate6 Feb 2026
38 more
- 2026Provable Learning of Random Hierarchy Models and Hierarchical Shallow-to-Deep Chaining27 Jan 2026
- 2026TEA: Temporal Adaptive Satellite Image Semantic Segmentation8 Jan 2026
- 2026On the Convergence Behavior of Preconditioned Gradient Descent Toward the Rich Learning Regime6 Jan 2026
- 2026Leveraging Flatness to Improve Information-Theoretic Generalization Bounds for SGD4 Jan 2026
- 2026Deep Networks Learn Deep Hierarchical Models1 Jan 2026
- 2025The Urysohn Ladder: Recursive Metric Contraction for Scalable Continual Learning20 Dec 2025
- 2025A Special Case of Quadratic Extrapolation Under the Neural Tangent Kernel11 Dec 2025
- 2025Efficient Continual Learning in Neural Machine Translation: A Low-Rank Adaptation Approach10 Dec 2025
- 2025Provably Learning from Modern Language Models via Low Logit Rank10 Dec 2025
- 2025Rethinking Decoupled Knowledge Distillation: A Predictive Distribution Perspective4 Dec 2025
- 2025Mining Useful General Data for Low-Resource Domain Adaptation10 Nov 2025
- 2025Block Coordinate Descent for Neural Networks Provably Finds Global Minima26 Oct 2025
- 2025The Graphon Limit Hypothesis: Understanding Neural Network Pruning via Infinite Width Analysis20 Oct 2025
- 2025Vanishing Contributions: A Unified Framework for Smooth and Iterative Model Compression9 Oct 2025
- 2025Accelerating Inference for Multilayer Neural Networks with Quantum Computers8 Oct 2025
- 2025Angular Constraint Embedding via SpherePair Loss for Constrained Clustering8 Oct 2025
- 2025How Well Can Preference Optimization Generalize Under Noisy Feedback?1 Oct 2025
- 2025How Can Mamba Learn In Context with Outliers and Generalize Provably?1 Oct 2025
- 2025Gradient Flow Convergence Guarantee for General Neural Network Architectures28 Sep 2025
- 2025Light Differentiable Logic Gate Networks26 Sep 2025
- 2025Geometry of Neural Reinforcement Learning in Continuous State and Action Spaces28 Jul 2025
- 2025On Arbitrary Predictions from Equally Valid Models25 Jul 2025
- 2025Optimizers Qualitatively Alter Solutions And We Should Leverage This16 Jul 2025
- 2025GPTailor: Large Language Model Pruning Through Layer Cutting and Stitching25 Jun 2025
- 2025Neural Network Reprogrammability: A Unified Theme on Model Reprogramming, Prompt Tuning, and Prompt Instruction5 Jun 2025
- 2025FrontierCO: Real-World and Large-Scale Evaluation of Machine Learning Solvers for Combinatorial Optimization22 May 2025
- 2025MergeBench: A Benchmark for Merging Domain-Specialized LLMs16 May 2025
- 2025The Power of Random Features and the Limits of Distribution-Free Gradient Descent15 May 2025
- 2025Block-Biased Mamba for Long-Range Sequence Processing13 May 2025
- 2025Deep Learning Optimization Using Self-Adaptive Weighted Auxiliary Variables30 Apr 2025
- 2025Do Two AI Scientists Agree?3 Apr 2025
- 2025Towards Understanding the Optimization Mechanisms in Deep Learning29 Mar 2025
- 2025Feature Learning beyond the Lazy-Rich Dichotomy: Insights from Representational Geometry23 Mar 2025
- 2025Understanding Inverse Reinforcement Learning under Overparameterization: Non-Asymptotic Analysis and Global Optimality22 Mar 2025
- 2025High-entropy Advantage in Neural Networks' Generalizability17 Mar 2025
- 2025Langevin Monte-Carlo Provably Learns Depth Two Neural Nets at Any Size and Data13 Mar 2025
- 2025Path Regularization: A Near-Complete and Optimal Nonasymptotic Generalization Theory for Multilayer Neural Networks and Double Descent Phenomenon3 Mar 2025
- 2025Scaling Law Phenomena Across Regression Paradigms: Multiple and Kernel Approaches3 Mar 2025
50 newest of 340 All 340 (JSON)
- 2021Knowledge Distillation Methods for Efficient Unsupervised Adaptation Across Multiple Domainscited by 15
- 2021Understanding Robustness in Teacher-Student Setting: A New Perspectivecited by 1
- 2021Boosting of Head Pose Estimation by Knowledge Distillationcited by 0
- 2025Rethinking Decoupled Knowledge Distillation: A Predictive Distribution Perspectivecited by 0
- 2025Rethinking Decoupled Knowledge Distillation: A Predictive Distribution Perspective4 Dec 2025
- 2021Boosting of Head Pose Estimation by Knowledge Distillation20 Aug 2021
- 2021Understanding Robustness in Teacher-Student Setting: A New Perspective25 Feb 2021
- 2021Knowledge Distillation Methods for Efficient Unsupervised Adaptation Across Multiple Domains18 Jan 2021
- 2021Don't Forget to Sign the Gradients!cited by 2
- 2021Don't Forget to Sign the Gradients!5 Mar 2021
- 2022Not All Patches are What You Need: Expediting Vision Transformers via Token Reorganizationscited by 363
- 2021Training Neural Networks with Fixed Sparse Maskscited by 214
- 2021The Values Encoded in Machine Learning Researchcited by 172
- 2022Neural Networks can Learn Representations with Gradient Descentcited by 161
- 2022High-dimensional Asymptotics of Feature Learning: How One Gradient Step Improves the Representationcited by 150
- 2022A Kernel-Based View of Language Model Fine-Tuningcited by 121
- 2021Toward Understanding the Feature Learning Process of Self-supervised Contrastive Learningcited by 119
- 2021Generalization error of random features and kernel methods: hypercontractivity and kernel matrix concentrationcited by 110
- 2021What Happens after SGD Reaches Zero Loss? --A Mathematical Frameworkcited by 107
- 2023Scan and Snap: Understanding Training Dynamics and Token Composition in 1-layer Transformercited by 102
- 2022The Design Space of E(3)-Equivariant Atom-Centered Interatomic Potentialscited by 101
- 2021Theory of overparametrization in quantum neural networkscited by 100
12 most cited of 320 All 320 (JSON)
- 2026Relative Generalization Invariance of LLM Pretraining26 Sep 2026
- 2026A Data-dependent Early Stopping Rule using Rademacher Complexity with L1-norm25 Aug 2026
- 2026FlexPooling with Simple Auxiliary Classifiers in Deep Networks12 Jun 2026
- 2026Activation Functions, Statistics and Learning of Higher-Order Interactions in Restricted Boltzmann Machines18 May 2026
- 2026Strategic Over-Parameterization for Generalizable Low-Rank Adaptation15 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
- 2026Towards Initialization-dependent and Non-vacuous Generalization Bounds for Overparameterized Shallow Neural Networks1 Apr 2026
- 2026Curvature-Guided LoRA: Matching Full Fine-Tuning in Function Space31 Mar 2026
- 2026Beyond Consistency: Inference for the Relative risk functional in Deep Nonparametric Cox Models25 Mar 2026
- 2026An Optimal Control Approach To Transformer Training10 Mar 2026
- 2026Convex Dominance in Deep Learning I: A Scaling Law of Loss and Learning Rate6 Feb 2026
12 newest of 320 All 320 (JSON)