Linear Transformers Implicitly Discover Unified Numerical Algorithms
Not on the map. Why?
Abstract (arXiv)
We train a linear attention transformer on millions of masked-block matrix completion tasks: each prompt is masked low-rank matrix whose missing block may be (i) a scalar prediction target or (ii) an unseen kernel slice of Nystr"om extrapolation. The model sees only input-output pairs and a mean-squared loss; it is given no normal equations, no handcrafted iterations, and no hint that the tasks are related. Surprisingly, after training, algebraic unrolling reveals the same parameter-free update rule across three distinct computational regimes (full visibility, rank-limited updates, and distributed computation). We prove that this rule achieves second-order convergence on full-batch problems, cuts distributed iteration complexity, and remains accurate with rank-limited attention. Thus, a transformer trained solely to patch missing blocks implicitly discovers a unified, resource-adaptive iterative solver spanning prediction, estimation, and Nystr"om extrapolation, highlighting a powerful capability of in-context learning.
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 17 of its 44 references that are in the map. First posted on arXiv 24 Sep 2025.
Builds on
17 of 44 references.
References · 17
- 2017Attention Is All You Needcited by 60,361
- 2020Language Models are Few-Shot Learnerscited by 32,419
- 2022In-context Learning and Induction Headscited by 964
- 2021An Explanation of In-context Learning as Implicit Bayesian Inferencecited by 938
- 2022Transformers learn in-context by gradient descentcited by 732
- 2022What Can Transformers Learn In-Context? A Case Study of Simple Function Classescited by 715
- 2022What learning algorithm is in-context learning? Investigations with linear modelscited by 684
- 2021MetaICL: Learning to Learn In Contextcited by 454
- 2023Transformers as Statisticians: Provable In-Context Learning with In-Context Algorithm Selectioncited by 298
- 2014Sketching as a Tool for Numerical Linear Algebracited by 288
- 2023Transformers learn to implement preconditioned gradient descent for in-context learningcited by 279
- 2023Looped Transformers as Programmable Computerscited by 237
5 more
- 2013Revisiting the Nystrom Method for Improved Large-Scale Machine Learningcited by 77
- 2023Transformers Learn to Achieve Second-Order Convergence Rates for In-Context Linear Regressioncited by 49
- 2024Linear Transformers are Versatile In-Context Learnerscited by 28
- 2024Transformers Can Learn Temporal Difference Methods for In-Context Reinforcement Learningcited by 23
- 2024MLPs Learn In-Context on Regression and Classification Taskscited by 13
Built on by
Citing papers per year: 2026: 1.
1 paper.