Linearized Attention Cannot Enter the Kernel Regime at Any Practical Width
Not on the map. Why?
Abstract (arXiv)
Understanding whether attention mechanisms converge to the kernel regime is foundational to the validity of influence functions for transformer accountability. Exact NTK characterization of softmax attention is precluded by its exponential nonlinearity; linearized attention is the canonical tractable proxy and the object of study here. This paper establishes that even this proxy does not converge to its NTK limit at any practical width, revealing a fundamental trade-off in the learning dynamics of attention. An exact correspondence is established between parameter-free linearized attention and a data-dependent Gram-induced kernel; spectral amplification analysis shows that the attention transformation cubes the Gram matrix's condition number, requiring width m = Omega(kappad(G)6 nlog n) for NTK convergence, where kappad(G) is the effective condition number of the rank-min(n,d) truncation of the input Gram matrix; for natural image datasets this threshold is physically infeasible (m gg 1024 for MNIST and m gg 1029 for CIFAR-10, 12--17 orders of magnitude beyond the largest known architectures). Influence malleability is introduced to characterize this non-convergence: linearized attention exhibits 2--9× higher malleability than ReLU networks under adversarial data perturbation, with the gap depending on dataset condition number and task setting. A dual implication is established: the same data-dependent kernel is shown theoretically to reduce approximation error when targets align with the data geometry, while, empirically, creating vulnerability to adversarial manipulation of the training data. The structural argument extends to trainable QKV attention under standard initialization, with direct consequences for influence methods applied to deployed transformer architectures.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 23 of its 30 references that are in the map. First posted on arXiv 13 Mar 2026.
Builds on
23 of 30 references.
References · 23
- 2017Attention Is All You Needcited by 59,932
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
- 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,616
- 2018Neural Tangent Kernel: Convergence and Generalization in Neural Networkscited by 2,597
- 2020Transformers are RNNs: Fast Autoregressive Transformers with Linear Attentioncited by 1,979
- 2017Understanding Black-box Predictions via Influence Functionscited by 1,714
- 2020Rethinking Attention with Performerscited by 1,584
- 2019Wide Neural Networks of Any Depth Evolve as Linear Models Under Gradient Descentcited by 732
- 2018On Lazy Training in Differentiable Programmingcited by 671
- 2018Gradient Descent Finds Global Minima of Deep Neural Networkscited by 572
- 2019On Exact Computation with an Infinitely Wide Neural Netcited by 572
- 2023Studying Large Language Model Generalization with Influence Functionscited by 311
11 more
- 2023TRAK: Attributing Model Behavior at Scalecited by 273
- 2020Spectrum Dependent Learning Curves in Kernel Regression and Wide Neural Networkscited by 200
- 2020Spectral Bias and Task-Model Alignment Explain Generalization in Kernel Regression and Infinitely Wide Neural Networkscited by 166
- 2022If Influence Functions are the Answer, Then What is the Question?cited by 142
- 2020Tensor Programs II: Neural Tangent Kernel for Any Architecturecited by 139
- 2023DataInf: Efficiently Estimating Data Influence in LoRA-tuned LLMs and Diffusion Modelscited by 121
- 2020Infinite attention: NNGP and NTK for deep attention networkscited by 117
- 2023Linear attention is (maybe) all you need (to understand transformer optimization)cited by 97
- 2024Transformers Learn Nonlinear Features In Context: Nonconvex Mean-field Dynamics on the Attention Landscapecited by 43
- 2021Rethinking Influence Functions of Neural Networks in the Over-parameterized Regimecited by 26
- 2025Infinite-Width Limit of a Single Attention Layer: Analysis via Tensor Programscited by 5
Built on by
Not cited yet.