Theoretical insights into the optimization landscape of over-parameterized shallow neural networks
Not on the map. Why?
Abstract (arXiv)
In this paper we study the problem of learning a shallow artificial neural network that best fits a training data set. We study this problem in the over-parameterized regime where the number of observations are fewer than the number of parameters in the model. We show that with quadratic activations the optimization landscape of training such shallow neural networks has certain favorable characteristics that allow globally optimal models to be found efficiently using a variety of local search heuristics. This result holds for an arbitrary training data of input/output pairs. For differentiable activation functions we also show that gradient descent, when suitably initialized, converges at a linear rate to a globally optimal model. This result focuses on a realizable model where the inputs are chosen i.i.d. from a Gaussian distribution and the labels are generated according to planted weight coefficients.
Key facts. Cited by 128 papers in the map (data through 14 Oct 2026). First posted on arXiv 16 Jul 2017.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 34; 2022: 35; 2023: 24; 2024: 11; 2025: 18; 2026: 6.
128 papers.
Sort:
By area:
- 2021A Field Guide to Federated Optimizationcited by 315
- 2022Neural Networks can Learn Representations with Gradient Descentcited by 161
- 2021Continual Learning in the Teacher-Student Setup: Impact of Task Similaritycited by 88
- 2022Benign Overfitting without Linearity: Neural Network Classifiers Trained by Gradient Descent for Noisy Linear Datacited by 74
- 2021Tensor Methods in Computer Vision and Deep Learningcited by 52
- 2022Neural Collapse with Normalized Features: A Geometric Analysis over the Riemannian Manifoldcited by 50
- 2024LoRA Training in the NTK Regime has No Spurious Local Minimacited by 48
- 2022Implicit Bias in Leaky ReLU Networks Trained on High-Dimensional Datacited by 46
- 2021Constrained Learning with Non-Convex Lossescited by 42
- 2021Recent Advances on Neural Network Pruning at Initializationcited by 41
- 2021Adversarial Robustness with Semi-Infinite Constrained Learningcited by 39
- 2022Probabilistically Robust Learning: Balancing Average- and Worst-case Performancecited by 37
38 more
- 2022Memorization and Optimization in Deep Neural Networks with Minimum Over-parameterizationcited by 36
- 2022Random Feature Amplification: Feature Learning and Generalization in Neural Networkscited by 35
- 2021Taxonomizing local versus global structure in neural network loss landscapescited by 34
- 2023The Power of Preconditioning in Overparameterized Low-Rank Matrix Sensingcited by 33
- 2022Green Learning: Introduction, Examples and Outlookcited by 30
- 2023NTK-SAP: Improving neural network pruning by aligning training dynamicscited by 30
- 2022Flat minima generalize for low-rank matrix recoverycited by 28
- 2023SPDF: Sparse Pre-training and Dense Fine-tuning for Large Language Modelscited by 28
- 2021Overparameterization of deep ResNet: zero loss and mean-field analysiscited by 24
- 2022Convergence of gradient descent for deep neural networkscited by 24
- 2024Bayes-optimal learning of an extensive-width neural network from quadratically many samplescited by 23
- 2022Theoretical Perspectives on Deep Learning Methods in Inverse Problemscited by 22
- 2021The Limitations of Large Width in Neural Networks: A Deep Gaussian Process Perspectivecited by 21
- 2022On the Effective Number of Linear Regions in Shallow Univariate ReLU Networks: Convergence Guarantees and Implicit Biascited by 21
- 2023Generalization and Stability of Interpolating Neural Networks with Minimal Widthcited by 21
- 2023On the Impact of Overparameterization on the Training of a Shallow Neural Network in High Dimensionscited by 21
- 2021Learning Transferable Parameters for Unsupervised Domain Adaptationcited by 19
- 2022How does unlabeled data improve generalization in self-training? A one-hidden-layer theoretical analysiscited by 19
- 2023Provable Guarantees for Nonlinear Feature Learning in Three-Layer Neural Networkscited by 19
- 2023On the Convergence and Sample Complexity Analysis of Deep Q-Networks with ε-Greedy Explorationcited by 17
- 2025Optimizers Qualitatively Alter Solutions And We Should Leverage Thiscited by 17
- 2023Traffic Prediction using Artificial Intelligence: Review of Recent Advances and Emerging Opportunitiescited by 16
- 2022Tensor-on-Tensor Regression: Riemannian Optimization, Over-parameterization, Statistical-computational Gap, and Their Interplaycited by 15
- 2023Resilient Constrained Learningcited by 15
- 2021Neural Spectrahedra and Semidefinite Lifts: Global Convex Optimization of Polynomial Activation Neural Networks in Fully Polynomial-Timecited by 14
- 2021Understanding Overparameterization in Generative Adversarial Networkscited by 14
- 2021Why Lottery Ticket Wins? A Theoretical Perspective of Sample Complexity on Pruned Neural Networkscited by 14
- 2022On Feature Learning in Neural Networks with Global Convergence Guaranteescited by 13
- 2022Identifying good directions to escape the NTK regime and efficiently learn low-degree plus sparse polynomialscited by 13
- 2023Beyond the Universal Law of Robustness: Sharper Laws for Random Features and Neural Tangent Kernelscited by 13
- 2024The Persistence of Neural Collapse Despite Low-Rank Biascited by 13
- 2021Fundamental Tradeoffs in Distributionally Adversarial Trainingcited by 12
- 2021On the Global Convergence of Gradient Descent for multi-layer ResNets in the mean-field regimecited by 12
- 2021Landscape analysis for shallow neural networks: complete classification of critical points for affine target functionscited by 11
- 2021On the Convergence of Shallow Neural Network Training with Randomly Masked Neuronscited by 11
- 2022On the (Non-)Robustness of Two-Layer Neural Networks in Different Learning Regimescited by 11
- 2022Non-convex online learning via algorithmic equivalencecited by 11
- 2023How Spurious Features Are Memorized: Precise Analysis for Random and NTK Featurescited by 11
50 most cited of 128 All 128 (JSON)
- 2026How Width and Data Shape Generalization Scaling Laws in Quadratic Neural Networks26 Jun 2026
- 2026When Both Layers Learn: Training Dynamics of Representing Linear Models via ReLU Networks3 Jun 2026
- 2026Towards Initialization-dependent and Non-vacuous Generalization Bounds for Overparameterized Shallow Neural Networks1 Apr 2026
- 2026A Gaussian Comparison Theorem for Training Dynamics in Machine Learning10 Mar 2026
- 2026Why ReLU? A Bit-Model Dichotomy for Deep Network Training22 Feb 2026
- 2026The Implicit Bias of Adam and Muon on Smooth Homogeneous Neural Networks18 Feb 2026
- 2025Nonlinear Dynamics In Optimization Landscape of Shallow Neural Networks with Tunable Leaky ReLU29 Oct 2025
- 2025Statistical physics of deep learning: Optimal learning of a multi-layer perceptron near interpolation28 Oct 2025
- 2025A Law of Data Reconstruction for Random Features (and Beyond)26 Sep 2025
- 2025Visualization and Analysis of the Loss Landscape in Graph Neural Networks15 Sep 2025
- 2025Optimizers Qualitatively Alter Solutions And We Should Leverage This16 Jul 2025
- 2025Understanding Learning Invariance in Deep Linear Networks16 Jun 2025
38 more
- 2025Models of Heavy-Tailed Mechanistic Universality4 Jun 2025
- 2025Statistical mechanics of extensive-width Bayesian neural networks near interpolation30 May 2025
- 2025Convexified Message-Passing Graph Neural Networks23 May 2025
- 2025Learning with Restricted Boltzmann Machines: Asymptotics of AMP and GD in High Dimensions23 May 2025
- 2025The Nuclear Route: Sharp Asymptotics of ERM in Overparameterized Quadratic Networks23 May 2025
- 2025Statistical Inference for Online Algorithms22 May 2025
- 2025Curse of Dimensionality in Neural Network Optimization7 Feb 2025
- 2025Information-Theoretic Guarantees for Recovering Low-Rank Tensors from Symmetric Rank-One Measurements7 Feb 2025
- 2025Optimal generalisation and learning transition in extensive-width shallow neural networks near interpolation30 Jan 2025
- 2025A Metric Topology of Deep Learning for Data Classification20 Jan 2025
- 2025Geometry and Optimization of Shallow Polynomial Networks10 Jan 2025
- 2025Linearly Separable Features in Shallow Nonlinear Networks: Width Scales Polynomially with Intrinsic Data Dimension4 Jan 2025
- 2024The Persistence of Neural Collapse Despite Low-Rank Bias30 Oct 2024
- 2024Entity Insertion in Multilingual Linked Corpora: The Case of Wikipedia5 Oct 2024
- 2024Bayes-optimal learning of an extensive-width neural network from quadratically many samples7 Aug 2024
- 2024SF-DQN: Provable Knowledge Transfer using Successor Feature for Deep Reinforcement Learning24 May 2024
- 2024Error Analysis of Three-Layer Neural Network Trained with PGD for Deep Ritz Method19 May 2024
- 2024Near-Optimal Solutions of Constrained Learning Problems18 Mar 2024
- 2024Early Directional Convergence in Deep Homogeneous Neural Networks for Small Initializations12 Mar 2024
- 2024Investigating the Histogram Loss in Regression20 Feb 2024
- 2024LoRA Training in the NTK Regime has No Spurious Local Minima19 Feb 2024
- 2024Uncertainty Quantification of Graph Convolution Neural Network Models of Evolving Processes17 Feb 2024
- 2024Loss Landscape of Shallow ReLU-like Neural Networks: Stationary Points, Saddle Escape, and Network Embedding8 Feb 2024
- 2023Boosting Defect Detection in Manufacturing using Tensor Convolutional Neural Networks29 Dec 2023
- 2023Hidden Minima in Two-Layer ReLU Networks28 Dec 2023
- 2023Learning Hierarchical Polynomials with Three-Layer Neural Networks23 Nov 2023
- 2023On the Impact of Overparameterization on the Training of a Shallow Neural Network in High Dimensions7 Nov 2023
- 2023Should Under-parameterized Student Networks Copy or Average Teacher Weights?3 Nov 2023
- 2023On the Convergence and Sample Complexity Analysis of Deep Q-Networks with ε-Greedy Exploration24 Oct 2023
- 2023Fundamental Limits of Deep Learning-Based Binary Classifiers Trained with Hinge Loss13 Sep 2023
- 2023Global Optimality in Bivariate Gradient-based DAG Learning30 Jun 2023
- 2023Efficient Online Processing with Deep Neural Networks23 Jun 2023
- 2023Gradient is All You Need? How Consensus-Based Optimization can be Interpreted as a Stochastic Relaxation of Gradient Descent16 Jun 2023
- 2023Solving Large-scale Spatial Problems with Convolutional Neural Networks14 Jun 2023
- 2023Resilient Constrained Learning4 Jun 2023
- 2023Traffic Prediction using Artificial Intelligence: Review of Recent Advances and Emerging Opportunities31 May 2023
- 2023How Spurious Features Are Memorized: Precise Analysis for Random and NTK Features20 May 2023
- 2023Better NTK Conditioning: A Free Lunch from (ReLU) Nonlinear Activation in Wide Neural Networks15 May 2023
50 newest of 128 All 128 (JSON)
- 2021Continual Learning in the Teacher-Student Setup: Impact of Task Similaritycited by 88
- 2023On the Impact of Overparameterization on the Training of a Shallow Neural Network in High Dimensionscited by 21
- 2025Geometry and Optimization of Shallow Polynomial Networkscited by 10
- 2022Knowledge Distillation Meets Open-Set Semi-Supervised Learningcited by 9
- 2025Optimal generalisation and learning transition in extensive-width shallow neural networks near interpolationcited by 1
- 2025Optimal generalisation and learning transition in extensive-width shallow neural networks near interpolation30 Jan 2025
- 2025Geometry and Optimization of Shallow Polynomial Networks10 Jan 2025
- 2023On the Impact of Overparameterization on the Training of a Shallow Neural Network in High Dimensions7 Nov 2023
- 2022Knowledge Distillation Meets Open-Set Semi-Supervised Learning13 May 2022
- 2021Continual Learning in the Teacher-Student Setup: Impact of Task Similarity9 Jul 2021
- 2021A Field Guide to Federated Optimizationcited by 315
- 2022Neural Networks can Learn Representations with Gradient Descentcited by 161
- 2022Benign Overfitting without Linearity: Neural Network Classifiers Trained by Gradient Descent for Noisy Linear Datacited by 74
- 2021Tensor Methods in Computer Vision and Deep Learningcited by 52
- 2022Neural Collapse with Normalized Features: A Geometric Analysis over the Riemannian Manifoldcited by 50
- 2024LoRA Training in the NTK Regime has No Spurious Local Minimacited by 48
- 2022Implicit Bias in Leaky ReLU Networks Trained on High-Dimensional Datacited by 46
- 2021Constrained Learning with Non-Convex Lossescited by 42
- 2021Recent Advances on Neural Network Pruning at Initializationcited by 41
- 2021Adversarial Robustness with Semi-Infinite Constrained Learningcited by 39
- 2022Probabilistically Robust Learning: Balancing Average- and Worst-case Performancecited by 37
- 2022Memorization and Optimization in Deep Neural Networks with Minimum Over-parameterizationcited by 36
12 most cited of 122 All 122 (JSON)
- 2026How Width and Data Shape Generalization Scaling Laws in Quadratic Neural Networks26 Jun 2026
- 2026When Both Layers Learn: Training Dynamics of Representing Linear Models via ReLU Networks3 Jun 2026
- 2026Towards Initialization-dependent and Non-vacuous Generalization Bounds for Overparameterized Shallow Neural Networks1 Apr 2026
- 2026A Gaussian Comparison Theorem for Training Dynamics in Machine Learning10 Mar 2026
- 2026Why ReLU? A Bit-Model Dichotomy for Deep Network Training22 Feb 2026
- 2026The Implicit Bias of Adam and Muon on Smooth Homogeneous Neural Networks18 Feb 2026
- 2025Nonlinear Dynamics In Optimization Landscape of Shallow Neural Networks with Tunable Leaky ReLU29 Oct 2025
- 2025Statistical physics of deep learning: Optimal learning of a multi-layer perceptron near interpolation28 Oct 2025
- 2025A Law of Data Reconstruction for Random Features (and Beyond)26 Sep 2025
- 2025Visualization and Analysis of the Loss Landscape in Graph Neural Networks15 Sep 2025
- 2025Optimizers Qualitatively Alter Solutions And We Should Leverage This16 Jul 2025
- 2025Understanding Learning Invariance in Deep Linear Networks16 Jun 2025
12 newest of 122 All 122 (JSON)