Enhancing Vision-Language Models Generalization via Diversity-Driven Novel Feature Synthesis

Siyuan Yan,Cheng Luo,Zhen Yu,Zongyuan Ge
2024-08-13
Abstract:Vision-language foundation models like CLIP have shown impressive zero-shot generalization, but finetuning on downstream datasets can cause overfitting and loss of its generalization ability on unseen domains. Although collecting additional data from new domains of interest is possible, this method is often impractical due to the challenges in obtaining annotated data. To address this, we propose a plug-and-play feature synthesis method called LDFS (Language-Guided Diverse Feature Synthesis) to synthesize new domain features and improve existing CLIP fine-tuning strategies. LDFS has three main contributions: 1) To synthesize novel domain features and promote diversity, we propose an instance-conditional feature augmentation strategy based on a text-guided feature augmentation loss. 2) To maintain feature quality after augmenting, we introduce a pairwise regularizer to preserve augmented feature coherence within the CLIP feature space. 3) We propose to use stochastic text feature augmentation to reduce the modality gap and further facilitate the process of text-guided feature synthesis. Extensive experiments show LDFS superiority in improving CLIP generalization ability on unseen domains without collecting data from those domains. The code will be made publicly available.
Computer Vision and Pattern Recognition
What problem does this paper attempt to address?
The problem that this paper attempts to solve is how to improve the generalization ability of the model for unseen domains without collecting new - domain data when fine - tuning on vision - language foundation models (such as CLIP). Specifically, the paper points out that although models such as CLIP perform well in zero - shot scenarios, when adapting to downstream tasks through fine - tuning, it may lead to over - fitting, thereby reducing their generalization performance in unseen domains. In addition, collecting annotated data in new domains is often impractical because it involves high costs and difficulties in data acquisition. To address this challenge, the paper proposes a method named LDFS (Language - Guided Diverse Feature Synthesis), which enhances the generalization ability of CLIP through the following three main contributions: 1. **Promote feature diversity**: A text - guided instance - conditional feature enhancement strategy is proposed. By introducing an instance - conditional feature enhancement loss to synthesize new domain features, the diversity of features is promoted. 2. **Maintain feature quality**: A pairwise regularizer is introduced to maintain the consistency of enhanced features within the CLIP feature space, thereby maintaining the quality of features. 3. **Reduce modality gap**: Random text feature enhancement is used to reduce the modality gap between text and image, further promoting the text - guided feature synthesis process. Through these methods, LDFS can significantly improve the generalization ability of CLIP in unseen domains without relying on target - domain data. The paper verifies the effectiveness of LDFS through experiments on multiple challenging domain - transfer benchmark datasets, including PACS, Office - Home, DomainNet and NICO++.