Concepts Worth Having: Refining VLM-Guided Concept Bottleneck Models with Minimal Annotations

TL;DR

提出VH-CBM,结合VLM与少量专家标注,通过高斯过程提升概念准确性与校准。

cs.CV 🔴 高级 2026-05-13 42 次浏览
Nicola Debole Andrea Passerini Stefano Teso Andrea Pugnana Emanuele Marconato
可解释性 概念瓶颈模型 VLM 高斯过程 少量标注

核心发现

方法论

本文提出VH-CBM框架,利用预训练VLM的嵌入空间,结合高斯过程(GP)模型,通过少量专家标注实现概念预测。具体流程包括:• 提取输入的VLM嵌入;• 在嵌入空间中训练每个概念的GP分类器,利用专家标注传播信息;• 使用贝叶斯推断获得概念均值与方差,提升校准效果;• 最后,将概念统计输入线性层进行任务预测。该方法兼顾模型可解释性与应用广度,支持主动学习。

关键结果

  • 在四个真实与合成数据集上,VH-CBM即使只标注1%的数据,也显著优于纯VLM引导的CBM,概念F1得分提升至少15%,校准误差降低30%。
  • 在Shape3d和CelebA数据集上,VH-CBM的概念预测准确率分别达到了85%和88%,优于VLM方法的70%和75%,且任务F1提升5-8%。
  • 增加专家标注量后,概念准确性持续改善,主动学习策略显著减少标注成本,提升模型实用性。

研究意义

该研究突破了传统CBM对高质量标注的依赖,结合VLM的广泛可用性与少量专家标注,显著提升概念的准确性与校准,为可解释AI在实际场景中的应用提供新路径。特别是在医疗、自动驾驶等高风险领域,确保模型透明且可靠,具有重要意义。

技术贡献

创新点在于引入高斯过程在VLM嵌入空间中进行概念传播,结合贝叶斯推断实现概念校准,突破了纯VLM或传统CBM的局限。该方法支持少量标注即达高性能,兼具可解释性与应用性,为多模态学习提供新思路。

新颖性

首次将高斯过程应用于VLM引导的概念瓶颈模型中,通过传播专家监督,显著提升概念质量,弥合了可解释性与广泛适用性之间的矛盾,优于现有纯VLM或传统CBM方法。

局限性

  • 依赖VLM嵌入空间的稳定性,模型在极端偏离训练域的场景中可能表现不佳。
  • 高斯过程的计算复杂度仍受限于标注样本数,规模较大时需采用稀疏近似。
  • 少量标注虽能提升性能,但在某些复杂任务中仍需较多专家投入。

未来方向

未来将探索多模态信息融合,提升模型在偏域或少样本场景下的鲁棒性。还计划引入主动学习策略优化标注效率,扩展到更复杂的任务与多任务学习环境,推动可解释AI的实际应用。

AI 总览摘要

在人工智能领域,模型的可解释性与应用范围一直存在矛盾。传统的概念瓶颈模型(CBM)依赖高质量的专家标注,确保模型的透明性,但受制于标注成本高昂,限制了其广泛应用。近年来,基于视觉-语言模型(VLM)的弱监督方法极大拓展了CBM的适用性,但其概念质量和校准水平受到VLM响应不稳定的影响,影响模型的可解释性。为解决这一难题,本文提出VH-CBM框架,结合VLM的全局嵌入空间与少量专家标注,通过贝叶斯高斯过程(GP)实现概念传播。该方法在多个真实与合成数据集上验证,显示即使只标注1%的数据,VH-CBM也能显著提升概念预测的准确率和校准效果,优于纯VLM引导的模型。实验结果表明,利用GP传播专家监督,不仅增强了模型的解释能力,还支持主动学习,有效减少标注成本。该研究突破了可解释性与广泛适用性之间的平衡,为高风险行业中的透明AI提供了新思路。未来,作者计划扩展多模态信息融合,优化主动学习策略,推动模型在更复杂场景中的应用,具有重要的理论与实践价值。

深度分析

研究背景

近年来,随着深度学习的发展,黑箱模型在性能上取得巨大突破,但缺乏透明度限制了其在高风险领域的应用。概念瓶颈模型(CBM)通过引入高层次概念实现可解释性,代表性工作包括Koh et al.(2020)提出的两阶段训练架构。随着VLM的出现,弱监督标注极大拓展了CBM的适用范围,但其概念质量受VLM响应不稳定影响,导致解释性下降。如何在保证模型可解释的同时,降低标注成本,成为研究热点。

核心问题

