An Improved Analysis of Training Over-parameterized Deep Neural Networks
Not on the map. Why?
Abstract (arXiv)
A recent line of research has shown that gradient-based algorithms with random initialization can converge to the global minima of the training loss for over-parameterized (i.e., sufficiently wide) deep neural networks. However, the condition on the width of the neural network to ensure the global convergence is very stringent, which is often a high-degree polynomial in the training sample size n (e.g., O(n24)). In this paper, we provide an improved analysis of the global convergence of (stochastic) gradient descent for training deep neural networks, which only requires a milder over-parameterization condition than previous work in terms of the training sample size and other problem-dependent parameters. The main technical contributions of our analysis include (a) a tighter gradient lower bound that leads to a faster convergence of the algorithm, and (b) a sharper characterization of the trajectory length of the algorithm. By specializing our result to two-layer (i.e., one-hidden-layer) neural networks, it also provides a milder over-parameterization condition than the best-known result in prior work.
Key facts. Cited by 120 papers in the map (data through 7 Oct 2026). First posted on arXiv 11 Jun 2019.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 30; 2022: 32; 2023: 21; 2024: 23; 2025: 9; 2026: 5.
120 papers.
Sort:
By area:
- 2021Do We Actually Need Dense Over-Parameterization? In-Time Over-Parameterization in Sparse Trainingcited by 107
- 2023A Theoretical Understanding of Shallow Vision Transformers: Learning, Generalization, and Sample Complexitycited by 77
- 2022DisPFL: Towards Communication-Efficient Personalized Federated Learning via Decentralized Sparse Trainingcited by 74
- 2021Training Multi-Layer Over-Parametrized Neural Network in Subquadratic Timecited by 62
- 2023Revisiting Weighted Aggregation in Federated Learning with Neural Networkscited by 57
- 2021What can linearized neural networks actually say about generalization?cited by 49
- 2021Understanding the Generalization of Adam in Learning Neural Networks with Proper Regularizationcited by 48
- 2021Asymmetric Loss Functions for Learning with Noisy Labelscited by 46
- 2021On the Proof of Global Convergence of Gradient Descent for Deep ReLU Networks with Linear Widthscited by 44
- 2022Memorization and Optimization in Deep Neural Networks with Minimum Over-parameterizationcited by 36
- 2024How Do Nonlinear Transformers Learn and Generalize in In-Context Learning?cited by 35
- 2021Continuous vs. Discrete Optimization of Deep Neural Networkscited by 34
38 more
- 2021Does Preprocessing Help Training Over-parameterized Neural Networks?cited by 32
- 2023A Fast Optimization View: Reformulating Single Layer Attention in LLM Based on Tensor and SVM Trick, and Solving It in Matrix Multiplication Timecited by 30
- 2022Bypass Exponential Time Preprocessing: Fast Neural Network Training via Weight-Data Correlation Preprocessingcited by 29
- 2023Catapults in SGD: spikes in the training loss and their impact on generalization through feature learningcited by 29
- 2021Properties of the After Kernelcited by 27
- 2021Subquadratic Overparameterization for Shallow Neural Networkscited by 26
- 2023How to Protect Copyright Data in Optimization of Large Language Models?cited by 26
- 2024Evaluating the design space of diffusion-based generative modelscited by 25
- 2022Characterizing the Spectrum of the NTK via a Power Series Expansioncited by 22
- 2023Generalization and Stability of Interpolating Neural Networks with Minimal Widthcited by 21
- 2024A Survey on Statistical Theory of Deep Learning: Approximation, Training Dynamics, and Generative Modelscited by 20
- 2023Six Lectures on Linearized Neural Networkscited by 19
- 2022Demystifying the Neural Tangent Kernel from a Practical Perspective: Can it be trusted for Neural Architecture Search without training?cited by 18
- 2021Provable Generalization of SGD-trained Neural Networks of Any Width in the Presence of Adversarial Label Noisecited by 17
- 2021A global convergence theory for deep ReLU implicit networks via over-parameterizationcited by 17
- 2022Quadratic models for understanding catapult dynamics of neural networkscited by 17
- 2021Learning with Gradient Descent and Weakly Convex Lossescited by 16
- 2022Bounding the Width of Neural Networks via Coupled Initialization -- A Worst Case Analysiscited by 15
- 2022An Analysis of Attention via the Lens of Exchangeability and Latent Variable Modelscited by 15
- 2024FedFisher: Leveraging Fisher Information for One-Shot Federated Learningcited by 15
- 2021Understanding Overparameterization in Generative Adversarial Networkscited by 14
- 2021The Separation Capacity of Random Neural Networkscited by 14
- 2021Why Lottery Ticket Wins? A Theoretical Perspective of Sample Complexity on Pruned Neural Networkscited by 14
- 2024What Improves the Generalization of Graph Transformers? A Theoretical Dive into the Self-attention and Positional Encodingcited by 14
- 2024Differential Privacy Mechanisms in Neural Tangent Kernel Regressioncited by 14
- 2024Wide Neural Networks Trained with Weight Decay Provably Exhibit Neural Collapsecited by 14
- 2023On the Convergence of Encoder-only Shallow Transformerscited by 13
- 2024Bounds for the smallest eigenvalue of the NTK for arbitrary spherical data of arbitrary dimensioncited by 13
- 2024The Persistence of Neural Collapse Despite Low-Rank Biascited by 12
- 2022Spectral Bias Outside the Training Set for Deep Networks in the Kernel Regimecited by 11
- 2023Efficient SGD Neural Network Training via Sublinear Activated Neuron Identificationcited by 11
- 2024From Activation to Initialization: Scaling Insights for Optimizing Neural Fieldscited by 11
- 2021Local SGD Optimizes Overparameterized Neural Networks in Polynomial Timecited by 10
- 2022Global Convergence of Over-parameterized Deep Equilibrium Modelscited by 10
- 2022Nonlinear Sufficient Dimension Reduction with a Stochastic Neural Networkcited by 10
- 2022Implicit Bias of MSE Gradient Optimization in Underparameterized Neural Networkscited by 9
- 2022A Kernel-Expanded Stochastic Neural Networkcited by 9
- 2023An Automatic Learning Rate Schedule Algorithm for Achieving Faster Convergence and Steeper Descentcited by 9
50 most cited of 120 All 120 (JSON)
- 2026A Geometric Characterization of the Stationary Plateau for Two-Layer Neural Networks3 Jun 2026
- 2026Label-NTK Alignments and A Tighter Convergence Bound in the NTK Regime24 May 2026
- 2026A Layer Separation Optimization Framework for Cross-Entropy Training in Deep Learning25 Apr 2026
- 2026Gradient Descent with Projection Finds Over-Parameterized Neural Networks for Learning Low-Degree Polynomials with Nearly Minimax Optimal Rate22 Mar 2026
- 2026Finite-Time Analysis of Gradient Descent for Shallow Transformers23 Jan 2026
- 2025Shallow Neural Networks Learn Low-Degree Spherical Polynomials with Feature Learning by Learnable Channel Attention23 Dec 2025
- 2025Quantitative convergence of trained single layer neural networks to Gaussian processes29 Sep 2025
- 2025Explaining Grokking and Information Bottleneck through Neural Collapse Emergence25 Sep 2025
- 2025Data Uniformity Improves Training Efficiency and More, with a Convergence Framework Beyond the NTK Regime30 Jun 2025
- 2025Training NTK to Generalize with KARE16 May 2025
- 2025Deep Learning Optimization Using Self-Adaptive Weighted Auxiliary Variables30 Apr 2025
- 2025High-entropy Advantage in Neural Networks' Generalizability17 Mar 2025
38 more
- 2025Global Convergence and Rich Feature Learning in L-Layer Infinite-Width Neural Networks under μP Parametrization12 Mar 2025
- 2025Curse of Dimensionality in Neural Network Optimization7 Feb 2025
- 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
- 2024The Persistence of Neural Collapse Despite Low-Rank Bias30 Oct 2024
- 2024On the Convergence Theory of Pipeline Gradient-based Analog In-memory Training19 Oct 2024
- 2024Wide Neural Networks Trained with Weight Decay Provably Exhibit Neural Collapse7 Oct 2024
- 2024Unraveling the Hessian: A Key to Smooth Convergence in Loss Function Landscapes18 Sep 2024
- 2024Hybrid Coordinate Descent for Efficient Neural Network Learning Using Line Search and Gradient Descent2 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
- 2024Evaluating the design space of diffusion-based generative models18 Jun 2024
- 2024What Improves the Generalization of Graph Transformers? A Theoretical Dive into the Self-attention and Positional Encoding4 Jun 2024
- 2024Novel Kernel Models and Exact Representor Theory for Neural Networks Beyond the Over-Parameterized Regime24 May 2024
- 2024Bounds for the smallest eigenvalue of the NTK for arbitrary spherical data of arbitrary dimension23 May 2024
- 2024Approximation and Gradient Descent Training with Neural Networks19 May 2024
- 2024Error Analysis of Three-Layer Neural Network Trained with PGD for Deep Ritz Method19 May 2024
- 2024Class-wise Activation Unravelling the Engima of Deep Double Descent13 May 2024
- 2024From Activation to Initialization: Scaling Insights for Optimizing Neural Fields28 Mar 2024
- 2024FedFisher: Leveraging Fisher Information for One-Shot Federated Learning19 Mar 2024
- 2024How Do Nonlinear Transformers Learn and Generalize in In-Context Learning?23 Feb 2024
- 2024Uncertainty Quantification of Graph Convolution Neural Network Models of Evolving Processes17 Feb 2024
- 2024A Resource Model For Neural Scaling Law7 Feb 2024
- 2024Analyzing the Neural Tangent Kernel of Periodically Activated Coordinate Networks7 Feb 2024
- 2024A Survey on Statistical Theory of Deep Learning: Approximation, Training Dynamics, and Generative Models14 Jan 2024
- 2023On the Convergence of Encoder-only Shallow Transformers2 Nov 2023
- 2023An Automatic Learning Rate Schedule Algorithm for Achieving Faster Convergence and Steeper Descent17 Oct 2023
- 2023Sparse Deep Learning for Time Series Data: Theory and Applications5 Oct 2023
- 2023On the Computational Entanglement of Distant Features in Adversarial Machine Learning27 Sep 2023
- 2023A Fast Optimization View: Reformulating Single Layer Attention in LLM Based on Tensor and SVM Trick, and Solving It in Matrix Multiplication Time14 Sep 2023
- 2023Approximation Results for Gradient Descent trained Neural Networks9 Sep 2023
- 2023Six Lectures on Linearized Neural Networks25 Aug 2023
- 2023How to Protect Copyright Data in Optimization of Large Language Models?23 Aug 2023
- 2023Convergence of Two-Layer Regression with Nonlinear Units16 Aug 2023
- 2023Efficient SGD Neural Network Training via Sublinear Activated Neuron Identification13 Jul 2023
- 2023Catapults in SGD: spikes in the training loss and their impact on generalization through feature learning7 Jun 2023
- 2023PGrad: Learning Principal Gradients For Domain Generalization2 May 2023
- 2023Connected Superlevel Set in (Deep) Reinforcement Learning and its Application to Minimax Theorems23 Mar 2023
50 newest of 120 All 120 (JSON)
- 2021Do We Actually Need Dense Over-Parameterization? In-Time Over-Parameterization in Sparse Trainingcited by 107
- 2023A Theoretical Understanding of Shallow Vision Transformers: Learning, Generalization, and Sample Complexitycited by 77
- 2022DisPFL: Towards Communication-Efficient Personalized Federated Learning via Decentralized Sparse Trainingcited by 74
- 2021Training Multi-Layer Over-Parametrized Neural Network in Subquadratic Timecited by 62
- 2023Revisiting Weighted Aggregation in Federated Learning with Neural Networkscited by 57
- 2021What can linearized neural networks actually say about generalization?cited by 49
- 2021Understanding the Generalization of Adam in Learning Neural Networks with Proper Regularizationcited by 48
- 2021Asymmetric Loss Functions for Learning with Noisy Labelscited by 46
- 2021On the Proof of Global Convergence of Gradient Descent for Deep ReLU Networks with Linear Widthscited by 44
- 2022Memorization and Optimization in Deep Neural Networks with Minimum Over-parameterizationcited by 36
- 2024How Do Nonlinear Transformers Learn and Generalize in In-Context Learning?cited by 35
- 2021Continuous vs. Discrete Optimization of Deep Neural Networkscited by 34
12 most cited of 119 All 119 (JSON)
- 2026A Geometric Characterization of the Stationary Plateau for Two-Layer Neural Networks3 Jun 2026
- 2026Label-NTK Alignments and A Tighter Convergence Bound in the NTK Regime24 May 2026
- 2026A Layer Separation Optimization Framework for Cross-Entropy Training in Deep Learning25 Apr 2026
- 2026Gradient Descent with Projection Finds Over-Parameterized Neural Networks for Learning Low-Degree Polynomials with Nearly Minimax Optimal Rate22 Mar 2026
- 2026Finite-Time Analysis of Gradient Descent for Shallow Transformers23 Jan 2026
- 2025Shallow Neural Networks Learn Low-Degree Spherical Polynomials with Feature Learning by Learnable Channel Attention23 Dec 2025
- 2025Quantitative convergence of trained single layer neural networks to Gaussian processes29 Sep 2025
- 2025Explaining Grokking and Information Bottleneck through Neural Collapse Emergence25 Sep 2025
- 2025Data Uniformity Improves Training Efficiency and More, with a Convergence Framework Beyond the NTK Regime30 Jun 2025
- 2025Training NTK to Generalize with KARE16 May 2025
- 2025Deep Learning Optimization Using Self-Adaptive Weighted Auxiliary Variables30 Apr 2025
- 2025High-entropy Advantage in Neural Networks' Generalizability17 Mar 2025
12 newest of 119 All 119 (JSON)