Gradient Descent Provably Optimizes Over-parameterized Neural Networks
Not on the map. Why?
Abstract (arXiv)
One of the mysteries in the success of neural networks is randomly initialized first order methods like gradient descent can achieve zero training loss even though the objective function is non-convex and non-smooth. This paper demystifies this surprising phenomenon for two-layer fully connected ReLU activated neural networks. For an m hidden node shallow neural network with ReLU activation and n training data, we show as long as m is large enough and no two inputs are parallel, randomly initialized gradient descent converges to a globally optimal solution at a linear convergence rate for the quadratic loss function. Our analysis relies on the following observation: over-parameterization and random initialization jointly restrict every weight vector to be close to its initialization for all iterations, which allows us to exploit a strong convexity-like property to show that gradient descent converges at a global linear rate to the global optimum. We believe these insights are also useful in analyzing deep models and other first order methods.
Key facts. Cited by 659 papers in the map (data through 14 Oct 2026). First posted on arXiv 4 Oct 2018.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2020: 1; 2021: 163; 2022: 146; 2023: 142; 2024: 80; 2025: 82; 2026: 45.
659 papers.
Sort:
By area:
- 2021FedBN: Federated Learning on Non-IID Features via Local Batch Normalizationcited by 571
- 2021AutoFormer: Searching Transformers for Visual Recognitioncited by 188
- 2022Hidden Progress in Deep Learning: SGD Learns Parities Near the Computational Limitcited by 187
- 2022Neural Networks can Learn Representations with Gradient Descentcited by 161
- 2021The Low-Rank Simplicity Bias in Deep Networkscited by 154
- 2022High-dimensional Asymptotics of Feature Learning: How One Gradient Step Improves the Representationcited by 150
- 2021Fit without fear: remarkable mathematical phenomena of deep learning through the prism of interpolationcited by 132
- 2021Geometry of the Loss Landscape in Overparameterized Neural Networks: Symmetries and Invariancescited by 127
- 2022Modality Competition: What Makes Joint Training of Multi-modal Network Fail in Deep Learning? (Provably)cited by 122
- 2022A Kernel-Based View of Language Model Fine-Tuningcited by 121
- 2023SAM-CLIP: Merging Vision Foundation Models towards Semantic and Spatial Understandingcited by 111
- 2021Generalization error of random features and kernel methods: hypercontractivity and kernel matrix concentrationcited by 110
38 more
- 2021What Happens after SGD Reaches Zero Loss? --A Mathematical Frameworkcited by 107
- 2021Statistically Meaningful Approximation: a Case Study on Approximating Turing Machines with Transformerscited by 102
- 2023Scan and Snap: Understanding Training Dynamics and Token Composition in 1-layer Transformercited by 102
- 2021Theory of overparametrization in quantum neural networkscited by 100
- 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
- 2021Saddle-to-Saddle Dynamics in Deep Linear Networks: Small Initialization Training, Symmetry, and Sparsitycited by 85
- 2021Information-Theoretic Generalization Bounds for Stochastic Gradient Descentcited by 84
- 2023DiffFit: Unlocking Transferability of Large Diffusion Models via Simple Parameter-Efficient Fine-Tuningcited by 84
- 2021Gradient Descent on Two-layer Nets: Margin Maximization and Simplicity Biascited by 83
- 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
- 2021Pixelated Butterfly: Simple and Efficient Sparse training for Neural Network Modelscited by 76
- 2022Benign Overfitting without Linearity: Neural Network Classifiers Trained by Gradient Descent for Noisy Linear Datacited by 74
- 2023Transformers as Support Vector Machinescited by 74
- 2021On the Validity of Modeling SGD with Stochastic Differential Equations (SDEs)cited by 73
- 2021On the Implicit Bias of Initialization Shape: Beyond Infinitesimal Mirror Descentcited by 70
- 2021Classifying high-dimensional Gaussian mixtures: Where kernel methods fail and neural networks succeedcited by 69
- 2022Domain-Adjusted Regression or: ERM May Already Learn Features Sufficient for Out-of-Distribution Generalizationcited by 69
- 2022Analyzing Sharpness along GD Trajectory: Progressive Sharpening and Edge of Stabilitycited by 63
- 2022Neural Networks Efficiently Learn Low-Dimensional Representations with SGDcited by 63
- 2021Lost in Pruning: The Effects of Pruning Neural Networks beyond Test Accuracycited by 62
- 2021Training Multi-Layer Over-Parametrized Neural Network in Subquadratic Timecited by 62
- 2022Theory of Graph Neural Networks: Representation and Learningcited by 61
- 2023ZiCo: Zero-shot NAS via Inverse Coefficient of Variation on Gradientscited by 58
- 2021Tensor Programs IIb: Architectural Universality of Neural Tangent Kernel Training Dynamicscited by 57
- 2021Optimization of Graph Neural Networks: Implicit Acceleration by Skip Connections and More Depthcited by 55
- 2022Towards Understanding Mixture of Experts in Deep Learningcited by 52
- 2022Theoretical Guarantees for Permutation-Equivariant Quantum Neural Networkscited by 52
- 2023JoMA: Demystifying Multilayer Transformers via JOint Dynamics of MLP and Attentioncited by 52
- 2022Learning threshold neurons via the "edge of stability"cited by 51
- 2022Mechanistic Mode Connectivitycited by 50
- 2021Deep Kronecker neural networks: A general framework for neural networks with adaptive activation functionscited by 49
- 2021Understanding the Generalization of Adam in Learning Neural Networks with Proper Regularizationcited by 49
- 2023A Theory of Non-Linear Feature Learning with One Gradient Step in Two-Layer Neural Networkscited by 49
- 2024From Lazy to Rich: Exact Learning Dynamics in Deep Linear Networkscited by 49
- 2021A Local Convergence Theory for Mildly Over-Parameterized Two-Layer Neural Networkcited by 47
50 most cited of 659 All 659 (JSON)
- 2026A Tropical Geometry View of Forgetting: A Per-Unit Projector for Knowledge-Preserving Fine-Tuning3 Oct 2026
- 2026Training-Loss Guarantees for Muon with Finite-Step Newton--Schulz Orthogonalization2 Oct 2026
- 2026Removing spurious minima for planar features by skip connections1 Oct 2026
- 2026Dimension-Free Rank Lifting from Random Hyperplane Arrangements30 Sep 2026
- 2026Not all solutions are created equal: An analytical dissociation of functional and representational similarity in deep linear neural networks30 Sep 2026
- 2026The Price of Locality: Why Forward-Forward Underperforms Backpropagation?27 Sep 2026
- 2026Relative Generalization Invariance of LLM Pretraining26 Sep 2026
- 2026Convergence Theory of Knowledge Distillation in Asynchronous P2P Gossip Learning Network1 Sep 2026
- 2026A Data-dependent Early Stopping Rule using Rademacher Complexity with L1-norm25 Aug 2026
- 2026Convergence Guarantees of Gradient Descent for Neural Networks via Generalized Lipschitz Smoothness11 Aug 2026
- 2026Tight Worst-Case Bounds for the Smallest Eigenvalue of ReLU NTK Gram Matrices4 Aug 2026
- 2026On the Implicit Flatness Bias of Sharpness-Aware Minimization: A Linear Stability Analysis with Quantitative Hyperparameter Bounds4 Aug 2026
38 more
- 2026Variable Importance Identification Through Lazy Training for Binary Classification25 Jul 2026
- 2026The Differential Neural Tangent Kernel and Its Positivity11 Jul 2026
- 2026Dynamics of Gradient Descent with Large Step Size Near a Manifold of Flat Minima9 Jul 2026
- 2026PromptDyG: Test-Time Prompt Adaptation on Dynamic Graphs22 Jun 2026
- 2026Minimax-Optimal Generalization Bounds for Smooth Deep Neural Networks Trained by (Stochastic) Gradient Descent4 Jun 2026
- 2026Optimal Rates for Generalization of Gradient Descent Methods with Deep Neural Networks4 Jun 2026
- 2026When Both Layers Learn: Training Dynamics of Representing Linear Models via ReLU Networks3 Jun 2026
- 2026A Geometric Characterization of the Stationary Plateau for Two-Layer Neural Networks3 Jun 2026
- 2026Provable Data Scaling Law for Meta Learning via Complexity Minimization1 Jun 2026
- 2026Bayesian Inference with Shaped Deep Non-linear MLPs29 May 2026
- 2026LARK: Learnability-Grounded Trajectory Selection for Efficient Reasoning Distillation28 May 2026
- 2026Label-NTK Alignments and A Tighter Convergence Bound in the NTK Regime24 May 2026
- 2026Uniform-in-Time Weak Propagation-of-Chaos in Shallow Neural Networks21 May 2026
- 2026Canonical Regularisation of Wide Feature-Learning Neural Networks18 May 2026
- 2026Strategic Over-Parameterization for Generalizable Low-Rank Adaptation15 May 2026
- 2026Zero-shot Quantum Neural Architecture Search12 May 2026
- 2026Man, Machine, and Mathematics29 Apr 2026
- 2026A Layer Separation Optimization Framework for Cross-Entropy Training in Deep Learning25 Apr 2026
- 2026Global Convergence of Policy Gradient Methods for ReLU Controllers in Linear Quadratic Regulation24 Apr 2026
- 2026Conservation Law Breaking at the Edge of Stability: A Spectral Theory of Non-Convex Neural Network Optimization8 Apr 2026
- 2026Towards Initialization-dependent and Non-vacuous Generalization Bounds for Overparameterized Shallow Neural Networks1 Apr 2026
- 2026Gradient Descent with Projection Finds Over-Parameterized Neural Networks for Learning Low-Degree Polynomials with Nearly Minimax Optimal Rate22 Mar 2026
- 2026On the Learning Dynamics of Two-layer Linear Networks with Label Noise SGD11 Mar 2026
- 2026Large Spikes in Stochastic Gradient Descent: A Large-Deviations View10 Mar 2026
- 2026An Optimal Control Approach To Transformer Training10 Mar 2026
- 2026Weak-SIGReg: Covariance Regularization for Stable Deep Learning6 Mar 2026
- 2026What Does Flow Matching Bring To TD Learning?4 Mar 2026
- 2026Plug, Play, and Fortify: A Low-Cost Module for Robust Multimodal Image Understanding Models26 Feb 2026
- 2026A Theoretical Framework for LLM Fine-tuning Using Early Stopping for Non-random Initialization15 Feb 2026
- 2026Convex Dominance in Deep Learning I: A Scaling Law of Loss and Learning Rate6 Feb 2026
- 2026Target noise: A pre-training based neural network initialization for efficient high resolution learning6 Feb 2026
- 2026The Multiple Ticket Hypothesis: Random Sparse Subnetworks Suffice for RLVR2 Feb 2026
- 2026Finite-Time Analysis of Gradient Descent for Shallow Transformers23 Jan 2026
- 2025Unifying Learning Dynamics and Generalization in Transformers Scaling Law26 Dec 2025
- 2025Shallow Neural Networks Learn Low-Degree Spherical Polynomials with Feature Learning by Learnable Channel Attention23 Dec 2025
- 2025Entropy-Reservoir Bregman Projection: An Information-Geometric Unification of Model Collapse16 Dec 2025
- 2025TPV: Parameter Perturbations Through the Lens of Test Prediction Variance11 Dec 2025
- 2025Dynamical Implicit Neural Representations26 Nov 2025
50 newest of 659 All 659 (JSON)
- 2023DiffFit: Unlocking Transferability of Large Diffusion Models via Simple Parameter-Efficient Fine-Tuningcited by 84
- 2024Neural Network-Based Score Estimation in Diffusion Models: Optimization and Generalizationcited by 42
- 2024Evaluating the design space of diffusion-based generative modelscited by 26
- 2023Discovering a reaction-diffusion model for Alzheimer's disease by combining PINNs with symbolic regressioncited by 25
- 2026What Does Flow Matching Bring To TD Learning?cited by 2
- 2020Why Adversarial Interaction Creates Non-Homogeneous Patterns: A Pseudo-Reaction-Diffusion Model for Turing Instabilitycited by 0
- 2026What Does Flow Matching Bring To TD Learning?4 Mar 2026
- 2024Evaluating the design space of diffusion-based generative models18 Jun 2024
- 2024Neural Network-Based Score Estimation in Diffusion Models: Optimization and Generalization28 Jan 2024
- 2023Discovering a reaction-diffusion model for Alzheimer's disease by combining PINNs with symbolic regression16 Jul 2023
- 2023DiffFit: Unlocking Transferability of Large Diffusion Models via Simple Parameter-Efficient Fine-Tuning13 Apr 2023
- 2020Why Adversarial Interaction Creates Non-Homogeneous Patterns: A Pseudo-Reaction-Diffusion Model for Turing Instability1 Oct 2020
- 2024Recurrent Natural Policy Gradient for POMDPscited by 8
- 2024SF-DQN: Provable Knowledge Transfer using Successor Feature for Deep Reinforcement Learningcited by 5
- 2024An Improved Finite-time Analysis of Temporal Difference Learning with Deep Neural Networkscited by 4
- 2026Global Convergence of Policy Gradient Methods for ReLU Controllers in Linear Quadratic Regulationcited by 0
- 2026Global Convergence of Policy Gradient Methods for ReLU Controllers in Linear Quadratic Regulation24 Apr 2026
- 2024Recurrent Natural Policy Gradient for POMDPs28 May 2024
- 2024SF-DQN: Provable Knowledge Transfer using Successor Feature for Deep Reinforcement Learning24 May 2024
- 2024An Improved Finite-time Analysis of Temporal Difference Learning with Deep Neural Networks7 May 2024
- 2021Locality defeats the curse of dimensionality in convolutional teacher-student scenarioscited by 26
- 2023On the Impact of Overparameterization on the Training of a Shallow Neural Network in High Dimensionscited by 21
- 2021On Learnability via Gradient Method for Two-Layer ReLU Neural Networks in Teacher-Student Settingcited by 16
- 2026Convergence Theory of Knowledge Distillation in Asynchronous P2P Gossip Learning Networkcited by 0
- 2026Convergence Theory of Knowledge Distillation in Asynchronous P2P Gossip Learning Network1 Sep 2026
- 2023On the Impact of Overparameterization on the Training of a Shallow Neural Network in High Dimensions7 Nov 2023
- 2021Locality defeats the curse of dimensionality in convolutional teacher-student scenarios16 Jun 2021
- 2021On Learnability via Gradient Method for Two-Layer ReLU Neural Networks in Teacher-Student Setting11 Jun 2021
- 2021Offline Neural Contextual Bandits: Pessimism, Optimization and Generalizationcited by 25
- 2022Structural Estimation of Markov Decision Processes in High-Dimensional State Space with Finite-Time Guaranteescited by 12
- 2021Sample Complexity of Offline Reinforcement Learning with Deep ReLU Networkscited by 8
- 2022Structural Estimation of Markov Decision Processes in High-Dimensional State Space with Finite-Time Guarantees4 Oct 2022
- 2021Offline Neural Contextual Bandits: Pessimism, Optimization and Generalization27 Nov 2021
- 2021Sample Complexity of Offline Reinforcement Learning with Deep ReLU Networks11 Mar 2021
- 2022Towards Understanding Mixture of Experts in Deep Learningcited by 52
- 2021FedBN: Federated Learning on Non-IID Features via Local Batch Normalizationcited by 571
- 2021AutoFormer: Searching Transformers for Visual Recognitioncited by 188
- 2022Hidden Progress in Deep Learning: SGD Learns Parities Near the Computational Limitcited by 187
- 2022Neural Networks can Learn Representations with Gradient Descentcited by 161
- 2021The Low-Rank Simplicity Bias in Deep Networkscited by 154
- 2022High-dimensional Asymptotics of Feature Learning: How One Gradient Step Improves the Representationcited by 150
- 2021Fit without fear: remarkable mathematical phenomena of deep learning through the prism of interpolationcited by 132
- 2021Geometry of the Loss Landscape in Overparameterized Neural Networks: Symmetries and Invariancescited by 127
- 2022Modality Competition: What Makes Joint Training of Multi-modal Network Fail in Deep Learning? (Provably)cited by 122
- 2022A Kernel-Based View of Language Model Fine-Tuningcited by 121
- 2023SAM-CLIP: Merging Vision Foundation Models towards Semantic and Spatial Understandingcited by 111
- 2021Generalization error of random features and kernel methods: hypercontractivity and kernel matrix concentrationcited by 110
12 most cited of 638 All 638 (JSON)
- 2026A Tropical Geometry View of Forgetting: A Per-Unit Projector for Knowledge-Preserving Fine-Tuning3 Oct 2026
- 2026Training-Loss Guarantees for Muon with Finite-Step Newton--Schulz Orthogonalization2 Oct 2026
- 2026Removing spurious minima for planar features by skip connections1 Oct 2026
- 2026Dimension-Free Rank Lifting from Random Hyperplane Arrangements30 Sep 2026
- 2026Not all solutions are created equal: An analytical dissociation of functional and representational similarity in deep linear neural networks30 Sep 2026
- 2026The Price of Locality: Why Forward-Forward Underperforms Backpropagation?27 Sep 2026
- 2026Relative Generalization Invariance of LLM Pretraining26 Sep 2026
- 2026A Data-dependent Early Stopping Rule using Rademacher Complexity with L1-norm25 Aug 2026
- 2026Convergence Guarantees of Gradient Descent for Neural Networks via Generalized Lipschitz Smoothness11 Aug 2026
- 2026Tight Worst-Case Bounds for the Smallest Eigenvalue of ReLU NTK Gram Matrices4 Aug 2026
- 2026On the Implicit Flatness Bias of Sharpness-Aware Minimization: A Linear Stability Analysis with Quantitative Hyperparameter Bounds4 Aug 2026
- 2026Variable Importance Identification Through Lazy Training for Binary Classification25 Jul 2026
12 newest of 638 All 638 (JSON)