The Global Empirical NTK: Self-Referential Bias and Dimensionality of Gradient Descent Learning
Not on the map. Why?
Abstract (arXiv)
In training a neural network with gradient descent (GD), each iteration induces a linear operator that governs first-order updates to a model's internal state variables. We define this operator as the Global Empirical Neural Tangent Kernel (NTK). In finite-width networks, the NTK is typically intractable to form, leading prior work to focus on restrictive settings such as tracking outputs only or taking infinite-width limits. Here, we study the structure of the NTK for a range of models. Formulating the model state as the solution to a single global implicit constraint, we derive the NTK as a product of two operators: K, accounting for immediate parameter-to-state interactions, and P, describing internal state-to-state dependencies. For a broad class of weight-based models, including RNNs and transformers, we prove a universal Kronecker-core theorem showing that K admits an exact, computable form given by the Gram matrix of weight-site variables. This core structure reveals that the NTK is structurally bottlenecked, constraining its effective rank and giving rise to a self-referential bias whereby GD preferentially learns within dominant modes of joint hidden and input activity. For recurrent models, we examine the spectrum of the NTK and show when it is biased and low-rank in space or time under the proposed decomposition. We further demonstrate that model dynamics at initialization bias the NTK, restricting learning and preventing task components from being learned effectively. Finally, we show that the NTK associated with a self-attention transformer is likewise structurally constrained to be low-rank. Overall, we show that the NTK possesses tractable structure that explains GD bias toward task solutions and the emergence of low-rank representations. To enable use of the NTK as a practical metric, we build kpflow, a library relying on randomized matrix-free numerical linear algebra.
Key facts. Cited by 2 papers in the map (data through 14 Oct 2026). Builds on 37 of its 73 references that are in the map. First posted on arXiv 9 May 2026.
Builds on
37 of 73 references.
References · 37
- 2018Neural Ordinary Differential Equationscited by 3,779
- 2018Neural Tangent Kernel: Convergence and Generalization in Neural Networkscited by 2,621
- 2020Fourier Features Let Networks Learn High Frequency Functions in Low Dimensional Domainscited by 2,200
- 2018On the Spectral Bias of Neural Networkscited by 1,341
- 2015Optimizing Neural Networks with Kronecker-factored Approximate Curvaturecited by 796
- 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
- 2019Deep Equilibrium Modelscited by 615
- 2014New insights and perspectives on the natural gradient methodcited by 443
- 2018Shampoo: Preconditioned Stochastic Tensor Optimizationcited by 426
- 2020Spectrum Dependent Learning Curves in Kernel Regression and Wide Neural Networkscited by 200
25 more
- 2019Mean-field theory of two-layers neural networks: dimension-free bounds and kernel limitcited by 187
- 2019Towards Understanding the Spectral Bias of Deep Learningcited by 174
- 2020Tensor Programs II: Neural Tangent Kernel for Any Architecturecited by 143
- 2016A Kronecker-factored approximate Fisher matrix for convolution layerscited by 141
- 2019Finite Depth and Width Corrections to the Neural Tangent Kernelcited by 115
- 2020Spectra of the Conjugate Kernel and Neural Tangent Kernel for linear-width neural networkscited by 77
- 2020The Recurrent Neural Tangent Kernelcited by 67
- 2020Hutch++: Optimal Stochastic Trace Estimationcited by 63
- 2022Fast Finite Width Neural Tangent Kernelcited by 62
- 2024How Feature Learning Can Improve Neural Scaling Lawscited by 56
- 2017Linking connectivity, dynamics and computations in low-rank recurrent neural networkscited by 51
- 2023Dynamics of Finite Width Kernel and Prediction Fluctuations in Mean Field Neural Networkscited by 43
- 2020Lyapunov spectra of chaotic recurrent neural networkscited by 37
- 2022A Fast, Well-Founded Approximation to the Empirical Neural Tangent Kernelcited by 30
- 2021A Theory of Neural Tangent Kernel Alignment and Its Influence on Trainingcited by 25
- 2020The interplay between randomness and structure during learning in RNNscited by 20
- 2020A Practical Sparse Approximation for Real Time Recurrent Learningcited by 18
- 2023Low Tensor Rank Learning of Neural Dynamicscited by 12
- 2023On the Neural Tangent Kernel of Equilibrium Modelscited by 7
- 2023The computational role of structure in neural activity and connectivitycited by 6
- 2025Dynamics of learning to integrate in linear recurrent neural networkscited by 4
- 2025Discovering alternative solutions beyond the simplicity bias in recurrent neural networkscited by 3
- 2025Fast Generalized Neural Tangent Kernel Statistics via Trace Estimationcited by 3
- 2025Neural Tangent Knowledge Distillation for Optical Convolutional Networkscited by 2
- 2025KPFlow: An Operator Perspective on Dynamic Collapse Under Gradient Descent Training of Recurrent Networkscited by 1
Built on by
Citing papers per year: 2026: 2.
2 papers.