English

CPT: Colorful Prompt Tuning for Pre-trained Vision-Language Models

Computer Vision and Pattern Recognition 2022-05-23 v3 Computation and Language

Abstract

Pre-Trained Vision-Language Models (VL-PTMs) have shown promising capabilities in grounding natural language in image data, facilitating a broad variety of cross-modal tasks. However, we note that there exists a significant gap between the objective forms of model pre-training and fine-tuning, resulting in a need for large amounts of labeled data to stimulate the visual grounding capability of VL-PTMs for downstream tasks. To address the challenge, we present Cross-modal Prompt Tuning (CPT, alternatively, Colorful Prompt Tuning), a novel paradigm for tuning VL-PTMs, which reformulates visual grounding into a fill-in-the-blank problem with color-based co-referential markers in image and text, maximally mitigating the gap. In this way, CPT enables strong few-shot and even zero-shot visual grounding capabilities of VL-PTMs. Comprehensive experimental results show that the prompt-tuned VL-PTMs outperform their fine-tuned counterparts by a large margin (e.g., 17.3% absolute accuracy improvement, and 73.8% relative standard deviation reduction on average with one shot in RefCOCO evaluation). We make the data and code for this paper publicly available at https://github.com/thunlp/CPT.

Keywords

Cite

@article{arxiv.2109.11797,
  title  = {CPT: Colorful Prompt Tuning for Pre-trained Vision-Language Models},
  author = {Yuan Yao and Ao Zhang and Zhengyan Zhang and Zhiyuan Liu and Tat-Seng Chua and Maosong Sun},
  journal= {arXiv preprint arXiv:2109.11797},
  year   = {2022}
}

Comments

Work in progress