arXiv:2505.23694cs.CV2025-05CVPR被引 26

用语义引导提示分布,提升ViT模型微调效果

DA-VPT: Semantic-Guided Visual Prompt Tuning for Vision Transformers

  • 通过度量学习构建提示与语义数据间的距离关系
  • 在分类与分割任务中均实现性能提升
  • 适合需要高效微调ViT的视觉任务研究者

视觉提示微调(VPT)作为一种参数高效微调方法,通过部分更新可学习提示词来保持大多数模型参数冻结。尽管已有研究尝试调整提示连接结构,但提示与图像块之间的分布关联仍不明确。本文提出分布感知视觉提示微调(DA-VPT),利用度量学习从类别相关语义数据中学习提示间的距离度量,以引导其分布。实验表明,提示可作为图像块与类别标记间共享语义信息的有效桥梁。我们在多个主流识别与分割基准上评估了该方法,结果证明,借助语义信息引导提示学习,能更有效地微调ViT模型,在各类下游视觉任务中实现性能提升。

原文摘要 · Abstract (English)

Visual Prompt Tuning (VPT) has become a promising solution for Parameter-Efficient Fine-Tuning (PEFT) approach for Vision Transformer (ViT) models by partially fine-tuning learnable tokens while keeping most model parameters frozen. Recent research has explored modifying the connection structures of the prompts. However, the fundamental correlation and distribution between the prompts and image tokens remain unexplored. In this paper, we leverage metric learning techniques to investigate how the distribution of prompts affects fine-tuning performance. Specifically, we propose a novel framework, Distribution Aware Visual Prompt Tuning (DA-VPT), to guide the distributions of the prompts by learning the distance metric from their class-related semantic data. Our method demonstrates that the prompts can serve as an effective bridge to share semantic information between image patches and the class token. We extensively evaluated our approach on popular benchmarks in both recognition and segmentation tasks. The results demonstrate that our approach enables more effective and efficient fine-tuning of ViT models by leveraging semantic information to guide the learning of the prompts, leading to improved performance on various downstream vision tasks.

视觉提示ViT微调度量学习

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