Label-NTK Alignments and A Tighter Convergence Bound in the NTK Regime
Not on the map. Why?
Abstract (arXiv)
The Neural Tangent Kernel (NTK) framework explains optimization in over-parameterized neural networks via approximately linearized dynamics, yielding exponential convergence guarantees. However, existing results are often overly pessimistic and do not match the fast training in practice, as they depend on the smallest NTK eigenvalue, which is typically extremely small in practice. In this work, we develop sharper convergence guarantees by characterizing the interaction between data labels and the NTK eigen-spectrum. We identify two key phenomena, Label-NTK alignment and Residual-NTK alignment, showing that projections of labels and residuals onto NTK eigenvectors scale with the corresponding eigenvalues. We provide empirical evidence and theoretical justification under mild data assumptions. Exploiting these alignment properties, we derive a refined convergence bound that depends on the full spectrum and closely matches practical training dynamics, significantly improving over classical worst-case results. We further obtain improved generalization bounds. Experiments on MLPs and CNNs across multiple datasets validate our theory.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 28 of its 33 references that are in the map. First posted on arXiv 24 May 2026.
Builds on
28 of 33 references.
References · 28
- 2018Neural Tangent Kernel: Convergence and Generalization in Neural Networkscited by 2,621
- 2018A Convergence Theory for Deep Learning via Over-Parameterizationcited by 787
- 2019Wide Neural Networks of Any Depth Evolve as Linear Models Under Gradient Descentcited by 738
- 2018On Lazy Training in Differentiable Programmingcited by 675
- 2018Gradient Descent Provably Optimizes Over-parameterized Neural Networkscited by 659
- 2018Gradient Descent Finds Global Minima of Deep Neural Networkscited by 574
- 2019Fine-Grained Analysis of Optimization and Generalization for Overparameterized Two-Layer Neural Networkscited by 566
- 2021Gradient Descent on Neural Networks Typically Occurs at the Edge of Stabilitycited by 427
- 2020Loss landscapes and optimization in over-parameterized non-linear systems and neural networkscited by 233
- 2020The large learning rate phase of deep learning: the catapult mechanismcited by 216
- 2020Spectrum Dependent Learning Curves in Kernel Regression and Wide Neural Networkscited by 200
- 2019Towards moderate overparameterization: global convergence guarantees for training shallow neural networkscited by 176
16 more
- 2020Spectral Bias and Task-Model Alignment Explain Generalization in Kernel Regression and Infinitely Wide Neural Networkscited by 166
- 2020Frequency Bias in Neural Networks for Input of Non-Uniform Densitycited by 140
- 2019The Convergence Rate of Neural Networks for Learned Functions of Different Frequenciescited by 135
- 2020On the linearity of large non-linear models: when and why the tangent kernel is constantcited by 129
- 2019An Improved Analysis of Training Over-parameterized Deep Neural Networkscited by 121
- 2019Polylogarithmic width suffices for gradient descent to achieve arbitrarily small test error with shallow ReLU networkscited by 121
- 2021Neural Networks as Kernel Learners: The Silent Alignment Effectcited by 106
- 2020On the Similarity between the Laplace and Neural Tangent Kernelscited by 76
- 2020Global Convergence of Deep Networks with One Wide Layer Followed by Pyramidal Topologycited by 66
- 2021A Theory of Neural Tangent Kernel Alignment and Its Influence on Trainingcited by 25
- 2022Characterizing the Spectrum of the NTK via a Power Series Expansioncited by 22
- 2023Aiming towards the minimizers: fast convergence of SGD for overparametrized problemscited by 22
- 2022Transition to Linearity of Wide Neural Networks is an Emerging Property of Assembling Weak Modelscited by 6
- 2023Better NTK Conditioning: A Free Lunch from (ReLU) Nonlinear Activation in Wide Neural Networkscited by 6
- 2025Understanding the Evolution of the Neural Tangent Kernel at the Edge of Stabilitycited by 6
- 2022Transition to Linearity of General Neural Networks with Directed Acyclic Graph Architecturecited by 4
Built on by
Not cited yet.