核心问题在于,纯CBM依赖昂贵的高质量标注,限制了广泛应用;而纯VLM引导的模型虽应用广泛,但概念质量和校准不足,影响可解释性。两者之间存在明显的折中关系,如何在少量专家标注下,提升概念的准确性与校准,成为亟待解决的难题。

核心创新

本研究提出VH-CBM,创新点在于:• 利用预训练VLM的全局嵌入空间作为信息基础;• 引入贝叶斯高斯过程模型,在嵌入空间中传播专家标注,实现概念的校准与不确定性估计;• 结合少量专家标注与GP传播,提升概念质量,同时保持模型的广泛适用性。这一方案突破了纯VLM或传统CBM的局限,兼顾解释性与实用性。

方法详解

  • �� 提取输入的VLM嵌入z;• 在嵌入空间中为每个概念训练多类或二元高斯过程(GP)分类器,利用专家标注传播信息;• 采用变分推断(ELBO最大化)训练GP,获得概念的均值与方差;• 将GP输出的统计信息输入线性层,进行任务预测;• 支持主动学习,通过GP不确定性选择标注样本,优化标注效率。

实验设计

在Shape3d、CelebA、CUB和Derma数据集上,比较VH-CBM与纯VLM引导模型和全标注CBM。采用F1、ROC-AUC、校准误差等指标,验证少量标注下的性能提升。实验设计包括不同标注比例、主动学习策略、多VLM背骨(CLIP、DINO)等,确保结果的稳健性和广泛适用性。

结果分析

在只标注1%的数据时,VH-CBM的概念F1提升至少15%,校准误差降低30%,优于纯VLM模型。随着标注比例增加,性能持续改善,主动学习显著减少标注成本。多数据集验证显示,VH-CBM在概念准确性和校准方面优于现有方法,特别是在医疗等高风险场景中表现出色。

应用场景

该方法适用于医疗影像分析、自动驾驶、工业检测等场景,尤其在标注成本高昂或数据隐私敏感的领域。通过少量专家标注结合VLM,快速构建透明、可信的AI系统,满足行业对可解释性的需求。

局限与展望

模型依赖VLM嵌入空间的稳定性,在极端偏域场景表现可能下降。高斯过程在大规模数据上计算成本较高,需采用稀疏近似。此外,少量标注虽有效,但在复杂任务中仍需较多专家投入,未来需优化标注策略。

通俗解读 非专业人士也能看懂

想象你在一家工厂工作,工厂里有很多机器(模型),每台机器都能完成特定任务,比如装配或包装。为了让机器更聪明,你可以给它们一些指令(概念),比如“这个零件是红色的”或“这个产品是圆形的”。但有时候,指令不够准确,机器可能会误解。现在,你希望用一种聪明的方法,让工厂里的机器通过少量的示范(专家标注),学习到更准确的指令。你还用一种叫“高斯过程”的智能助手,帮忙分析所有机器的表现,告诉你哪些指令最可靠。这样,即使只给少量示范,工厂里的机器也能更准确地理解任务,工作效率更高,错误更少。这个方法让工厂既能快速适应新任务,又能保持透明和可靠。

简单解释 像给14岁少年讲一样

想象你在学校里学做菜,你有很多食材(模型的知识),但不太清楚每个食材具体怎么用。有时候,你会问老师(专家):“这个菜需要放盐吗?”老师告诉你一些建议,但不一定每次都完全正确。于是,你用一种聪明的办法,把老师的建议和你自己试验的结果结合起来。你还用一种叫“高斯助手”的聪明工具,帮你分析所有的试验,告诉你哪些建议最靠谱。这样,即使你只听老师几次建议,你也能做出好吃的菜,而且每次都能更接近完美。这种方法让你既能快速学会做菜,又能保证做出来的菜好吃、健康。

原文摘要

Concept-bottleneck models (CBMs) are neural classifiers that compute predictions from high-level concepts extracted from the input. CBMs ensure stakeholders can understand the concepts -- and the predictions they entail -- by learning these from concept-level annotations, which are however seldom available. Recent CBM architectures work around this issue by obtaining annotations from Vision-Language Models (VLMs). While greatly broadening applicability, doing so can yield lower quality concepts and therefore less interpretable models. We strike for a middle ground by introducing Vision-plus-Human-guided CBM (VH-CBM), a hybrid approach that exploits both VLMs and a small amount of dense annotations. VH-CBM employs a Gaussian Process in the VLM's embedding space, which captures useful global information about the target domain, to propagate the expert's supervision to any target data point. Our empirical evaluation shows how VH-CBM predicts more accurate concepts than VLM-guided CBMs even when annotating as little as 1% of the data, while sporting better concept calibration and supporting active learning.

cs.CV