English

CLIP meets Model Zoo Experts: Pseudo-Supervision for Visual Enhancement

Machine Learning 2023-10-24 v1 Artificial Intelligence Computer Vision and Pattern Recognition

Abstract

Contrastive language image pretraining (CLIP) is a standard method for training vision-language models. While CLIP is scalable, promptable, and robust to distribution shifts on image classification tasks, it lacks object localization capabilities. This paper studies the following question: Can we augment CLIP training with task-specific vision models from model zoos to improve its visual representations? Towards this end, we leverage open-source task-specific vision models to generate pseudo-labels for an uncurated and noisy image-text dataset. Subsequently, we train CLIP models on these pseudo-labels in addition to the contrastive training on image and text pairs. This simple setup shows substantial improvements of up to 16.3% across different vision tasks, including segmentation, detection, depth estimation, and surface normal estimation. Importantly, these enhancements are achieved without compromising CLIP's existing capabilities, including its proficiency in promptable zero-shot classification.

Keywords

Cite

@article{arxiv.2310.14108,
  title  = {CLIP meets Model Zoo Experts: Pseudo-Supervision for Visual Enhancement},
  author = {Mohammadreza Salehi and Mehrdad Farajtabar and Maxwell Horton and Fartash Faghri and Hadi Pouransari and Raviteja Vemulapalli and Oncel Tuzel and Ali Farhadi and Mohammad Rastegari and Sachin Mehta},
  journal= {arXiv preprint arXiv:2310.14108},
  year   = {2023}
}