Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Texts as Images in Prompt Tuning for Multi-Label Image Recognition

About

Prompt tuning has been employed as an efficient way to adapt large vision-language pre-trained models (e.g. CLIP) to various downstream tasks in data-limited or label-limited settings. Nonetheless, visual data (e.g., images) is by default prerequisite for learning prompts in existing methods. In this work, we advocate that the effectiveness of image-text contrastive learning in aligning the two modalities (for training CLIP) further makes it feasible to treat texts as images for prompt tuning and introduce TaI prompting. In contrast to the visual data, text descriptions are easy to collect, and their class labels can be directly derived. Particularly, we apply TaI prompting to multi-label image recognition, where sentences in the wild serve as alternatives to images for prompt tuning. Moreover, with TaI, double-grained prompt tuning (TaI-DPT) is further presented to extract both coarse-grained and fine-grained embeddings for enhancing the multi-label recognition performance. Experimental results show that our proposed TaI-DPT outperforms zero-shot CLIP by a large margin on multiple benchmarks, e.g., MS-COCO, VOC2007, and NUS-WIDE, while it can be combined with existing methods of prompting from images to improve recognition performance further. Code is released at https://github.com/guozix/TaI-DPT.

Zixian Guo, Bowen Dong, Zhilong Ji, Jinfeng Bai, Yiwen Guo, Wangmeng Zuo• 2022

Related benchmarks

TaskDatasetResultRank
Multi-Label ClassificationPASCAL VOC 2007 (test)
mAP88.3
125
Multi-Label ClassificationNUS-WIDE (test)
mAP46.5
112
Multi-Label ClassificationVOC-MLT (test)
Total mAP83.75
34
Multi-label Image ClassificationPASCAL VOC 2007
mAP93.3
25
Long-Tailed Multi-Label Visual RecognitionCOCO Long-Tailed (test)
mAP Total56.23
21
Multi-label image recognitionPASCAL VOC 2007
mAP @ 10% Threshold93.3
18
Multi-label recognitionMS-COCO (val)--
18
Multi-Label ClassificationCOCO 2014
mAP81.5
14
Multi-label recognitionMS-COCO 16 novel classes
mAP59.2
5
Multi-label recognitionMS-COCO
IoU@10%81.5
4
Showing 10 of 11 rows

Other info

Code

Follow for update