English

Beyond Sole Strength: Customized Ensembles for Generalized Vision-Language Models

Computer Vision and Pattern Recognition 2024-06-04 v2

Abstract

Fine-tuning pre-trained vision-language models (VLMs), e.g., CLIP, for the open-world generalization has gained increasing popularity due to its practical value. However, performance advancements are limited when relying solely on intricate algorithmic designs for a single model, even one exhibiting strong performance, e.g., CLIP-ViT-B/16. This paper, for the first time, explores the collaborative potential of leveraging much weaker VLMs to enhance the generalization of a robust single model. The affirmative findings motivate us to address the generalization problem from a novel perspective, i.e., ensemble of pre-trained VLMs. We introduce three customized ensemble strategies, each tailored to one specific scenario. Firstly, we introduce the zero-shot ensemble, automatically adjusting the logits of different models based on their confidence when only pre-trained VLMs are available. Furthermore, for scenarios with extra few-shot samples, we propose the training-free and tuning ensemble, offering flexibility based on the availability of computing resources. The proposed ensemble strategies are evaluated on zero-shot, base-to-new, and cross-dataset generalization, achieving new state-of-the-art performance. Notably, this work represents an initial stride toward enhancing the generalization performance of VLMs via ensemble. The code is available at https://github.com/zhiheLu/Ensemble_VLM.git.

Keywords

Cite

@article{arxiv.2311.17091,
  title  = {Beyond Sole Strength: Customized Ensembles for Generalized Vision-Language Models},
  author = {Zhihe Lu and Jiawang Bai and Xin Li and Zeyu Xiao and Xinchao Wang},
  journal= {arXiv preprint arXiv:2311.17091},
  year   = {2024}
}

Comments

Accepted on ICML 2024

R2 v1 2026-06-28T13:34:35.536Z