What can linear attention learn from nonlinear teachers in-context?
Not on the map. Why?
Abstract (arXiv)
Linear attention is a tractable model for understanding the mechanisms governing in-context learning in transformers. For linear regression tasks, recent asymptotic analyses have characterised its learning and generalisation behaviour. We extend this theory to nonlinear single-index targets, y=f(x^top w)+varepsilon . Our main result establishes a nonlinearity-noise equivalence: linear attention extracts only the linear Hermite component of f, while the remaining nonlinear structure contributes to the generalisation error as effective noise. This reduction allows results from the corresponding linear theory to be transferred to nonlinear tasks. We illustrate its implications for finite pretraining data and for the transition from task memorisation to task generalisation as task diversity increases. These results identify a limitation of the reduced linear-attention model and provide a tractable starting point for studying nonlinear in-context learning.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 5 of its 21 references that are in the map. First posted on arXiv 7 Oct 2026.
Builds on
5 of 21 references.
References · 5
- 2023Gated Linear Attention Transformers with Hardware-Efficient Trainingcited by 490
- 2023What and How does In-Context Learning Learn? Bayesian Model Averaging, Parameterization, and Generalizationcited by 112
- 2024Pretrained transformer efficiently learns low-dimensional target functions in-contextcited by 29
- 2025Theory of Scaling Laws for In-Context Regression: Depth, Width, Context and Timecited by 15
- 2026In-context Learning of Single-index Targets: Comparing Kernel and Feature Learnerscited by 1
Built on by
Not cited yet.