On the linearity of large non-linear models: when and why the tangent kernel is constant
Not on the map. Why?
Abstract (arXiv)
The goal of this work is to shed light on the remarkable phenomenon of transition to linearity of certain neural networks as their width approaches infinity. We show that the transition to linearity of the model and, equivalently, constancy of the (neural) tangent kernel (NTK) result from the scaling properties of the norm of the Hessian matrix of the network as a function of the network width. We present a general framework for understanding the constancy of the tangent kernel via Hessian scaling applicable to the standard classes of neural networks. Our analysis provides a new perspective on the phenomenon of constant tangent kernel, which is different from the widely accepted "lazy training". Furthermore, we show that the transition to linearity is not a general property of wide neural networks and does not hold when the last layer of the network is non-linear. It is also not necessary for successful optimization by gradient descent.
Key facts. Cited by 129 papers in the map (data through 7 Oct 2026). First posted on arXiv 2 Oct 2020.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 20; 2022: 30; 2023: 22; 2024: 29; 2025: 15; 2026: 13.
129 papers.
Sort:
By area:
- 2022How Do Vision Transformers Work?cited by 305
- 2021Fit without fear: remarkable mathematical phenomena of deep learning through the prism of interpolationcited by 130
- 2024Challenges in Training PINNs: A Loss Landscape Perspectivecited by 109
- 2021Neural Networks as Kernel Learners: The Silent Alignment Effectcited by 106
- 2024Simplifying Deep Temporal Difference Learningcited by 71
- 2024Scalable Optimization in the Modular Normcited by 64
- 2021Training Multi-Layer Over-Parametrized Neural Network in Subquadratic Timecited by 62
- 2021Score-based Generative Neural Networks for Large-Scale Optimal Transportcited by 59
- 2021Optimization of Graph Neural Networks: Implicit Acceleration by Skip Connections and More Depthcited by 55
- 2021What can linearized neural networks actually say about generalization?cited by 49
- 2024Scaling and renormalization in high-dimensional regressioncited by 45
- 2021Early Stopping for Deep Image Priorcited by 44
38 more
- 2021Random Neural Networks in the Infinite Width Limit as Gaussian Processescited by 43
- 2022Graph Neural Networks are Inherently Good Generalizers: Insights by Bridging GNNs and MLPscited by 42
- 2024Numerical analysis of physics-informed neural networks and related models in physics-informed machine learningcited by 39
- 2023On the Optimization and Generalization of Multi-head Attentioncited by 37
- 2023Catapults in SGD: spikes in the training loss and their impact on generalization through feature learningcited by 29
- 2021Analyzing Monotonic Linear Interpolation in Neural Network Loss Landscapescited by 27
- 2022Do Residual Neural Networks discretize Neural Ordinary Differential Equations?cited by 27
- 2023An operator preconditioning perspective on training in physics-informed machine learningcited by 27
- 2021Subquadratic Overparameterization for Shallow Neural Networkscited by 26
- 2021Stability & Generalisation of Gradient Descent for Shallow Neural Networks without the Neural Tangent Kernelcited by 25
- 2022On the Generalization Mystery in Deep Learningcited by 25
- 2021Overparameterization of deep ResNet: zero loss and mean-field analysiscited by 24
- 2022Mechanism of feature learning in deep fully connected networks and kernel machines that recursively learn featurescited by 24
- 2023Aiming towards the minimizers: fast convergence of SGD for overparametrized problemscited by 22
- 2021A Neural Tangent Kernel Perspective of GANscited by 21
- 2021Uniform Generalization Bounds for Overparameterized Neural Networkscited by 21
- 2022What Can the Neural Tangent Kernel Tell Us About Adversarial Robustness?cited by 21
- 2023Unifying GANs and Score-Based Diffusion as Generative Particle Modelscited by 21
- 2024The Challenges of the Nonlinear Regime for Physics-Informed Neural Networkscited by 21
- 2026There Will Be a Scientific Theory of Deep Learningcited by 21
- 2023Six Lectures on Linearized Neural Networkscited by 19
- 2022Demystifying the Neural Tangent Kernel from a Practical Perspective: Can it be trusted for Neural Architecture Search without training?cited by 18
- 2021On the Equivalence between Neural Network and Support Vector Machinecited by 17
- 2022Quadratic models for understanding catapult dynamics of neural networkscited by 17
- 2024Reparameterization invariance in approximate Bayesian inferencecited by 17
- 2021Learning with Gradient Descent and Weakly Convex Lossescited by 16
- 2023Graph Neural Tangent Kernel: Convergence on Large Graphscited by 16
- 2023On the Convergence of Encoder-only Shallow Transformerscited by 13
- 2024Bounds for the smallest eigenvalue of the NTK for arbitrary spherical data of arbitrary dimensioncited by 13
- 2022Wide and Deep Neural Networks Achieve Optimality for Classificationcited by 12
- 2022Spectral Bias Outside the Training Set for Deep Networks in the Kernel Regimecited by 11
- 2023Non-Parametric Representation Learning with Kernelscited by 11
- 2021On the Global Convergence of Gradient Descent for multi-layer ResNets in the mean-field regimecited by 10
- 2022Implicit Bias of MSE Gradient Optimization in Underparameterized Neural Networkscited by 9
- 2024Deep Learning Through A Telescoping Lens: A Simple Model Provides Empirical Insights On Grokking, Gradient Boosting & Beyondcited by 9
- 2022Kernel Methods and Multi-layer Perceptrons Learn Linear Models in High Dimensionscited by 8
- 2024Peer-to-Peer Learning Dynamics of Wide Neural Networkscited by 8
- 2025Enhancing Physics-Informed Neural Networks Through Feature Engineeringcited by 8
50 most cited of 129 All 129 (JSON)
- 2026Relative Generalization Invariance of LLM Pretraining26 Sep 2026
- 2026Different Statistical Perspectives for Understanding Generalisation in Graph Neural Networks25 May 2026
- 2026Label-NTK Alignments and A Tighter Convergence Bound in the NTK Regime24 May 2026
- 2026The Devil is in the Condition Numbers: Why is GLU Better than non-GLU Structure?20 May 2026
- 2026Training Infinitely Deep and Wide Transformers17 May 2026
- 2026The Neural Tangent Kernel for Classification17 May 2026
- 2026On the global convergence of gradient flow for wide shallow models beyond homogeneous nonlinearities11 May 2026
- 2026Implicit Target Shift in Online Learning: Characterization and Correction8 May 2026
- 2026There Will Be a Scientific Theory of Deep Learning23 Apr 2026
- 2026Exact Certification of Neural Networks and Partition Aggregation Ensembles against Label Poisoning13 Apr 2026
- 2026Towards Initialization-dependent and Non-vacuous Generalization Bounds for Overparameterized Shallow Neural Networks1 Apr 2026
- 2026Unveiling the Mechanism of Continuous Representation Full-Waveform Inversion: A Wave Based Neural Tangent Kernel Framework23 Mar 2026
38 more
- 2026Model Evolution Under Zeroth-Order Optimization: A Neural Tangent Kernel Perspective22 Mar 2026
- 2025Revisiting the Neural Tangent Kernel: the role of large width and depth10 Nov 2025
- 2025A Communication-Efficient Decentralized Actor-Critic Algorithm22 Oct 2025
- 2025Optimal Rates for Generalization of Gradient Descent for Deep ReLU Classification3 Oct 2025
- 2025Low Rank Gradients and Where to Find Them1 Oct 2025
- 2025Theoretical Foundations of Representation Learning using Unlabeled Data: Statistics and Optimization23 Sep 2025
- 2025Objective Value Change and Shape-Based Accelerated Optimization for the Neural Network Approximation27 Aug 2025
- 2025Beyond Scaling Curves: Internal Dynamics of Neural Networks Through the NTK Lens7 Jul 2025
- 2025Thompson Sampling in Function Spaces via Neural Operators27 Jun 2025
- 2025Universal Value-Function Uncertainties27 May 2025
- 2025Is Supervised Learning Really That Different from Unsupervised?16 May 2025
- 2025Ultra-fast feature learning for the training of two-layer neural networks in the two-timescale regime25 Apr 2025
- 2025Divergence of Empirical Neural Tangent Kernel in Classification Problems15 Apr 2025
- 2025Enhancing Physics-Informed Neural Networks Through Feature Engineering11 Feb 2025
- 2025A Gaussian Process View on Observation Noise and Initialization in Wide Neural Networks3 Feb 2025
- 2025Globally Convergent Variational Inference14 Jan 2025
- 2024Exact Certification of (Graph) Neural Networks Against Label Poisoning30 Nov 2024
- 2024Infinite Width Limits of Self Supervised Neural Networks17 Nov 2024
- 2024Deep Learning Through A Telescoping Lens: A Simple Model Provides Empirical Insights On Grokking, Gradient Boosting & Beyond31 Oct 2024
- 2024Sharper Guarantees for Learning Neural Network Classifiers with Gradient Methods13 Oct 2024
- 2024NTK-DFL: Enhancing Decentralized Federated Learning in Heterogeneous Settings via Neural Tangent Kernel2 Oct 2024
- 2024Peer-to-Peer Learning Dynamics of Wide Neural Networks23 Sep 2024
- 2024Unrolled denoising networks provably learn optimal Bayesian inference19 Sep 2024
- 2024Variational Search Distributions10 Sep 2024
- 2024Continual learning with the neural tangent ensemble30 Aug 2024
- 2024Provable Robustness of (Graph) Neural Networks Against Data Poisoning and Backdoor Attacks15 Jul 2024
- 2024Simplifying Deep Temporal Difference Learning5 Jul 2024
- 2024Accelerated Stochastic Min-Max Optimization Based on Bias-corrected Momentum18 Jun 2024
- 2024Loss Gradient Gaussian Width based Generalization and Optimization Guarantees11 Jun 2024
- 2024Reparameterization invariance in approximate Bayesian inference5 Jun 2024
- 2024Scalable Optimization in the Modular Norm23 May 2024
- 2024Bounds for the smallest eigenvalue of the NTK for arbitrary spherical data of arbitrary dimension23 May 2024
- 2024Error Analysis of Three-Layer Neural Network Trained with PGD for Deep Ritz Method19 May 2024
- 2024An Improved Finite-time Analysis of Temporal Difference Learning with Deep Neural Networks7 May 2024
- 2024Physics-Informed Neural Networks: Minimizing Residual Loss with Wide Networks and Effective Activations2 May 2024
- 2024Scaling and renormalization in high-dimensional regression1 May 2024
- 2024The lazy (NTK) and rich (μP) regimes: a gentle tutorial30 Apr 2024
- 2024Understanding the training of infinitely deep and wide ResNets with Conditional Optimal Transport19 Mar 2024
50 newest of 129 All 129 (JSON)
- 2025Universal Value-Function Uncertaintiescited by 1
- 2025Universal Value-Function Uncertainties27 May 2025
- 2022How Do Vision Transformers Work?cited by 305
- 2021Fit without fear: remarkable mathematical phenomena of deep learning through the prism of interpolationcited by 130
- 2024Challenges in Training PINNs: A Loss Landscape Perspectivecited by 109
- 2021Neural Networks as Kernel Learners: The Silent Alignment Effectcited by 106
- 2024Simplifying Deep Temporal Difference Learningcited by 71
- 2024Scalable Optimization in the Modular Normcited by 64
- 2021Training Multi-Layer Over-Parametrized Neural Network in Subquadratic Timecited by 62
- 2021Score-based Generative Neural Networks for Large-Scale Optimal Transportcited by 59
- 2021Optimization of Graph Neural Networks: Implicit Acceleration by Skip Connections and More Depthcited by 55
- 2021What can linearized neural networks actually say about generalization?cited by 49
- 2024Scaling and renormalization in high-dimensional regressioncited by 45
- 2021Early Stopping for Deep Image Priorcited by 44
12 most cited of 127 All 127 (JSON)
- 2026Relative Generalization Invariance of LLM Pretraining26 Sep 2026
- 2026Different Statistical Perspectives for Understanding Generalisation in Graph Neural Networks25 May 2026
- 2026Label-NTK Alignments and A Tighter Convergence Bound in the NTK Regime24 May 2026
- 2026The Devil is in the Condition Numbers: Why is GLU Better than non-GLU Structure?20 May 2026
- 2026Training Infinitely Deep and Wide Transformers17 May 2026
- 2026The Neural Tangent Kernel for Classification17 May 2026
- 2026On the global convergence of gradient flow for wide shallow models beyond homogeneous nonlinearities11 May 2026
- 2026Implicit Target Shift in Online Learning: Characterization and Correction8 May 2026
- 2026There Will Be a Scientific Theory of Deep Learning23 Apr 2026
- 2026Exact Certification of Neural Networks and Partition Aggregation Ensembles against Label Poisoning13 Apr 2026
- 2026Towards Initialization-dependent and Non-vacuous Generalization Bounds for Overparameterized Shallow Neural Networks1 Apr 2026
- 2026Unveiling the Mechanism of Continuous Representation Full-Waveform Inversion: A Wave Based Neural Tangent Kernel Framework23 Mar 2026
12 newest of 127 All 127 (JSON)