Convergence of Gradient Descent for General Neural Network Architectures Beyond the NTK Regime
Not on the map. Why?
Abstract (arXiv)
Training dynamics is central to understanding neural networks, yet its theoretical analysis remains difficult even for simple architectures and becomes substantially more challenging for general modern architectures. In this paper, we propose a convergence framework for analyzing gradient descent (GD) dynamics under a broad family of neural network architectures and datasets beyond the neural tangent kernel (NTK) regime. The framework is formulated at the level of network blocks and covers architectures including pre-normalized multi-layer transformers. More precisely, under mild assumptions, we prove that for almost all initializations, GD with regular learning rates converges to the neighbourhood of a stationary point. This is mainly proved by establishing an iterate-dependent PL-type inequality through analyticity and measure-zero arguments, and by proving Lipschitz smoothness along the GD trajectory through polynomial generalized smoothness and a local relaxed dissipative condition. We further interpret the theorem under Xavier initialization and practical architectural scaling, showing that the learning rate scale depends on the depth and effective bottleneck dimensions rather than the largest width. Finally, we derive structural nondegeneracy implications for residual connections and function composition, and provide a generic characterization of global minimizers within our framework.
Key facts. Cited by 2 papers in the map (data through 14 Oct 2026). Builds on 17 of its 60 references that are in the map. First posted on arXiv 22 Jun 2026.
Builds on
17 of 60 references.
References · 17
- 2016Layer Normalizationcited by 5,669
- 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,718
- 2019Why gradient clipping accelerates training: A theoretical justification for adaptivitycited by 341
- 2020Towards Understanding Ensemble, Knowledge Distillation and Self-Distillation in Deep Learningcited by 297
- 2020The large learning rate phase of deep learning: the catapult mechanismcited by 216
- 2020Feature Learning in Infinite-Width Neural Networkscited by 190
- 2016Identity Matters in Deep Learningcited by 109
- 2022Benign Overfitting in Two-layer Convolutional Neural Networkscited by 91
- 2019Quadratic Suffices for Over-parametrization via Matrix Chernoff Boundcited by 59
- 2024Residual Connections and Normalization Can Provably Prevent Oversmoothing in GNNscited by 34
- 2023Good regularity creates large learning rate implicit biases: edge of stability, balancing, and catapultcited by 25
- 2022Feature selection with gradient descent on two-layer networks in low-rotation regimescited by 14
5 more
- 2022On Feature Learning in Neural Networks with Global Convergence Guaranteescited by 13
- 2026Deep Delta Learningcited by 9
- 2025On the Convergence of Gradient Descent on Learning Transformers with Residual Connectionscited by 6
- 2025Precise gradient descent training dynamics for finite-width multi-layer neural networkscited by 5
- 2025Global Convergence and Rich Feature Learning in L-Layer Infinite-Width Neural Networks under μP Parametrizationcited by 4
Built on by
Citing papers per year: 2026: 2.
2 papers.