mlsanityData to 7 Oct 2026

FAAST: Forward-Only Associative Learning via Closed-Form Fast Weights for Test-Time Supervised Adaptation

arXiv 2605.04651 ↗First posted 6 May 2026cs.LG cs.CLcited by 0

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

  1. 2017Attention Is All You Needcited by 59,932
  2. 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,456
  3. 2020Language Models are Few-Shot Learnerscited by 32,199
  4. 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,084
  5. 2020Retrieval-Augmented Generation for Knowledge-Intensive NLP Taskscited by 9,160
  6. 2018Representation Learning with Contrastive Predictive Codingcited by 7,662
  7. 2022PaLM: Scaling Language Modeling with Pathwayscited by 5,304
  8. 2017Model-Agnostic Meta-Learning for Fast Adaptation of Deep Networkscited by 4,861
  9. 2022Flamingo: a Visual Language Model for Few-Shot Learningcited by 4,337
  10. 2019Parameter-Efficient Transfer Learning for NLPcited by 4,195
  11. 2021Prefix-Tuning: Optimizing Continuous Prompts for Generationcited by 3,878
  12. 2021The Power of Scale for Parameter-Efficient Prompt Tuningcited by 3,769
24 more
  1. 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,611
  2. 2016Pointer Sentinel Mixture Modelscited by 2,992
  3. 2016Matching Networks for One Shot Learningcited by 2,246
  4. 2018Unsupervised Feature Learning via Non-Parametric Instance-level Discriminationcited by 1,838
  5. 2018World Modelscited by 1,444
  6. 2016Understanding intermediate layers using linear classifier probescited by 1,279
  7. 2021Improving language models by retrieving from trillions of tokenscited by 1,278
  8. 2021Train Short, Test Long: Attention with Linear Biases Enables Input Length Extrapolationcited by 1,085
  9. 2022Atlas: Few-shot Learning with Retrieval Augmented Language Modelscited by 940
  10. 2019Generalization through Memorization: Nearest Neighbor Language Modelscited by 788
  11. 2022What Can Transformers Learn In-Context? A Case Study of Simple Function Classescited by 700
  12. 2014Neural Turing Machinescited by 630
  13. 2018Do Better ImageNet Models Transfer Better?cited by 560
  14. 2020Hopfield Networks is All You Needcited by 461
  15. 2014Memory Networkscited by 354
  16. 2023Fine-Tuning Language Models with Just Forward Passescited by 337
  17. 2023Transformers learn to implement preconditioned gradient descent for in-context learningcited by 274
  18. 2022The Forward-Forward Algorithm: Some Preliminary Investigationscited by 262
  19. 2019Revisiting Self-Supervised Visual Representation Learningcited by 250
  20. 2016Using Fast Weights to Attend to the Recent Pastcited by 161
  21. 2021On the Effectiveness of Adapter-based Tuning for Pretrained Language Model Adaptationcited by 147
  22. 2018Differentiable plasticity: training plastic neural networks with backpropagationcited by 83
  23. 2024Parameter-Efficient Fine-Tuning in Large Models: A Survey of Methodologiescited by 44
  24. 2025FrEVL: Leveraging Frozen Pretrained Embeddings for Efficient Vision-Language Understandingcited by 2

Built on by

Not cited yet.