用贝叶斯方法构建动态标签映射,提升视觉重编程效果
Bayesian-guided Label Mapping for Visual Reprogramming
- 基于贝叶斯概率构建可迭代更新的标签映射矩阵
- 在多个模型上优于现有标签映射方法,显著提升性能
- 适合研究视觉重编程、标签对齐及模型可解释性的读者
视觉重编程(VR)通过调整预训练视觉模型的输入或输出接口,解决下游任务中标签与预训练标签完全不同问题。当调整输出接口时,传统标签映射方法建立无梯度的一一对应关系。本文发现这种一一映射可能忽略预训练标签与下游标签间的复杂关联。为此提出贝叶斯引导的标签映射(BLM)方法:构建一个迭代更新的概率标签映射矩阵,每个元素表示预训练标签与下游标签之间的关联强度,其赋值基于贝叶斯条件概率,考虑下游标签与预训练模型在下游样本上的预测联合分布。在ResNeXt和CLIP等预训练视觉模型上的实验表明,BLM显著优于现有方法。该方法还为理解视觉重编程的有效性提供了概率视角。代码已开源。
原文摘要 · Abstract (English)
Visual reprogramming (VR) leverages the intrinsic capabilities of pretrained vision models by adapting their input or output interfaces to solve downstream tasks whose labels (i.e., downstream labels) might be totally different from the labels associated with the pretrained models (i.e., pretrained labels). When adapting the output interface, label mapping methods transform the pretrained labels to downstream labels by establishing a gradient-free one-to-one correspondence between the two sets of labels. However, in this paper, we reveal that one-to-one mappings may overlook the complex relationship between pretrained and downstream labels. Motivated by this observation, we propose a Bayesian-guided Label Mapping (BLM) method. BLM constructs an iteratively-updated probabilistic label mapping matrix, with each element quantifying a pairwise relationship between pretrained and downstream labels. The assignment of values to the constructed matrix is guided by Bayesian conditional probability, considering the joint distribution of the downstream labels and the labels predicted by the pretrained model on downstream samples. Experiments conducted on both pretrained vision models (e.g., ResNeXt) and vision-language models (e.g., CLIP) demonstrate the superior performance of BLM over existing label mapping methods. The success of BLM also offers a probabilistic lens through which to understand and analyze the effectiveness of VR. Our code is available at https://github.com/tmlr-group/BayesianLM.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。