为CLIP设计可解释的视觉提示解耦方法,提升分类性能与可理解性。
Understanding Model Reprogramming for CLIP via Decoupling Visual Prompts
- 将视觉提示按描述特征分组优化,避免单一提示覆盖不足
- 在11个下游数据集上平均表现优于基线,最高提升6.2%
- 提供概率权重机制,揭示每个提示对分类的影响
模型重编程通过仅修改输入和输出空间来适应下游任务。视觉重编程(VR)是视觉任务的一种实例,通过向输入图像添加可训练噪声模式(即视觉提示)来辅助分类。现有针对CLIP的VR方法使用单一视觉提示整合所有类别描述,但学习能力有限,难以捕捉描述中的多样化特征(如形状、颜色、纹理),且可能偏向于区分度低的属性。本文提出解耦-重加权框架:将视觉提示按显式原因(DVP-cse)或无监督聚类(DVP-cls)分组优化,再通过概率重加权矩阵(PRM)融合各提示输出,衡量其对各下游类别的贡献。理论上,该方法降低了经验风险界;实验上,在11个下游数据集上平均表现优于基线,显著提升分类效果。尤其,DVP-PRM可揭示个体提示如何影响决策,提供可解释的重编程分析框架。代码已开源。
原文摘要 · Abstract (English)
Model reprogramming adapts pretrained models to downstream tasks by modifying only the input and output spaces. Visual reprogramming (VR) is one instance for vision tasks that adds a trainable noise pattern (i.e., a visual prompt) to input images to facilitate downstream classification. The existing VR approaches for CLIP train a single visual prompt using all descriptions of different downstream classes. However, the limited learning capacity may result in (1) a failure to capture diverse aspects of the descriptions (e.g., shape, color, and texture), and (2) a possible bias toward less informative attributes that do not help distinguish between classes. In this paper, we introduce a decoupling-and-reweighting framework. Our decoupled visual prompts (DVP) are optimized using descriptions grouped by explicit causes (DVP-cse) or unsupervised clusters (DVP-cls). Then, we integrate the outputs of these visual prompts with a probabilistic reweighting matrix (PRM) that measures their contributions to each downstream class. Theoretically, DVP lowers the empirical risk bound. Experimentally, DVP outperforms baselines on average across 11 downstream datasets. Notably, the DVP-PRM integration enables insights into how individual visual prompts influence classification decisions, providing a probabilistic framework for understanding reprogramming. Our code is available at https://github.com/tmlr-group/DecoupledVP.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。