Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

ComMem: Complementary Memory Systems for Test-Time Adaptation of Vision-Language Models

About

Test-time adaptation (TTA) of vision-language models (VLMs) is essential for their robust deployment in dynamic, real-world environments. However, existing TTA methods often adapt locally without accumulating knowledge over time, or operating within a single modality without exploiting VLMs' inherently multi-modal nature. Inspired by the \textbf{Com}plementary \textbf{Mem}ory systems of the biological brain, we propose \textbf{ComMem}, an innovative approach that mimics the distinct but cooperative roles of the hippocampus and neocortex to enable effective TTA for VLMs. ComMem consists of two key components: a fast-adapting detailed memory, akin to the hippocampus, that forms a dynamic visual cache from high-confidence test samples; and a slow-integrating abstract memory, akin to the neocortex, that continually refines global textual prototypes. For each test instance, ComMem jointly optimizes both memory systems to ensure cross-modal consistency. Extensive experiments on 15 benchmark datasets show that ComMem significantly outperforms state-of-the-art methods under both natural distribution shifts and cross-dataset generalization, offering a promising direction for enhancing VLMs' practical adaptability.

Guanglong Sun, Shuang Cui, Bo Lei, Liyuan Wang, Zihan Zhai, Hongwei Yan, Hang Su, Jun Zhu, Yi Zhong• 2026

Related benchmarks

TaskDatasetResultRank
Image ClassificationFGVC-Aircraft (test)--
322
Image ClassificationStanford Cars (test)--
320
Image ClassificationSUN397 (test)
Top-1 Accuracy70.48
251
Image ClassificationOxford Flowers-102 (test)
Top-1 Accuracy75.56
221
Image ClassificationCaltech101 (test)--
204
Image ClassificationEuroSAT (test)--
195
Image ClassificationUCF-101 (test)
Accuracy71.74
127
Image ClassificationDTD (test)
Accuracy (DTD Test)55.5
65
Image ClassificationImageNet Natural Distribution Shifts suite (ImageNet, ImageNet-A, ImageNet-V2, ImageNet-R, ImageNet-Sketch) (test)
Top-1 Accuracy (ImageNet)72.16
55
Image ClassificationFood-101 (test)
Accuracy86.48
36
Showing 10 of 12 rows

Other info

Follow for update