Learning One-hidden-layer ReLU Networks via Gradient Descent
Not on the map. Why?
Abstract (arXiv)
We study the problem of learning one-hidden-layer neural networks with Rectified Linear Unit (ReLU) activation function, where the inputs are sampled from standard Gaussian distribution and the outputs are generated from a noisy teacher network. We analyze the performance of gradient descent for training such kind of neural networks based on empirical risk minimization, and provide algorithm-dependent guarantees. In particular, we prove that tensor initialization followed by gradient descent can converge to the ground-truth parameters at a linear rate up to some statistical error. To the best of our knowledge, this is the first work characterizing the recovery guarantee for practical learning of one-hidden-layer ReLU networks with multiple neurons. Numerical experiments verify our theoretical findings.
Key facts. Cited by 50 papers in the map (data through 7 Oct 2026). First posted on arXiv 20 Jun 2018.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 12; 2022: 13; 2023: 12; 2024: 8; 2025: 2; 2026: 3.
50 papers.
Sort:
By area:
- 2022Benign Overfitting in Two-layer Convolutional Neural Networkscited by 90
- 2023DiffFit: Unlocking Transferability of Large Diffusion Models via Simple Parameter-Efficient Fine-Tuningcited by 82
- 2022Neural Networks Efficiently Learn Low-Dimensional Representations with SGDcited by 61
- 2023ZiCo: Zero-shot NAS via Inverse Coefficient of Variation on Gradientscited by 57
- 2021A Local Convergence Theory for Mildly Over-Parameterized Two-Layer Neural Networkcited by 47
- 2023Generalization Ability of Wide Neural Networks on ℝcited by 28
- 2022Hardness of Noise-Free Learning for Two-Hidden-Layer Neural Networkscited by 27
- 2023Over-Parameterization Exponentially Slows Down Gradient Descent for Learning a Single Neuroncited by 25
- 2021Parameter identifiability of a deep feedforward ReLU neural networkcited by 24
- 2024Provably learning a multi-head attention layercited by 21
- 2022How does unlabeled data improve generalization in self-training? A one-hidden-layer theoretical analysiscited by 18
- 2021On Learnability via Gradient Method for Two-Layer ReLU Neural Networks in Teacher-Student Settingcited by 16
38 more
- 2022An Analysis of Attention via the Lens of Exchangeability and Latent Variable Modelscited by 15
- 2023Learning Narrow One-Hidden-Layer ReLU Networkscited by 14
- 2022Local Identifiability of Deep ReLU Neural Networks: the Theorycited by 12
- 2022Excess Risk of Two-Layer ReLU Neural Networks in Teacher-Student Settings and its Superiority to Kernel Methodscited by 10
- 2024Towards Robust Out-of-Distribution Generalization Bounds via Sharpnesscited by 10
- 2021Stable Recovery of Entangled Weights: Towards Robust Identification of Deep Neural Networks from Minimal Samplescited by 8
- 2021Spurious Local Minima Are Common for Deep Neural Networks with Piecewise Linear Activationscited by 7
- 2023A faster and simpler algorithm for learning shallow networkscited by 7
- 2021Efficiently Learning Any One Hidden Layer ReLU Network From Queriescited by 6
- 2024How does promoting the minority fraction affect generalization? A theoretical study of the one-hidden-layer neural network on group imbalancecited by 6
- 2022Magnitude and Angle Dynamics in Training Single ReLU Neuronscited by 5
- 2022Finite Sample Identification of Wide Shallow Neural Networks with Biasescited by 5
- 2023Continuized Acceleration for Quasar Convex Functions in Non-Convex Optimizationcited by 5
- 2021A Convergence Theory Towards Practical Over-parameterized Deep Neural Networkscited by 4
- 2021How much pre-training is enough to discover a good subnetwork?cited by 4
- 2022Deep Architecture Connectivity Matters for Its Convergence: A Fine-Grained Analysiscited by 4
- 2023Toward L_∞-recovery of Nonlinear Functions: A Polynomial Sample Complexity Bound for Gaussian Random Fieldscited by 4
- 2022Overparameterized ReLU Neural Networks Learn the Simplest Models: Neural Isometry and Exact Recoverycited by 3
- 2023Max-affine regression via first-order methodscited by 3
- 2024When Are Bias-Free ReLU Networks Effectively Linear Networks?cited by 3
- 2025Sobolev acceleration for neural networkscited by 3
- 2021Towards Searching Efficient and Accurate Neural Network Architectures in Binary Classification Problemscited by 2
- 2021Coordinate Descent Methods for DC Minimization: Optimality Conditions and Global Convergencecited by 2
- 2023Most Neural Networks Are Almost Learnablecited by 2
- 2024Understanding Representation Learnability of Nonlinear Self-Supervised Learningcited by 2
- 2024Stochastic Bandits with ReLU Neural Networkscited by 2
- 2021Structured Directional Pruning via Perturbation Orthogonal Projectioncited by 1
- 2022Explicitising The Implicit Intrepretability of Deep Neural Networks Via Dualitycited by 1
- 2023Convergence Analysis for Learning Orthonormal Deep Linear Neural Networkscited by 1
- 2024On the Hardness of Learning One Hidden Layer Neural Networkscited by 1
- 2025A Teacher-Student Perspective on the Dynamics of Learning Near the Optimal Pointcited by 1
- 2026When Both Layers Learn: Training Dynamics of Representing Linear Models via ReLU Networkscited by 1
- 2021Self-Regularity of Non-Negative Output Weights for Overparameterized Two-Layer Neural Networkscited by 0
- 2022Demystifying the Global Convergence Puzzle of Learning Over-parameterized ReLU Nets in Very High Dimensionscited by 0
- 2023Improving the Expressive Power of Deep Neural Networks through Integral Activation Transformcited by 0
- 2024Iterative thresholding for non-linear learning in the strong varepsilon-contamination modelcited by 0
- 2026Understanding the Impact of Differentially Private Training on Memorization of Long-Tailed Datacited by 0
- 2026Two-Timescale Fine-tuning Provably Learns New Features for Two-Layer ReLU Networkscited by 0
- 2026Two-Timescale Fine-tuning Provably Learns New Features for Two-Layer ReLU Networks28 Sep 2026
- 2026When Both Layers Learn: Training Dynamics of Representing Linear Models via ReLU Networks3 Jun 2026
- 2026Understanding the Impact of Differentially Private Training on Memorization of Long-Tailed Data1 Feb 2026
- 2025A Teacher-Student Perspective on the Dynamics of Learning Near the Optimal Point17 Dec 2025
- 2025Sobolev acceleration for neural networks24 Sep 2025
- 2024On the Hardness of Learning One Hidden Layer Neural Networks4 Oct 2024
- 2024Iterative thresholding for non-linear learning in the strong varepsilon-contamination model5 Sep 2024
- 2024When Are Bias-Free ReLU Networks Effectively Linear Networks?18 Jun 2024
- 2024Stochastic Bandits with ReLU Neural Networks12 May 2024
- 2024How does promoting the minority fraction affect generalization? A theoretical study of the one-hidden-layer neural network on group imbalance12 Mar 2024
- 2024Towards Robust Out-of-Distribution Generalization Bounds via Sharpness11 Mar 2024
- 2024Provably learning a multi-head attention layer6 Feb 2024
38 more
- 2024Understanding Representation Learnability of Nonlinear Self-Supervised Learning6 Jan 2024
- 2023Improving the Expressive Power of Deep Neural Networks through Integral Activation Transform19 Dec 2023
- 2023Convergence Analysis for Learning Orthonormal Deep Linear Neural Networks24 Nov 2023
- 2023Max-affine regression via first-order methods15 Aug 2023
- 2023A faster and simpler algorithm for learning shallow networks24 Jul 2023
- 2023Most Neural Networks Are Almost Learnable25 May 2023
- 2023Toward L_∞-recovery of Nonlinear Functions: A Polynomial Sample Complexity Bound for Gaussian Random Fields29 Apr 2023
- 2023Learning Narrow One-Hidden-Layer ReLU Networks20 Apr 2023
- 2023DiffFit: Unlocking Transferability of Large Diffusion Models via Simple Parameter-Efficient Fine-Tuning13 Apr 2023
- 2023Over-Parameterization Exponentially Slows Down Gradient Descent for Learning a Single Neuron20 Feb 2023
- 2023Continuized Acceleration for Quasar Convex Functions in Non-Convex Optimization15 Feb 2023
- 2023Generalization Ability of Wide Neural Networks on ℝ12 Feb 2023
- 2023ZiCo: Zero-shot NAS via Inverse Coefficient of Variation on Gradients26 Jan 2023
- 2022An Analysis of Attention via the Lens of Exchangeability and Latent Variable Models30 Dec 2022
- 2022Finite Sample Identification of Wide Shallow Neural Networks with Biases8 Nov 2022
- 2022Overparameterized ReLU Neural Networks Learn the Simplest Models: Neural Isometry and Exact Recovery30 Sep 2022
- 2022Neural Networks Efficiently Learn Low-Dimensional Representations with SGD29 Sep 2022
- 2022Magnitude and Angle Dynamics in Training Single ReLU Neurons27 Sep 2022
- 2022Local Identifiability of Deep ReLU Neural Networks: the Theory15 Jun 2022
- 2022Demystifying the Global Convergence Puzzle of Learning Over-parameterized ReLU Nets in Very High Dimensions5 Jun 2022
- 2022Excess Risk of Two-Layer ReLU Neural Networks in Teacher-Student Settings and its Superiority to Kernel Methods30 May 2022
- 2022Deep Architecture Connectivity Matters for Its Convergence: A Fine-Grained Analysis11 May 2022
- 2022Explicitising The Implicit Intrepretability of Deep Neural Networks Via Duality1 Mar 2022
- 2022Benign Overfitting in Two-layer Convolutional Neural Networks14 Feb 2022
- 2022Hardness of Noise-Free Learning for Two-Hidden-Layer Neural Networks10 Feb 2022
- 2022How does unlabeled data improve generalization in self-training? A one-hidden-layer theoretical analysis21 Jan 2022
- 2021Parameter identifiability of a deep feedforward ReLU neural network24 Dec 2021
- 2021Efficiently Learning Any One Hidden Layer ReLU Network From Queries8 Nov 2021
- 2021Coordinate Descent Methods for DC Minimization: Optimality Conditions and Global Convergence9 Sep 2021
- 2021How much pre-training is enough to discover a good subnetwork?31 Jul 2021
- 2021Structured Directional Pruning via Perturbation Orthogonal Projection12 Jul 2021
- 2021On Learnability via Gradient Method for Two-Layer ReLU Neural Networks in Teacher-Student Setting11 Jun 2021
- 2021Self-Regularity of Non-Negative Output Weights for Overparameterized Two-Layer Neural Networks2 Mar 2021
- 2021Spurious Local Minima Are Common for Deep Neural Networks with Piecewise Linear Activations25 Feb 2021
- 2021A Local Convergence Theory for Mildly Over-Parameterized Two-Layer Neural Network4 Feb 2021
- 2021Stable Recovery of Entangled Weights: Towards Robust Identification of Deep Neural Networks from Minimal Samples18 Jan 2021
- 2021Towards Searching Efficient and Accurate Neural Network Architectures in Binary Classification Problems16 Jan 2021
- 2021A Convergence Theory Towards Practical Over-parameterized Deep Neural Networks12 Jan 2021
- 2022Benign Overfitting in Two-layer Convolutional Neural Networkscited by 90
- 2022Neural Networks Efficiently Learn Low-Dimensional Representations with SGDcited by 61
- 2023ZiCo: Zero-shot NAS via Inverse Coefficient of Variation on Gradientscited by 57
- 2021A Local Convergence Theory for Mildly Over-Parameterized Two-Layer Neural Networkcited by 47
- 2023Generalization Ability of Wide Neural Networks on ℝcited by 28
- 2022Hardness of Noise-Free Learning for Two-Hidden-Layer Neural Networkscited by 27
- 2023Over-Parameterization Exponentially Slows Down Gradient Descent for Learning a Single Neuroncited by 25
- 2021Parameter identifiability of a deep feedforward ReLU neural networkcited by 24
- 2024Provably learning a multi-head attention layercited by 21
- 2022How does unlabeled data improve generalization in self-training? A one-hidden-layer theoretical analysiscited by 18
- 2021On Learnability via Gradient Method for Two-Layer ReLU Neural Networks in Teacher-Student Settingcited by 16
- 2022An Analysis of Attention via the Lens of Exchangeability and Latent Variable Modelscited by 15
12 most cited of 49 All 49 (JSON)
- 2026Two-Timescale Fine-tuning Provably Learns New Features for Two-Layer ReLU Networks28 Sep 2026
- 2026When Both Layers Learn: Training Dynamics of Representing Linear Models via ReLU Networks3 Jun 2026
- 2026Understanding the Impact of Differentially Private Training on Memorization of Long-Tailed Data1 Feb 2026
- 2025A Teacher-Student Perspective on the Dynamics of Learning Near the Optimal Point17 Dec 2025
- 2025Sobolev acceleration for neural networks24 Sep 2025
- 2024On the Hardness of Learning One Hidden Layer Neural Networks4 Oct 2024
- 2024Iterative thresholding for non-linear learning in the strong varepsilon-contamination model5 Sep 2024
- 2024When Are Bias-Free ReLU Networks Effectively Linear Networks?18 Jun 2024
- 2024Stochastic Bandits with ReLU Neural Networks12 May 2024
- 2024How does promoting the minority fraction affect generalization? A theoretical study of the one-hidden-layer neural network on group imbalance12 Mar 2024
- 2024Towards Robust Out-of-Distribution Generalization Bounds via Sharpness11 Mar 2024
- 2024Provably learning a multi-head attention layer6 Feb 2024
12 newest of 49 All 49 (JSON)