Convergence of Gradient Descent for Recurrent Neural Networks: A Nonasymptotic Analysis
Not on the map. Why?
Abstract (arXiv)
We analyze recurrent neural networks with diagonal hidden-to-hidden weight matrices, trained with gradient descent in the supervised learning setting, and prove that gradient descent can achieve optimality without massive overparameterization. Our in-depth nonasymptotic analysis (i) provides improved bounds on the network size m in terms of the sequence length T, sample size n and ambient dimension d, and (ii) identifies the significant impact of long-term dependencies in the dynamical system on the convergence and network width bounds characterized by a cutoff point that depends on the Lipschitz continuity of the activation function. Remarkably, this analysis reveals that an appropriately-initialized recurrent neural network trained with n samples can achieve optimality with a network size m that scales only logarithmically with n. This sharply contrasts with the prior works that require high-order polynomial dependency of m on n to establish strong regularity conditions. Our results are based on an explicit characterization of the class of dynamical systems that can be approximated and learned by recurrent neural networks via norm-constrained transportation mappings, and establishing local smoothness properties of the hidden state with respect to the learnable parameters.
Key facts. Cited by 7 papers in the map (data through 7 Oct 2026). Builds on 24 of its 43 references that are in the map. First posted on arXiv 19 Feb 2024.
Builds on
24 of 43 references.
References · 24
- 2014Learning Phrase Representations using RNN Encoder-Decoder for Statistical Machine Translationcited by 4,016
- 2018Neural Tangent Kernel: Convergence and Generalization in Neural Networkscited by 2,597
- 2013An Empirical Investigation of Catastrophic Forgetting in Gradient-Based Neural Networkscited by 711
- 2018On Lazy Training in Differentiable Programmingcited by 671
- 2013Speech Recognition with Deep Recurrent Neural Networkscited by 666
- 2018Gradient Descent Finds Global Minima of Deep Neural Networkscited by 572
- 2013Generating Sequences With Recurrent Neural Networkscited by 549
- 2014Recurrent Models of Visual Attentioncited by 344
- 2020Kernel and Rich Regimes in Overparametrized Modelscited by 302
- 2013Maxout Networkscited by 186
- 2019Towards moderate overparameterization: global convergence guarantees for training shallow neural networkscited by 176
- 2019Polylogarithmic width suffices for gradient descent to achieve arbitrarily small test error with shallow ReLU networkscited by 120
12 more
- 2018On the Convergence Rate of Training Recurrent Neural Networkscited by 82
- 2016Gradient Descent Learns Linear Dynamical Systemscited by 78
- 2020The Recurrent Neural Tangent Kernelcited by 67
- 2013How to Construct Deep Recurrent Neural Networkscited by 64
- 2020Approximation Bounds for Random Neural Networks and Reservoir Systemscited by 52
- 2019Neural tangent kernels, transportation mappings, and universal approximationcited by 30
- 2019Can SGD Learn Recurrent Neural Networks with Provable Generalization?cited by 17
- 2023Robust Implicit Regularization via Weight Normalizationcited by 10
- 2021Sample Complexity and Overparameterization Bounds for Temporal Difference Learning with Neural Network Approximationcited by 9
- 2022Recurrent Neural Networks and Universal Approximation of Bayesian Filterscited by 5
- 2023Kernel Limit for a Class of Recurrent Neural Networks Trained on Ergodic Data Sequencescited by 2
- 2021Path classification by stochastic linear recurrent neural networkscited by 1
Built on by
Citing papers per year: 2024: 2; 2025: 3; 2026: 2.
7 papers.
- 2024Recurrent Natural Policy Gradient for POMDPscited by 8
- 2025A Theoretical Justification for Asymmetric Actor-Critic Algorithmscited by 5
- 2024A Riemannian Optimization Perspective of the Gauss-Newton Method for Feedforward Neural Networkscited by 2
- 2025Convergence of Stochastic Gradient Langevin Dynamics in the Lazy Training Regimecited by 0
- 2025Non-Asymptotic Optimization and Generalization Bounds for Stochastic Gauss-Newton in Overparameterized Modelscited by 0
- 2026Finite-Time Analysis of Gradient Descent for Shallow Transformerscited by 0
- 2026Learning Suffers More Than the Policy Class Under Partial Observability: A Closed-Form Analysiscited by 0
- 2026Learning Suffers More Than the Policy Class Under Partial Observability: A Closed-Form Analysis7 Aug 2026
- 2026Finite-Time Analysis of Gradient Descent for Shallow Transformers23 Jan 2026
- 2025Non-Asymptotic Optimization and Generalization Bounds for Stochastic Gauss-Newton in Overparameterized Models6 Nov 2025
- 2025Convergence of Stochastic Gradient Langevin Dynamics in the Lazy Training Regime24 Oct 2025
- 2025A Theoretical Justification for Asymmetric Actor-Critic Algorithms31 Jan 2025
- 2024A Riemannian Optimization Perspective of the Gauss-Newton Method for Feedforward Neural Networks18 Dec 2024
- 2024Recurrent Natural Policy Gradient for POMDPs28 May 2024