English

Test-Time Training with KV Binding Is Secretly Linear Attention

Machine Learning 2026-05-14 v4 Artificial Intelligence Computer Vision and Pattern Recognition

Abstract

Test-time training (TTT) with KV binding as sequence modeling layer is commonly interpreted as a form of online meta-learning that memorizes a key-value mapping at test time. However, our analysis reveals multiple phenomena that contradict this memorization-based interpretation. Motivated by these findings, we revisit the formulation of TTT and show that a broad class of TTT architectures can be expressed as a form of learned linear attention operator. Beyond explaining previously puzzling model behaviors, this perspective yields multiple practical benefits: it enables principled architectural simplifications, admits fully parallel formulations that preserve performance while improving efficiency, and provides a systematic reduction of diverse TTT variants to a standard linear attention form. Overall, our results reframe TTT not as test-time memorization, but as learned linear attention with enhanced representational capacity. Project page: https://research.nvidia.com/labs/sil/projects/tttla/.

Keywords

Cite

@article{arxiv.2602.21204,
  title  = {Test-Time Training with KV Binding Is Secretly Linear Attention},
  author = {Junchen Liu and Sven Elflein and Or Litany and Zan Gojcic and Ruilong Li},
  journal= {arXiv preprint arXiv:2602.21204},
  year   = {2026}
}

Comments

ICML 2026, Webpage: https://research.nvidia.com/labs/sil/projects/tttla/

R2 v1 2026-07-01T10:50:31.549Z