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

Unified Vision and Language Prompt Learning

About

Prompt tuning, a parameter- and data-efficient transfer learning paradigm that tunes only a small number of parameters in a model's input space, has become a trend in the vision community since the emergence of large vision-language models like CLIP. We present a systematic study on two representative prompt tuning methods, namely text prompt tuning and visual prompt tuning. A major finding is that none of the unimodal prompt tuning methods performs consistently well: text prompt tuning fails on data with high intra-class visual variances while visual prompt tuning cannot handle low inter-class variances. To combine the best from both worlds, we propose a simple approach called Unified Prompt Tuning (UPT), which essentially learns a tiny neural network to jointly optimize prompts across different modalities. Extensive experiments on over 11 vision datasets show that UPT achieves a better trade-off than the unimodal counterparts on few-shot learning benchmarks, as well as on domain generalization benchmarks. Code and models will be released to facilitate future research.

Yuhang Zang, Wei Li, Kaiyang Zhou, Chen Huang, Chen Change Loy• 2022

Related benchmarks

TaskDatasetResultRank
Image ClassificationImageNet V2--
767
Image ClassificationImageNet-R
Top-1 Acc76.24
622
Image ClassificationDTD
Accuracy36.91
610
Image ClassificationEuroSAT
Accuracy60.43
569
Image ClassificationFlowers102
Accuracy68.03
558
Image ClassificationRESISC45
Accuracy62.84
539
Image ClassificationMNIST
Accuracy64.44
417
Image ClassificationImageNet
Top-1 Accuracy72.63
384
Image ClassificationFGVC Aircraft
Accuracy15.89
223
Image ClassificationImageNet Domain Generalization (Source: ImageNet, Targets: ImageNetV2, ImageNet-Sketch, ImageNet-A, ImageNet-R) (test)
Accuracy (ImageNetV2)64.35
105
Showing 10 of 18 rows

Other info

Follow for update