为视觉提示设计分角色注意力机制,提升特征区分度
Token Coordinated Prompt Attention is Needed for Visual Prompting
- 按类别分离提示:为图像块和全局聚合令牌分别设计专用提示
- 自动分配协同提示,使每个图像块获得适配的注意力交互
- 在多个基准测试中显著提升特征多样性与模型性能
视觉提示技术通过学习少量共享提示来高效微调预训练视觉变换器(ViT),但现有方法忽略不同令牌在传递判别信息中的独特作用,对所有令牌使用相同提示,限制了ViT的表征能力。这常导致提示提取的特征难以区分且存在偏差。为此,本文提出即插即用的令牌协调提示注意力(TCPA)模块,为不同令牌分配特定协同提示,实现基于注意力的交互。首先,识别CLS令牌(全局信息聚合)与图像令牌(局部特征提取)的不同功能,将提示解耦为CLS提示和图像提示,分别通过注意力机制与对应令牌交互,增强其判别能力。其次,针对不同图像令牌对应不同图像块、包含多样化信息的特点,引入匹配函数自动为各令牌分配协调提示,实现更精准的注意力交互,提升提取特征的多样性和表征能力。在多个基准上的广泛实验表明,TCPA显著增强了特征的多样性和判别力。代码已公开于https://github.com/zhoujiahuan1991/ICML2025-TCPA。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。