无需CLIP和标注,直接将视觉模型转为可解释的因果模型。
CLIP-Free, Label Free, Unsupervised Concept Bottleneck Models
- 用文本类别分布对齐替代图像-概念标注,实现无监督映射。
- 在40多个模型上超越有监督的CLIP基线,准确率提升显著。
- 适合需要可解释性又缺乏标注数据的研究与应用。
概念瓶颈模型(CBM)通过将密集特征映射到人类可理解的概念并线性组合预测结果,提升模型可解释性。然而现有方法依赖CLIP获取图像-概念标注,且需人工标注,同时必须训练线性分类器。本文提出一种新方法,可将任意冻结的视觉分类器转化为无需标签、不依赖CLIP、且线性分类器由无监督方式推导的CBM。核心思想是将原始分类器输出的类别分布,与基于文本类别名称生成的视觉-语言分布进行对齐,同时保持原模型性能。该方法无需真实图像-类别标注,数据效率高,并保留原模型推理过程。在超过40个视觉分类器上测试,所提无监督、无标签、无CLIP的CBM(U-F²-CBM)达到新基准,优于有监督的CLIP基线。此外,该方法还可用于零样本图像描述生成,在多个数据集上表现超越现有基于CLIP的方法,达到最先进水平。
原文摘要 · Abstract (English)
Concept Bottleneck Models (CBMs) map dense feature representations into human-interpretable concepts which are then combined linearly to make a prediction. However, modern CBMs rely on the CLIP model to obtain image-concept annotations, and it remains unclear how to design CBMs without the CLIP bottleneck. Methods that do not use CLIP instead require manual, labor intensive annotation to associate feature representations with concepts. Furthermore, all CBMs necessitate training a linear classifier to map the extracted concepts to class labels. In this work, we lift all three limitations simultaneously by proposing a method that converts any frozen visual classifier into a CBM without requiring image-concept labels (label-free), without relying on the CLIP model (CLIP-free), and by deriving the linear classifier in an unsupervised manner. Our method is formulated by aligning the original classifier's distribution (over discrete class indices) with its corresponding vision-language counterpart distribution derived from textual class names, while preserving the classifier's performance. The approach requires no ground-truth image-class annotations, and is highly data-efficient and preserves the classifier's reasoning process. Applied and tested on over 40 visual classifiers, our resulting unsupervised, label-free and CLIP-free CBM (U-F$^2$-CBM) sets a new state of the art, surpassing even supervised CLIP-based CBMs. We also show that our method can be used for zero-shot image captioning, outperforming existing methods based on CLIP, and achieving state-of-art.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。