Learning to (Learn at Test Time)
Not on the map. Why?
Abstract (arXiv)
We reformulate the problem of supervised learning as learning to learn with two nested loops (i.e. learning problems). The inner loop learns on each individual instance with self-supervision before final prediction. The outer loop learns the self-supervised task used by the inner loop, such that its final prediction improves. Our inner loop turns out to be equivalent to linear attention when the inner-loop learner is only a linear model, and to self-attention when it is a kernel estimator. For practical comparison with linear or self-attention layers, we replace each of them in a transformer with an inner loop, so our outer loop is equivalent to training the architecture. When each inner-loop learner is a neural network, our approach vastly outperforms transformers with linear attention on ImageNet from 224 x 224 raw pixels in both accuracy and FLOPs, while (regular) transformers cannot run.
Key facts. Cited by 17 papers in the map (data through 14 Oct 2026). Builds on 38 of its 67 references that are in the map. First posted on arXiv 20 Oct 2023.
Builds on
38 of 67 references.
References · 38
- 2020Language Models are Few-Shot Learnerscited by 32,419
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
- 2021Masked Autoencoders Are Scalable Vision Learnerscited by 7,403
- 2017Model-Agnostic Meta-Learning for Fast Adaptation of Deep Networkscited by 4,897
- 2014Going Deeper with Convolutionscited by 4,241
- 2020Longformer: The Long-Document Transformercited by 3,413
- 2021BEiT: BERT Pre-Training of Image Transformerscited by 2,375
- 2020Transformers are RNNs: Fast Autoregressive Transformers with Linear Attentioncited by 2,022
- 2020Linformer: Self-Attention with Linear Complexitycited by 1,474
- 2016Context Encoders: Feature Learning by Inpaintingcited by 1,428
- 2022Transformers learn in-context by gradient descentcited by 732
- 2022What learning algorithm is in-context learning? Investigations with linear modelscited by 684
26 more
- 2016Learning to learn by gradient descent by gradient descentcited by 620
- 2017Evolution Strategies as a Scalable Alternative to Reinforcement Learningcited by 574
- 2021Linear Transformers Are Secretly Fast Weight Programmerscited by 502
- 2023Hyena Hierarchy: Towards Larger Convolutional Language Modelscited by 393
- 2023Fine-Tuning Language Models with Just Forward Passescited by 345
- 2023Trained Transformers Learn Linear Models In-Contextcited by 337
- 2017"Zero-Shot" Super-Resolution using Deep Internal Learningcited by 286
- 2023Transformers learn to implement preconditioned gradient descent for in-context learningcited by 279
- 2022The Forward-Forward Algorithm: Some Preliminary Investigationscited by 266
- 2020Consistent Video Depth Estimationcited by 220
- 2022Test-Time Training with Masked Autoencoderscited by 212
- 2016Using Fast Weights to Attend to the Recent Pastcited by 170
- 2023One Step of Gradient Descent is Provably the Optimal In-Context Learner with One Layer of Linear Self-Attentioncited by 170
- 2021SOFT: Softmax-free Transformer with Linear Complexitycited by 128
- 2020Self-Supervised Policy Adaptation during Deploymentcited by 123
- 2013Learning by Transductioncited by 121
- 2017A Tutorial on Kernel Density Estimation and Recent Advancescited by 101
- 2022Better plain ViT baselines for ImageNet-1kcited by 98
- 2021Going Beyond Linear Transformers with Recurrent Fast Weight Programmerscited by 89
- 2023Test-Time Training on Nearest Neighbors for Large Language Modelscited by 88
- 2023Transformers as Support Vector Machinescited by 74
- 2018Online Model Distillation for Efficient Video Inferencecited by 60
- 2020Meta Learning Backpropagation And Improving Itcited by 49
- 2018Meta-Learning Update Rules for Unsupervised Representation Learningcited by 37
- 2023Test-Time Training on Video Streamscited by 37
- 2022Meta-Learning Fast Weight Language Modelscited by 21
Built on by
Citing papers per year: 2023: 1; 2024: 2; 2025: 5; 2026: 9.
17 papers.
Sort:
By area:
- 2024Learning to (Learn at Test Time): RNNs with Expressive Hidden Statescited by 355
- 2024TextGrad: Automatic "Differentiation" via Textcited by 273
- 2025End-to-End Test-Time Training for Long Contextcited by 62
- 2025SLOT: Sample-specific Language Model Optimization at Test-timecited by 18
- 2025Learning to Reason from Feedback at Test-Timecited by 10
- 2026AdaMEM: Test-Time Adaptive Memory for Language Agentscited by 6
- 2023MyPortrait: Morphable Prior-Guided Personalized Portrait Generationcited by 3
- 2025UrbanMind: Urban Dynamics Prediction with Multifaceted Spatial-Temporal Large Language Modelscited by 2
- 2026Learning to Remember, Learn, and Forget in Attention-Based Modelscited by 2
- 2026Rethinking the State Update Gate for Long-Sequence Recurrent 3D Reconstructioncited by 1
- 2026The Power of Test-Time Training for Approximate Samplingcited by 1
- 2025Titans Revisited: A Lightweight Reimplementation and Critical Analysis of a Test-Time Memory Modelcited by 0
5 more
- 2026From Migration to Calibration: Aligning Agents with Deployment Requirementscited by 0
- 2026Universal Test-Time Trainingcited by 0
- 2026Evolving in Thought Space: Training a Small Model at Test Time Unlocks Better Discoveriescited by 0
- 2026Towards In-Parameter Memory Augmentation for Large Language Modelscited by 0
- 2026Travel Back To Move Forward: Retrospective Exploration and Evolving for Long Horizon Agentscited by 0
- 2026Travel Back To Move Forward: Retrospective Exploration and Evolving for Long Horizon Agents9 Oct 2026
- 2026Towards In-Parameter Memory Augmentation for Large Language Models6 Oct 2026
- 2026Evolving in Thought Space: Training a Small Model at Test Time Unlocks Better Discoveries5 Oct 2026
- 2026Universal Test-Time Training4 Oct 2026
- 2026From Migration to Calibration: Aligning Agents with Deployment Requirements28 Sep 2026
- 2026The Power of Test-Time Training for Approximate Sampling9 Jun 2026
- 2026AdaMEM: Test-Time Adaptive Memory for Language Agents4 Jun 2026
- 2026Rethinking the State Update Gate for Long-Sequence Recurrent 3D Reconstruction16 May 2026
- 2026Learning to Remember, Learn, and Forget in Attention-Based Models9 Feb 2026
- 2025End-to-End Test-Time Training for Long Context29 Dec 2025
- 2025Titans Revisited: A Lightweight Reimplementation and Critical Analysis of a Test-Time Memory Model10 Oct 2025
- 2025SLOT: Sample-specific Language Model Optimization at Test-time18 May 2025
5 more
- 2025UrbanMind: Urban Dynamics Prediction with Multifaceted Spatial-Temporal Large Language Models16 May 2025
- 2025Learning to Reason from Feedback at Test-Time16 Feb 2025
- 2024Learning to (Learn at Test Time): RNNs with Expressive Hidden States5 Jul 2024
- 2024TextGrad: Automatic "Differentiation" via Text11 Jun 2024
- 2023MyPortrait: Morphable Prior-Guided Personalized Portrait Generation5 Dec 2023
- 2026Universal Test-Time Trainingcited by 0
- 2026Universal Test-Time Training4 Oct 2026
- 2024TextGrad: Automatic "Differentiation" via Textcited by 273
- 2025End-to-End Test-Time Training for Long Contextcited by 62
- 2025SLOT: Sample-specific Language Model Optimization at Test-timecited by 18
- 2023MyPortrait: Morphable Prior-Guided Personalized Portrait Generationcited by 3
- 2025UrbanMind: Urban Dynamics Prediction with Multifaceted Spatial-Temporal Large Language Modelscited by 2
- 2026Learning to Remember, Learn, and Forget in Attention-Based Modelscited by 2
- 2026Rethinking the State Update Gate for Long-Sequence Recurrent 3D Reconstructioncited by 1
- 2026The Power of Test-Time Training for Approximate Samplingcited by 1
- 2025Titans Revisited: A Lightweight Reimplementation and Critical Analysis of a Test-Time Memory Modelcited by 0
- 2026From Migration to Calibration: Aligning Agents with Deployment Requirementscited by 0
- 2026From Migration to Calibration: Aligning Agents with Deployment Requirements28 Sep 2026
- 2026The Power of Test-Time Training for Approximate Sampling9 Jun 2026
- 2026Rethinking the State Update Gate for Long-Sequence Recurrent 3D Reconstruction16 May 2026
- 2026Learning to Remember, Learn, and Forget in Attention-Based Models9 Feb 2026
- 2025End-to-End Test-Time Training for Long Context29 Dec 2025
- 2025Titans Revisited: A Lightweight Reimplementation and Critical Analysis of a Test-Time Memory Model10 Oct 2025
- 2025SLOT: Sample-specific Language Model Optimization at Test-time18 May 2025
- 2025UrbanMind: Urban Dynamics Prediction with Multifaceted Spatial-Temporal Large Language Models16 May 2025
- 2024TextGrad: Automatic "Differentiation" via Text11 Jun 2024
- 2023MyPortrait: Morphable Prior-Guided Personalized Portrait Generation5 Dec 2023