English

Linear Transformers Implicitly Discover Unified Numerical Algorithms

Machine Learning 2025-09-25 v1 Artificial Intelligence

Abstract

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.

Keywords

Cite

@article{arxiv.2509.19702,
  title  = {Linear Transformers Implicitly Discover Unified Numerical Algorithms},
  author = {Patrick Lutz and Aditya Gangrade and Hadi Daneshmand and Venkatesh Saligrama},
  journal= {arXiv preprint arXiv:2509.19702},
  year   = {2025}
}

Comments

To appear at NeurIPS 2025

R2 v1 2026-07-01T05:53:25.155Z