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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Image Classification | ImageNet V2 | -- | 767 | |
| Image Classification | ImageNet-R | Top-1 Acc76.24 | 622 | |
| Image Classification | DTD | Accuracy36.91 | 610 | |
| Image Classification | EuroSAT | Accuracy60.43 | 569 | |
| Image Classification | Flowers102 | Accuracy68.03 | 558 | |
| Image Classification | RESISC45 | Accuracy62.84 | 539 | |
| Image Classification | MNIST | Accuracy64.44 | 417 | |
| Image Classification | ImageNet | Top-1 Accuracy72.63 | 384 | |
| Image Classification | FGVC Aircraft | Accuracy15.89 | 223 | |
| Image Classification | ImageNet Domain Generalization (Source: ImageNet, Targets: ImageNetV2, ImageNet-Sketch, ImageNet-A, ImageNet-R) (test) | Accuracy (ImageNetV2)64.35 | 105 |