arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2505.02406cs.CV
收藏

视觉提示需要令牌协调提示注意力

Token Coordinated Prompt Attention is Needed for Visual Prompting

Amazon.com, Inc(亚马逊公司)

Zichen Liu, Xu Zou, Gang Hua, Jiahuan Zhou

AI总结:

针对视觉提示中共享提示忽视令牌差异的问题,本文提出令牌协调提示注意力模块TCPA,通过分离CLS与图像提示并自动匹配令牌,提升ViT微调的特征判别力和多样性。

摘要:

视觉提示技术被广泛用于高效微调预训练的视觉Transformer(ViT),其方式是为所有令牌学习一小组共享提示。然而,现有方法忽视了不同令牌在传递判别信息中的独特作用,并使用相同的提示与所有令牌交互,从而限制了ViT的表示能力。这通常导致提取的提示特征难以区分且存在偏差,阻碍了性能提升。为解决这一问题,我们提出了一种即插即用的令牌协调提示注意力(TCPA)模块,该模块为不同令牌分配特定的协调提示以进行基于注意力的交互。首先,认识到CLS令牌和图像令牌的不同功能——全局信息聚合和局部特征提取,我们将提示解耦为CLS提示和图像提示,它们分别通过注意力机制仅与CLS令牌和图像令牌交互。这增强了它们各自的判别能力。此外,由于不同的图像令牌对应不同的图像块并包含多样化的信息,我们采用匹配函数自动为各个令牌分配协调提示。这使得注意力交互更加精确,提高了提取特征的多样性和表示能力。在多个基准上的大量实验表明,TCPA显著增强了提取特征的多样性和判别力。代码可在https://github.com/zhoujiahuan1991/ICML2025-TCPA获取。

英文摘要:

Visual prompting techniques are widely used to efficiently fine-tune pretrained Vision Transformers (ViT) by learning a small set of shared prompts for all tokens. However, existing methods overlook the unique roles of different tokens in conveying discriminative information and interact with all tokens using the same prompts, thereby limiting the representational capacity of ViT. This often leads to indistinguishable and biased prompt-extracted features, hindering performance. To address this issue, we propose a plug-and-play Token Coordinated Prompt Attention (TCPA) module, which assigns specific coordinated prompts to different tokens for attention-based interactions. Firstly, recognizing the distinct functions of CLS and image tokens-global information aggregation and local feature extraction, we disentangle the prompts into CLS Prompts and Image Prompts, which interact exclusively with CLS tokens and image tokens through attention mechanisms. This enhances their respective discriminative abilities. Furthermore, as different image tokens correspond to distinct image patches and contain diverse information, we employ a matching function to automatically assign coordinated prompts to individual tokens. This enables more precise attention interactions, improving the diversity and representational capacity of the extracted features. Extensive experiments across various benchmarks demonstrate that TCPA significantly enhances the diversity and discriminative power of the extracted features. The code is available at https://github.com/zhoujiahuan1991/ICML2025-TCPA.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