English

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

Machine Learning 2026-05-11 v2 Computation and Language

Abstract

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.

Keywords

Cite

@article{arxiv.2605.04651,
  title  = {FAAST: Forward-Only Associative Learning via Closed-Form Fast Weights for Test-Time Supervised Adaptation},
  author = {Guangsheng Bao and Hongbo Zhang and Han Cui and Ke Sun and Yanbin Zhao and Juncai He and Yue Zhang},
  journal= {arXiv preprint arXiv:2605.04651},
  year   = {2026}
}

Comments

9 pages, 6 figures, 10 tables

R2 v1 2026-07-01T12:52:23.101Z