Fast Convergence of Natural Gradient Descent for Overparameterized Neural Networks
Not on the map. Why?
Abstract (arXiv)
Natural gradient descent has proven effective at mitigating the effects of pathological curvature in neural network optimization, but little is known theoretically about its convergence properties, especially for nonlinear networks. In this work, we analyze for the first time the speed of convergence of natural gradient descent on nonlinear neural networks with squared-error loss. We identify two conditions which guarantee efficient convergence from random initializations: (1) the Jacobian matrix (of network's output for all training cases with respect to the parameters) has full row rank, and (2) the Jacobian matrix is stable for small perturbations around the initialization. For two-layer ReLU neural networks, we prove that these two conditions do in fact hold throughout the training, under the assumptions of nondegenerate inputs and overparameterization. We further extend our analysis to more general loss functions. Lastly, we show that K-FAC, an approximate natural gradient descent method, also converges to global minima under the same assumptions, and we give a bound on the rate of this convergence.
Key facts. Cited by 76 papers in the map (data through 14 Oct 2026). First posted on arXiv 27 May 2019.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 18; 2022: 15; 2023: 22; 2024: 9; 2025: 6; 2026: 6.
76 papers.
Sort:
By area:
- 2021The Low-Rank Simplicity Bias in Deep Networkscited by 154
- 2021Improved architectures and training algorithms for deep operator networkscited by 86
- 2023A Distributed Data-Parallel PyTorch Implementation of the Distributed Shampoo Optimizer for Training Neural Networks At-Scalecited by 77
- 2021Training Multi-Layer Over-Parametrized Neural Network in Subquadratic Timecited by 62
- 2023An Over-parameterized Exponential Regressioncited by 45
- 2023Solving Regularized Exp, Cosh and Sinh Regression Problemscited by 44
- 2021Does Preprocessing Help Training Over-parameterized Neural Networks?cited by 32
- 2022Gradient Descent on Neurons and its Link to Approximate Second-Order Optimizationcited by 30
- 2021Credit Assignment in Neural Networks through Deep Feedback Controlcited by 29
- 2022Bypass Exponential Time Preprocessing: Fast Neural Network Training via Weight-Data Correlation Preprocessingcited by 29
- 2023A Mathematical Abstraction for Balancing the Trade-off Between Creativity and Reality in Large Language Modelscited by 29
- 2022Dual Lottery Ticket Hypothesiscited by 27
38 more
- 2023How to Protect Copyright Data in Optimization of Large Language Models?cited by 26
- 2024A Kaczmarz-inspired approach to accelerate the optimization of neural network wavefunctionscited by 26
- 2021Fractal Structure and Generalization Properties of Stochastic Optimization Algorithmscited by 23
- 2022Training Overparametrized Neural Networks in Sublinear Timecited by 22
- 2021Single-Shot Pruning for Offline Reinforcement Learningcited by 21
- 2024The Challenges of the Nonlinear Regime for Physics-Informed Neural Networkscited by 21
- 2021KAISA: An Adaptive Second-Order Optimizer Framework for Deep Neural Networkscited by 20
- 2022Finite-Time Analysis of Entropy-Regularized Neural Natural Actor-Critic Algorithmcited by 20
- 2023Sampling via Gradient Flows in the Space of Probability Measurescited by 20
- 2023The Geometry of Neural Nets' Parameter Spaces Under Reparametrizationcited by 16
- 2023Gradient Flows for Sampling: Mean-Field Models, Gaussian Approximations and Affine Invariancecited by 16
- 2023On the Parameterization of Second-Order Optimization Effective Towards the Infinite Widthcited by 15
- 2024An Improved Empirical Fisher Approximation for Natural Gradient Descentcited by 15
- 2021Second-Order Neural ODE Optimizercited by 11
- 2022Invariance Properties of the Natural Gradient in Overparametrised Systemscited by 11
- 2023Efficient SGD Neural Network Training via Sublinear Activated Neuron Identificationcited by 11
- 2021A proof of convergence for the gradient descent optimization method with random initializations in the training of neural networks with ReLU activation for piecewise linear target functionscited by 10
- 2021Existence, uniqueness, and convergence rates for gradient flows in the training of artificial neural networks with ReLU activationcited by 9
- 2021Convergence proof for stochastic gradient descent in the training of deep neural networks with ReLU activation for constant target functionscited by 9
- 2022Amortized Proximal Optimizationcited by 9
- 2023Controlling the Inductive Bias of Wide Neural Networks by Modifying the Kernel's Spectrumcited by 9
- 2023An Automatic Learning Rate Schedule Algorithm for Achieving Faster Convergence and Steeper Descentcited by 9
- 2024Efficient, Multimodal, and Derivative-Free Bayesian Inference With Fisher-Rao Gradient Flowscited by 9
- 2022Generalization in Supervised Learning Through Riemannian Contractioncited by 8
- 2022A Mini-Block Fisher Method for Deep Neural Networkscited by 8
- 2023Natural Gradient Methods: Perspectives, Efficient-Scalable Approximations, and Analysiscited by 8
- 2024Understanding Stochastic Natural Gradient Variational Inferencecited by 8
- 2021TENGraD: Time-Efficient Natural Gradient Descent with Exact Fisher-Block Inversioncited by 7
- 2022Efficient Natural Gradient Descent Methods for Large-Scale PDE-Based Optimization Problemscited by 7
- 2022Scalable K-FAC Training for Deep Neural Networks with Distributed Preconditioningcited by 7
- 2023Rethinking Gauss-Newton for learning over-parameterized modelscited by 6
- 2023On the Disconnect Between Theory and Practice of Neural Networks: Limits of the NTK Perspectivecited by 6
- 2022Riemannian Natural Gradient Methodscited by 5
- 2023Neural Characteristic Activation Analysis and Geometric Parameterization for ReLU Networkscited by 5
- 2021Generalization Performance of Empirical Risk Minimization on Over-parameterized Deep ReLU Netscited by 4
- 2024TENG: Time-Evolving Natural Gradient for Solving PDEs With Deep Neural Nets Toward Machine Precisioncited by 4
- 2021SCORE: Approximating Curvature Information under Self-Concordant Regularizationcited by 3
- 2022Provably efficient variational generative modeling of quantum many-body systems via quantum-probabilistic information geometrycited by 3
50 most cited of 76 All 76 (JSON)
- 2026A Gradient Flow Perspective on Minimum MMD Estimation4 Jul 2026
- 2026Canonical Regularisation of Wide Feature-Learning Neural Networks18 May 2026
- 2026Understanding Sample Efficiency in Predictive Coding12 May 2026
- 2026Convergence Analysis of Newton's Method for Neural Networks in the Overparameterized Limit8 May 2026
- 2026Descent-Guided Policy Gradient for Scalable Cooperative Multi-Agent Learning23 Feb 2026
- 2026Gradient Regularized Natural Gradients26 Jan 2026
- 2025A Unified Perspective on Optimization in Machine Learning and Neuroscience: From Gradient Descent to Neural Adaptation21 Oct 2025
- 2025MAC: An Efficient Gradient Preconditioning using Mean Activation Approximated Curvature10 Jun 2025
- 2025Sharper Convergence Rates for Nonconvex Optimisation via Reduction Mappings10 Jun 2025
- 2025pFedSOP : Accelerating Training Of Personalized Federated Learning Using Second-Order Optimization8 Jun 2025
- 2025A Flat Minima Perspective on Understanding Augmentations and Model Robustness30 May 2025
- 2025Convergence Properties of Natural Gradient Descent for Minimizing KL Divergence27 Apr 2025
38 more
- 2024A Riemannian Optimization Perspective of the Gauss-Newton Method for Feedforward Neural Networks18 Dec 2024
- 2024Efficient, Multimodal, and Derivative-Free Bayesian Inference With Fisher-Rao Gradient Flows25 Jun 2024
- 2024An Improved Empirical Fisher Approximation for Natural Gradient Descent10 Jun 2024
- 2024Understanding Stochastic Natural Gradient Variational Inference4 Jun 2024
- 2024Regularized Gauss-Newton for Optimizing Overparameterized Neural Networks23 Apr 2024
- 2024TENG: Time-Evolving Natural Gradient for Solving PDEs With Deep Neural Nets Toward Machine Precision16 Apr 2024
- 2024Inverse-Free Fast Natural Gradient Descent Method for Deep Learning6 Mar 2024
- 2024The Challenges of the Nonlinear Regime for Physics-Informed Neural Networks6 Feb 2024
- 2024A Kaczmarz-inspired approach to accelerate the optimization of neural network wavefunctions18 Jan 2024
- 2023Exploiting hidden structures in non-convex games for convergence to Nash equilibrium27 Dec 2023
- 2023On the Parameterization of Second-Order Optimization Effective Towards the Infinite Width19 Dec 2023
- 2023Gathering and Exploiting Higher-Order Information when Training Large Structured Models6 Dec 2023
- 2023DePAint: A Decentralized Safe Multi-Agent Reinforcement Learning Algorithm considering Peak and Average Constraints22 Oct 2023
- 2023An Automatic Learning Rate Schedule Algorithm for Achieving Faster Convergence and Steeper Descent17 Oct 2023
- 2023Sampling via Gradient Flows in the Space of Probability Measures5 Oct 2023
- 2023On the Disconnect Between Theory and Practice of Neural Networks: Limits of the NTK Perspective29 Sep 2023
- 2023A Distributed Data-Parallel PyTorch Implementation of the Distributed Shampoo Optimizer for Training Neural Networks At-Scale12 Sep 2023
- 2023How to Protect Copyright Data in Optimization of Large Language Models?23 Aug 2023
- 2023Controlling the Inductive Bias of Wide Neural Networks by Modifying the Kernel's Spectrum26 Jul 2023
- 2023Efficient SGD Neural Network Training via Sublinear Activated Neuron Identification13 Jul 2023
- 2023A Mathematical Abstraction for Balancing the Trade-off Between Creativity and Reality in Large Language Models4 Jun 2023
- 2023Neural Characteristic Activation Analysis and Geometric Parameterization for ReLU Networks25 May 2023
- 2023Deep Metric Tensor Regularized Policy Gradient18 May 2023
- 2023An Over-parameterized Exponential Regression29 Mar 2023
- 2023Solving Regularized Exp, Cosh and Sinh Regression Problems28 Mar 2023
- 2023Natural Gradient Methods: Perspectives, Efficient-Scalable Approximations, and Analysis6 Mar 2023
- 2023Natural Gradient Hybrid Variational Inference with Application to Deep Mixed Models27 Feb 2023
- 2023Gauss-Newton Temporal Difference Learning with Nonlinear Function Approximation25 Feb 2023
- 2023Gradient Flows for Sampling: Mean-Field Models, Gaussian Approximations and Affine Invariance21 Feb 2023
- 2023The Geometry of Neural Nets' Parameter Spaces Under Reparametrization14 Feb 2023
- 2023Rethinking Gauss-Newton for learning over-parameterized models6 Feb 2023
- 2022Bypass Exponential Time Preprocessing: Fast Neural Network Training via Weight-Data Correlation Preprocessing25 Nov 2022
- 2022Component-Wise Natural Gradient Descent -- An Efficient Neural Network Optimization11 Oct 2022
- 2022Training Overparametrized Neural Networks in Sublinear Time9 Aug 2022
- 2022Riemannian Natural Gradient Methods15 Jul 2022
- 2022Invariance Properties of the Natural Gradient in Overparametrised Systems30 Jun 2022
- 2022Scalable K-FAC Training for Deep Neural Networks with Distributed Preconditioning30 Jun 2022
- 2022Provably efficient variational generative modeling of quantum many-body systems via quantum-probabilistic information geometry9 Jun 2022
50 newest of 76 All 76 (JSON)
- 2021Single-Shot Pruning for Offline Reinforcement Learningcited by 21
- 2021Single-Shot Pruning for Offline Reinforcement Learning31 Dec 2021
- 2021The Low-Rank Simplicity Bias in Deep Networkscited by 154
- 2021Improved architectures and training algorithms for deep operator networkscited by 86
- 2023A Distributed Data-Parallel PyTorch Implementation of the Distributed Shampoo Optimizer for Training Neural Networks At-Scalecited by 77
- 2021Training Multi-Layer Over-Parametrized Neural Network in Subquadratic Timecited by 62
- 2023An Over-parameterized Exponential Regressioncited by 45
- 2023Solving Regularized Exp, Cosh and Sinh Regression Problemscited by 44
- 2021Does Preprocessing Help Training Over-parameterized Neural Networks?cited by 32
- 2022Gradient Descent on Neurons and its Link to Approximate Second-Order Optimizationcited by 30
- 2021Credit Assignment in Neural Networks through Deep Feedback Controlcited by 29
- 2022Bypass Exponential Time Preprocessing: Fast Neural Network Training via Weight-Data Correlation Preprocessingcited by 29
- 2023A Mathematical Abstraction for Balancing the Trade-off Between Creativity and Reality in Large Language Modelscited by 29
- 2022Dual Lottery Ticket Hypothesiscited by 27
12 most cited of 75 All 75 (JSON)
- 2026A Gradient Flow Perspective on Minimum MMD Estimation4 Jul 2026
- 2026Canonical Regularisation of Wide Feature-Learning Neural Networks18 May 2026
- 2026Understanding Sample Efficiency in Predictive Coding12 May 2026
- 2026Convergence Analysis of Newton's Method for Neural Networks in the Overparameterized Limit8 May 2026
- 2026Descent-Guided Policy Gradient for Scalable Cooperative Multi-Agent Learning23 Feb 2026
- 2026Gradient Regularized Natural Gradients26 Jan 2026
- 2025A Unified Perspective on Optimization in Machine Learning and Neuroscience: From Gradient Descent to Neural Adaptation21 Oct 2025
- 2025MAC: An Efficient Gradient Preconditioning using Mean Activation Approximated Curvature10 Jun 2025
- 2025Sharper Convergence Rates for Nonconvex Optimisation via Reduction Mappings10 Jun 2025
- 2025pFedSOP : Accelerating Training Of Personalized Federated Learning Using Second-Order Optimization8 Jun 2025
- 2025A Flat Minima Perspective on Understanding Augmentations and Model Robustness30 May 2025
- 2025Convergence Properties of Natural Gradient Descent for Minimizing KL Divergence27 Apr 2025
12 newest of 75 All 75 (JSON)