arXiv:2503.06901cs.CVcs.LG2025-03ICCV被引 12

通过动态调整提示位置,让视觉提示调优更适配不同任务

PRO-VPT: Distribution-Adaptive Visual Prompt Tuning via Prompt Relocation

  • 基于嵌套优化设计提示重定位策略,动态分配提示令牌
  • 在VTAB-1k和FGVC上比传统VPT提升2.0个百分点准确率
  • 适合需要高效微调大模型的视觉任务研究者

视觉提示调优(VPT)通过微调少量轻量级提示标记,为适应预训练模型到下游任务提供高效方案。然而,以往方法在不同任务间使用固定提示分布,忽视了各模块对提示需求的差异。本文提出自适应分布优化(ADO),通过实证分析确认合理调整分布可显著提升性能,并发现ADO与VPT存在嵌套关系。基于此,我们提出PRO-VPT(基于迭代提示重定位的VPT),通过两步策略:从提示饱和块中剔除闲置提示,再将它们分配给最需要的块。通过迭代执行提示重定位与VPT,实现嵌套优化下的最优提示分布学习,充分释放VPT潜力。大量实验表明,PRO-VPT在VTAB-1k和FGVC基准上分别优于VPT 1.6和2.0个百分点,达到提示类方法当前最佳性能。代码已开源。

原文摘要 · Abstract (English)

Visual prompt tuning (VPT), i.e., fine-tuning some lightweight prompt tokens, provides an efficient and effective approach for adapting pre-trained models to various downstream tasks. However, most prior art indiscriminately uses a fixed prompt distribution across different tasks, neglecting the importance of each block varying depending on the task. In this paper, we introduce adaptive distribution optimization (ADO) by tackling two key questions: (1) How to appropriately and formally define ADO, and (2) How to design an adaptive distribution strategy guided by this definition? Through empirical analysis, we first confirm that properly adjusting the distribution significantly improves VPT performance, and further uncover a key insight that a nested relationship exists between ADO and VPT. Based on these findings, we propose a new VPT framework, termed PRO-VPT (iterative Prompt RelOcation-based VPT), which adaptively adjusts the distribution built upon a nested optimization formulation. Specifically, we develop a prompt relocation strategy derived from this formulation, comprising two steps: pruning idle prompts from prompt-saturated blocks, followed by allocating these prompts to the most prompt-needed blocks. By iteratively performing prompt relocation and VPT, our proposal can adaptively learn the optimal prompt distribution in a nested optimization-based manner, thereby unlocking the full potential of VPT. Extensive experiments demonstrate that our proposal significantly outperforms advanced VPT methods, e.g., PRO-VPT surpasses VPT by 1.6 pp and 2.0 pp average accuracy, leading prompt-based methods to state-of-the-art performance on VTAB-1k and FGVC benchmarks. The code is available at https://github.com/ckshang/PRO-VPT.

视觉提示提示调优自适应分布

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。