Visual Prompt Tuning
Menglin JiaLuming TangBor-Chun ChenClaire CardieSerge BelongieBharath HariharanSer-Nam Lim
Proposes Visual Prompt Tuning (VPT), a parameter-efficient adaptation technique that prepends learnable tokens to frozen vision transformers, outperforming full fine-tuning across downstream recognition tasks while updating less than one percent of model parameters.
Large pre-trained vision Transformers deliver strong results on recognition tasks but adapting them to new problems typically requires full fine-tuning. This stores a separate copy of the entire backbone for every task, an increasingly expensive proposition as models grow to hundreds of millions of parameters.
The article evaluates whether a far lighter approach, Visual Prompt Tuning (VPT), can match or exceed full fine-tuning. VPT freezes the backbone and adds fewer than 1 % extra trainable parameters as task-specific prompt tokens inserted into the input sequence of each Transformer layer.
The authors test both shallow and deep variants of VPT on 24 classification tasks spanning fine-grained recognition and the VTAB-1k benchmark, using supervised and self-supervised ViT and Swin backbones. They compare against full fine-tuning and seven established parameter-efficient baselines, measuring accuracy, total storage across tasks, and behavior across data scales and model sizes. Additional experiments cover semantic segmentation and ConvNet backbones.
VPT-deep surpasses full fine-tuning on 20 of the 24 classification tasks while requiring only 1.18× versus 24× total parameters across all tasks. It also outperforms every other efficient baseline on every task group. Gains hold across data regimes, model scales, and hierarchical Transformers, and remain competitive on segmentation. Random prompt initialization and prompts placed in early layers work best.
These results show that visual prompting can deliver higher accuracy at dramatically lower storage cost than conventional adaptation, making multi-task deployment of large vision models practical. The approach is especially attractive when storage or serving many task-specific models is the bottleneck.
Organizations facing many downstream tasks should pilot VPT-deep on representative workloads and compare storage and accuracy trade-offs against full fine-tuning and bias tuning. Further work is needed to understand why gains are smaller or absent with certain self-supervised pre-training objectives and to extend prompting to additional dense-prediction and video tasks.
The main limitations are that prompt length must be tuned per task, results with self-supervised backbones are inconclusive, and the method offers smaller relative gains on smaller ConvNets. remains high for supervised ViT models on classification benchmarks but lower for other settings until more data are collected.
- Paper: An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale, Alexey Dosovitskiy et al. (2021). Introduces the Vision Transformer (ViT) architecture that serves as the primary frozen backbone evaluated throughout Visual Prompt Tuning.
- Paper: The Power of Scale for Parameter-Efficient Prompt Tuning, Brian Lester et al. (2021). Establishes continuous prompt tuning for Transformer language models, providing the direct conceptual foundation adapted to visual inputs in VPT.
- Paper: Prefix-Tuning: Optimizing Continuous Prompts for Generation, Xiang Lisa Li et al. (2021). Proposes prepending continuous, trainable prefix tokens across Transformer layers, directly inspiring the deep variant of Visual Prompt Tuning.
- Paper: Learning to Prompt for Vision-Language Models, Kaiyang Zhou et al. (2021). Demonstrates prompt learning for vision-language models like CLIP, motivating the extension of prompt tuning directly into vision-only backbones.
- Paper: Swin Transformer: Hierarchical Vision Transformer using Shifted Windows, Ze Liu et al. (2021). Introduces the hierarchical Swin Transformer architecture, which VPT adopts as a key backbone to evaluate prompting across vision Transformer designs.
- Paper: Parameter-Efficient Transfer Learning for NLP, Neil Houlsby et al. (2019). Establishes parameter-efficient transfer learning with bottleneck adapters in Transformers, serving as a primary baseline and motivation for VPT.
- Paper: Towards a Unified View of Parameter-Efficient Transfer Learning, Junxian He et al. (2022). Presents a unified structural taxonomy connecting adapters, prefix tuning, and prompt tuning in frozen Transformers.
- Paper: Training data-efficient image transformers & distillation through attention, Hugo Touvron et al. (2021). Establishes data-efficient training and tokenization recipes for Vision Transformers (DeiT) used extensively in VPT's experiments.
- Paper: Pre-train, Prompt, and Predict: A Systematic Survey of Prompting Methods in Natural Language Processing, Pengfei Liu et al. (2021). Provides a comprehensive taxonomy of continuous and discrete prompt learning techniques foundational to visual prompting methods.
- Paper: Emerging Properties in Self-Supervised Vision Transformers, Mathilde Caron et al. (2021). Introduces DINO self-supervised Vision Transformers, which VPT examines to analyze how pre-training objectives affect prompt tuning efficacy.
- Paper: Conditional Prompt Learning for Vision-Language Models, Kaiyang Zhou et al. (2022). Extends static prompt learning in vision models by conditioning prompt tokens on specific visual inputs to enhance generalization.
