arXiv:2506.22908cs.CV2025-06ICCV被引 5

通过消除视觉提示调优中的突变性,显著提升模型精度与训练速度。

Attention to the Burstiness in Visual Prompt Tuning!

  • 提出白化预处理,使提示向量分布更接近高斯分布。
  • 在CUB数据集上提升超25个精度点,且训练速度大幅加快。
  • 设计低秩双线性结构,减少参数量同时保持高性能。

视觉提示调优(VPT)是一种参数高效微调方法,通过在输入空间学习少量提示参数来适配预训练视觉Transformer(ViT)。研究发现,图像块嵌入与Transformer自注意力模块中键/查询投影器的交互值存在显著突变性,且块嵌入值呈拉普拉斯分布,键/查询投影器值呈超拉普拉斯分布。这些非高斯分布为提示学习带来挑战。为此,本文提出对数据进行白化处理,消除相关性并均衡方差,使其趋近高斯分布后再学习提示。我们推导出基于随机图像块嵌入和ViT键/查询投影器的白化矩阵,并以双线性方式作用于待学习提示。令人惊讶的是,该方法显著加速了提示调优并大幅提升精度,如在CUB数据集上提升超过25个精度点;有趣的是,它仍能学习到具有突变性的提示。进一步地,针对双线性模型引入突变性的特性,我们提出紧凑的低秩版本,通过学习两个小矩阵的乘积生成最终提示,称为双线性提示调优(BPT)。大量实验证明,BPT不仅优于多种VPT方法,还显著降低参数量与计算开销。

原文摘要 · Abstract (English)

Visual Prompt Tuning (VPT) is a parameter-efficient fune-tuning technique that adapts a pre-trained vision Transformer (ViT) by learning a small set of parameters in the input space, known as prompts. In VPT, we uncover ``burstiness'' in the values arising from the interaction of image patch embeddings, and the key and query projectors within Transformer's self-attention module. Furthermore, the values of patch embeddings and the key and query projectors exhibit Laplacian and hyper-Laplacian distribution, respectively. Intuitively, these non-Gaussian distributions pose challenges for learning prompts. To address this, we propose whitening these data, de-correlating them and equalizing their variance towards more Gaussian before learning prompts. We derive the whitening matrix over random image patch embeddings and ViT's key and query projectors, and multiply it with the prompt to be learned in a bilinear manner. Surprisingly, this method significantly accelerates prompt tuning and boosts accuracy, e.g., $>$25 accuracy points on the CUB dataset; interestingly, it learns ``bursty prompts''. Extending the bilinear model which is known to introduce burstiness, we present a compact, low-rank version by learning two smaller matrices whose multiplication yields the final prompts. We call the proposed methods Bilinear Prompt Tuning (BPT). Extensive experiments across multiple benchmark datasets demonstrate that BPT methods not only outperform various VPT methods but also reduce parameter count and computation overhead.

提示调优视觉Transformer参数效率白化处理

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