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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Image Classification | FGVC-Aircraft (test) | -- | 322 | |
| Image Classification | Stanford Cars (test) | -- | 320 | |
| Image Classification | SUN397 (test) | Top-1 Accuracy70.48 | 251 | |
| Image Classification | Oxford Flowers-102 (test) | Top-1 Accuracy75.56 | 221 | |
| Image Classification | Caltech101 (test) | -- | 204 | |
| Image Classification | EuroSAT (test) | -- | 195 | |
| Image Classification | UCF-101 (test) | Accuracy71.74 | 127 | |
| Image Classification | DTD (test) | Accuracy (DTD Test)55.5 | 65 | |
| Image Classification | ImageNet Natural Distribution Shifts suite (ImageNet, ImageNet-A, ImageNet-V2, ImageNet-R, ImageNet-Sketch) (test) | Top-1 Accuracy (ImageNet)72.16 | 55 | |
| Image Classification | Food-101 (test) | Accuracy86.48 | 36 |