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

TransAgent: Transfer Vision-Language Foundation Models with Heterogeneous Agent Collaboration

About

Vision-language foundation models (such as CLIP) have recently shown their power in transfer learning, owing to large-scale image-text pre-training. However, target domain data in the downstream tasks can be highly different from the pre-training phase, which makes it hard for such a single model to generalize well. Alternatively, there exists a wide range of expert models that contain diversified vision and/or language knowledge pre-trained on different modalities, tasks, networks, and datasets. Unfortunately, these models are "isolated agents" with heterogeneous structures, and how to integrate their knowledge for generalizing CLIP-like models has not been fully explored. To bridge this gap, we propose a general and concise TransAgent framework, which transports the knowledge of the isolated agents in a unified manner, and effectively guides CLIP to generalize with multi-source knowledge distillation. With such a distinct framework, we flexibly collaborate with 11 heterogeneous agents to empower vision-language foundation models, without further cost in the inference phase. Finally, our TransAgent achieves state-of-the-art performance on 11 visual recognition datasets. Under the same low-shot setting, it outperforms the popular CoOp with around 10% on average, and 20% on EuroSAT which contains large domain shifts.

Yiwei Guo, Shaobin Zhuang, Kunchang Li, Yu Qiao, Yali Wang• 2024

Related benchmarks

TaskDatasetResultRank
Image ClassificationFlowers102--
558
Image ClassificationFood101--
457
Image ClassificationStanfordCars--
312
Image ClassificationFGVCAircraft--
261
Image ClassificationCaltech101
Base Accuracy98.9
148
Image ClassificationOxfordPets
Base Accuracy96.33
137
Image ClassificationSUN397
Accuracy (Base)82.9
131
Image ClassificationEuroSAT
Base Accuracy97.43
104
Image ClassificationDTD
Base Score84.37
96
Image ClassificationAverage 11 datasets
Base Accuracy85.29
83
Showing 10 of 14 rows

Other info

Code

Follow for update