FAAST: Forward-Only Associative Learning via Closed-Form Fast Weights for Test-Time Supervised Adaptation
Not on the map. Why?
Abstract (arXiv)
Adapting pretrained models typically involves a trade-off between the high training costs of backpropagation and the heavy inference overhead of memory-based or in-context learning. We propose FAAST, a forward-only associative adaptation method that analytically compiles labeled examples into fast weights in a single pass. By eliminating memory or context dependence, FAAST achieves constant-time inference and decouples task adaptation from pretrained representation. Across image classification and language modeling benchmarks, FAAST matches or exceeds backprop-based adaptation while reducing adaptation time by over 90% and is competitive to memory/context-based adaptation while saving memory usage by up to 95%. These results demonstrate FAAST as a highly efficient, scalable solution for supervised task adaptation, particularly for resource-constrained models. We release the code and models at https://github.com/baoguangsheng/faast.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 36 of its 67 references that are in the map. First posted on arXiv 6 May 2026.
Builds on
36 of 67 references.
References · 36
- 2017Attention Is All You Needcited by 59,932
- 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,456
- 2020Language Models are Few-Shot Learnerscited by 32,199
- 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,084
- 2020Retrieval-Augmented Generation for Knowledge-Intensive NLP Taskscited by 9,160
- 2018Representation Learning with Contrastive Predictive Codingcited by 7,662
- 2022PaLM: Scaling Language Modeling with Pathwayscited by 5,304
- 2017Model-Agnostic Meta-Learning for Fast Adaptation of Deep Networkscited by 4,861
- 2022Flamingo: a Visual Language Model for Few-Shot Learningcited by 4,337
- 2019Parameter-Efficient Transfer Learning for NLPcited by 4,195
- 2021Prefix-Tuning: Optimizing Continuous Prompts for Generationcited by 3,878
- 2021The Power of Scale for Parameter-Efficient Prompt Tuningcited by 3,769
24 more
- 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,611
- 2016Pointer Sentinel Mixture Modelscited by 2,992
- 2016Matching Networks for One Shot Learningcited by 2,246
- 2018Unsupervised Feature Learning via Non-Parametric Instance-level Discriminationcited by 1,838
- 2018World Modelscited by 1,444
- 2016Understanding intermediate layers using linear classifier probescited by 1,279
- 2021Improving language models by retrieving from trillions of tokenscited by 1,278
- 2021Train Short, Test Long: Attention with Linear Biases Enables Input Length Extrapolationcited by 1,085
- 2022Atlas: Few-shot Learning with Retrieval Augmented Language Modelscited by 940
- 2019Generalization through Memorization: Nearest Neighbor Language Modelscited by 788
- 2022What Can Transformers Learn In-Context? A Case Study of Simple Function Classescited by 700
- 2014Neural Turing Machinescited by 630
- 2018Do Better ImageNet Models Transfer Better?cited by 560
- 2020Hopfield Networks is All You Needcited by 461
- 2014Memory Networkscited by 354
- 2023Fine-Tuning Language Models with Just Forward Passescited by 337
- 2023Transformers learn to implement preconditioned gradient descent for in-context learningcited by 274
- 2022The Forward-Forward Algorithm: Some Preliminary Investigationscited by 262
- 2019Revisiting Self-Supervised Visual Representation Learningcited by 250
- 2016Using Fast Weights to Attend to the Recent Pastcited by 161
- 2021On the Effectiveness of Adapter-based Tuning for Pretrained Language Model Adaptationcited by 147
- 2018Differentiable plasticity: training plastic neural networks with backpropagationcited by 83
- 2024Parameter-Efficient Fine-Tuning in Large Models: A Survey of Methodologiescited by 44
- 2025FrEVL: Leveraging Frozen Pretrained Embeddings for Efficient Vision-Language Understandingcited by 2
Built on by
Not cited yet.