English

Understanding Retrieval-Augmented Task Adaptation for Vision-Language Models

Machine Learning 2024-05-03 v1 Artificial Intelligence Computer Vision and Pattern Recognition

Abstract

Pre-trained contrastive vision-language models have demonstrated remarkable performance across a wide range of tasks. However, they often struggle on fine-trained datasets with categories not adequately represented during pre-training, which makes adaptation necessary. Recent works have shown promising results by utilizing samples from web-scale databases for retrieval-augmented adaptation, especially in low-data regimes. Despite the empirical success, understanding how retrieval impacts the adaptation of vision-language models remains an open research question. In this work, we adopt a reflective perspective by presenting a systematic study to understand the roles of key components in retrieval-augmented adaptation. We unveil new insights on uni-modal and cross-modal retrieval and highlight the critical role of logit ensemble for effective adaptation. We further present theoretical underpinnings that directly support our empirical observations.

Keywords

Cite

@article{arxiv.2405.01468,
  title  = {Understanding Retrieval-Augmented Task Adaptation for Vision-Language Models},
  author = {Yifei Ming and Yixuan Li},
  journal= {arXiv preprint arXiv:2405.01468},
  year   = {2024}
}

Comments

The paper is accepted at ICML 2024

R2 v1 2026-06-28T16:14:26.101Z