Meta-Learning Fast Weight Language Models
Not on the map. Why?
Abstract (arXiv)
Dynamic evaluation of language models (LMs) adapts model parameters at test time using gradient information from previous tokens and substantially improves LM performance. However, it requires over 3x more compute than standard inference. We present Fast Weight Layers (FWLs), a neural component that provides the benefits of dynamic evaluation much more efficiently by expressing gradient updates as linear attention. A key improvement over dynamic evaluation is that FWLs can also be applied at training time so the model learns to make good use of gradient updates. FWLs can easily be added on top of existing transformer models, require relatively little extra compute or memory to run, and significantly improve language modeling perplexity.
Key facts. Cited by 20 papers in the map (data through 7 Oct 2026). Builds on 21 of its 24 references that are in the map. First posted on arXiv 5 Dec 2022.
Builds on
21 of 24 references.
References · 21
- 2020Language Models are Few-Shot Learnerscited by 32,199
- 2016Layer Normalizationcited by 5,579
- 2017Model-Agnostic Meta-Learning for Fast Adaptation of Deep Networkscited by 4,861
- 2016Pointer Sentinel Mixture Modelscited by 2,992
- 2019The Curious Case of Neural Text Degenerationcited by 2,881
- 2019Transformer-XL: Attentive Language Models Beyond a Fixed-Length Contextcited by 1,830
- 2019Generating Long Sequences with Sparse Transformerscited by 1,523
- 2019Generalization through Memorization: Nearest Neighbor Language Modelscited by 788
- 2021Fast Model Editing at Scalecited by 491
- 2021Linear Transformers Are Secretly Fast Weight Programmerscited by 471
- 2020Efficient Content-Based Sparse Attention with Routing Transformerscited by 469
- 2019Rapid Learning or Feature Reuse? Towards Understanding the Effectiveness of MAMLcited by 359
9 more
- 2017Meta Networkscited by 268
- 2022Transformer Quality in Linear Timecited by 189
- 2016Using Fast Weights to Attend to the Recent Pastcited by 161
- 2021Going Beyond Linear Transformers with Recurrent Fast Weight Programmerscited by 84
- 2016Efficient softmax approximation for GPUscited by 67
- 2017Dynamic Evaluation of Neural Sequence Modelscited by 53
- 2019Metalearned Neural Memorycited by 44
- 2019Dynamic Evaluation of Transformer Language Modelscited by 23
- 2021Reconsidering the Past: Optimizing Hidden States in Language Modelscited by 3
Built on by
Citing papers per year: 2023: 6; 2024: 4; 2025: 5; 2026: 5.
20 papers.
Sort:
By area:
- 2024Learning to (Learn at Test Time): RNNs with Expressive Hidden Statescited by 329
- 2025One-Minute Video Generation with Test-Time Trainingcited by 100
- 2024What is Wrong with Perplexity for Long-context Language Modeling?cited by 68
- 2025Test-time regression: a unifying framework for designing sequence models with associative memorycited by 63
- 2025End-to-End Test-Time Training for Long Contextcited by 53
- 2023Meta-Learning Online Adaptation of Language Modelscited by 43
- 2023Compressed Context Memory For Online Language Model Interactioncited by 42
- 2025MesaNet: Sequence Modeling by Locally Optimal Test-Time Trainingcited by 37
- 2024Generative Adapter: Contextualizing Language Models in Parameters with A Single Forward Passcited by 26
- 2025Lattice: Learning to Efficiently Compress the Memorycited by 18
- 2023Trainable Transformer in Transformercited by 17
- 2023kNN-Adapter: Efficient Domain Adaptation for Black-Box Language Modelscited by 13
8 more
- 2023Learning to (Learn at Test Time)cited by 12
- 2023When Meta-Learning Meets Online and Continual Learning: A Surveycited by 12
- 2024Online Test-Time Adaptation of Spatial-Temporal Traffic Flow Forecastingcited by 11
- 2026Rethinking Personalization in Large Language Models at the Token Levelcited by 3
- 2026MDN: Parallelizing Stepwise Momentum for Delta Linear Attentioncited by 3
- 2026Reinforced Fast Weights with Next-Sequence Predictioncited by 1
- 2026FocuSFT: Bilevel Optimization for Dilution-Aware Long-Context Fine-Tuningcited by 0
- 2026GradLev: Token-Parallel Test-Time Training Via Costate Predictioncited by 0
- 2026GradLev: Token-Parallel Test-Time Training Via Costate Prediction28 Sep 2026
- 2026FocuSFT: Bilevel Optimization for Dilution-Aware Long-Context Fine-Tuning11 May 2026
- 2026MDN: Parallelizing Stepwise Momentum for Delta Linear Attention7 May 2026
- 2026Reinforced Fast Weights with Next-Sequence Prediction18 Feb 2026
- 2026Rethinking Personalization in Large Language Models at the Token Level4 Feb 2026
- 2025End-to-End Test-Time Training for Long Context29 Dec 2025
- 2025MesaNet: Sequence Modeling by Locally Optimal Test-Time Training5 Jun 2025
- 2025Lattice: Learning to Efficiently Compress the Memory8 Apr 2025
- 2025One-Minute Video Generation with Test-Time Training7 Apr 2025
- 2025Test-time regression: a unifying framework for designing sequence models with associative memory21 Jan 2025
- 2024Generative Adapter: Contextualizing Language Models in Parameters with A Single Forward Pass8 Nov 2024
- 2024What is Wrong with Perplexity for Long-context Language Modeling?31 Oct 2024
8 more
- 2024Learning to (Learn at Test Time): RNNs with Expressive Hidden States5 Jul 2024
- 2024Online Test-Time Adaptation of Spatial-Temporal Traffic Flow Forecasting8 Jan 2024
- 2023Compressed Context Memory For Online Language Model Interaction6 Dec 2023
- 2023When Meta-Learning Meets Online and Continual Learning: A Survey9 Nov 2023
- 2023Learning to (Learn at Test Time)20 Oct 2023
- 2023Trainable Transformer in Transformer3 Jul 2023
- 2023Meta-Learning Online Adaptation of Language Models24 May 2023
- 2023kNN-Adapter: Efficient Domain Adaptation for Black-Box Language Models21 Feb 2023
- 2024Learning to (Learn at Test Time): RNNs with Expressive Hidden Statescited by 329
- 2025One-Minute Video Generation with Test-Time Trainingcited by 100
- 2024What is Wrong with Perplexity for Long-context Language Modeling?cited by 68
- 2025Test-time regression: a unifying framework for designing sequence models with associative memorycited by 63
- 2025End-to-End Test-Time Training for Long Contextcited by 53
- 2023Meta-Learning Online Adaptation of Language Modelscited by 43
- 2023Compressed Context Memory For Online Language Model Interactioncited by 42
- 2025MesaNet: Sequence Modeling by Locally Optimal Test-Time Trainingcited by 37
- 2024Generative Adapter: Contextualizing Language Models in Parameters with A Single Forward Passcited by 26
- 2025Lattice: Learning to Efficiently Compress the Memorycited by 18
- 2023Trainable Transformer in Transformercited by 17
- 2023kNN-Adapter: Efficient Domain Adaptation for Black-Box Language Modelscited by 13
- 2023Learning to (Learn at Test Time)cited by 12
- 2023When Meta-Learning Meets Online and Continual Learning: A Surveycited by 12
12 most cited of 17 All 17 (JSON)
- 2026GradLev: Token-Parallel Test-Time Training Via Costate Prediction28 Sep 2026
- 2026FocuSFT: Bilevel Optimization for Dilution-Aware Long-Context Fine-Tuning11 May 2026
- 2026MDN: Parallelizing Stepwise Momentum for Delta Linear Attention7 May 2026
- 2026Reinforced Fast Weights with Next-Sequence Prediction18 Feb 2026
- 2025End-to-End Test-Time Training for Long Context29 Dec 2025
- 2025MesaNet: Sequence Modeling by Locally Optimal Test-Time Training5 Jun 2025
- 2025Lattice: Learning to Efficiently Compress the Memory8 Apr 2025
- 2025Test-time regression: a unifying framework for designing sequence models with associative memory21 Jan 2025
- 2024Generative Adapter: Contextualizing Language Models in Parameters with A Single Forward Pass8 Nov 2024
- 2024What is Wrong with Perplexity for Long-context Language Modeling?31 Oct 2024
- 2024Online Test-Time Adaptation of Spatial-Temporal Traffic Flow Forecasting8 Jan 2024
- 2023Compressed Context Memory For Online Language Model Interaction6 Dec 2023
12 newest of 17 All 17 (JSON)