From Local Details to Global Context: Advancing Vision-Language Models with Attention-Based Selection
This addresses a specific bottleneck in vision-language models for researchers and practitioners, offering a training-free improvement for zero-shot tasks.
The paper tackles the problem of random augmentations in vision-language models causing overemphasis on local details and background artifacts, compromising global understanding, by proposing an attention-based selection method that achieves state-of-the-art performance on out-of-distribution generalization and zero-shot classification tasks, rivaling few-shot and test-time adaptation methods.
Pretrained vision-language models (VLMs), e.g., CLIP, demonstrate impressive zero-shot capabilities on downstream tasks. Prior research highlights the crucial role of visual augmentation techniques, like random cropping, in alignment with fine-grained class descriptions generated by large language models (LLMs), significantly enhancing zero-shot performance by incorporating multi-view information. However, the inherent randomness of these augmentations can inevitably introduce background artifacts and cause models to overly focus on local details, compromising global semantic understanding. To address these issues, we propose an \textbf{A}ttention-\textbf{B}ased \textbf{S}election (\textbf{ABS}) method from local details to global context, which applies attention-guided cropping in both raw images and feature space, supplement global semantic information through strategic feature selection. Additionally, we introduce a soft matching technique to effectively filter LLM descriptions for better alignment. \textbf{ABS} achieves state-of-the-art performance on out-of-distribution generalization and zero-shot classification tasks. Notably, \textbf{ABS} is training-free and even rivals few-shot and test-time adaptation methods. Our code is available at \href{https://github.com/BIT-DA/ABS}{\textcolor{darkgreen}{https://github.com/BIT-DA/ABS}}.