arXiv:2607.01416cs.CV2026-07中稿 · IJCAI

无需标注即可发现视觉概念,用图结构提升解释性与模型性能

Beyond Heatmaps: Unsupervised Concept-Graph Reasoning for Interpretable Visual Explanation

论文配图:Beyond Heatmaps: Unsupervised Concept-Graph Reasoning for Interpretable Visual Explanation
图 1 · 摘自论文原文
  • 通过非负矩阵分解无监督发现视觉概念,并构建图像级概念图
  • 引入可调阈值过滤弱特征,使模型仅依赖关键概念,最高达0.96准确率
  • 支持概念间非线性推理,解释结果可信且媲美有标注方法

概念瓶颈模型(CBMs)提供了内在可解释的替代方案。然而现有方法通常依赖预定义概念词表或人工标注,缺乏明确的概念定位,且每个概念仅用单一图像级分数表示,忽略了空间重复性和概念间依赖关系。本文提出基于图的概念瓶颈模型(G-CBM),通过非负矩阵分解(NMF)实现无监督概念发现,并将发现的概念作为图像级概念图中的节点。G-CBM将区域特征匹配到这些概念节点,实现概念定位并捕捉其在图像中的重复出现;同时引入可调概念过滤阈值τ,抑制弱区域特征。图注意力网络(GAT)建模节点间的非线性依赖关系,完成概念级推理。在ImageNet、HAM10000、PH2和Derm7pt数据集上,G-CBM相比ResNet-50基线平均提升3.7% AUC。概念过滤常提升预测性能,实现选择性使用概念:在PH2上仅用10个概念中的2个即达0.96的峰值AUC;在HAM10000上仅用9个概念中的3.8个即达0.92。在皮肤镜数据集上,G-CBM表现媲美需外部标注的监督方法。随机消融控制下的删除/插入分析表明,学习到的概念排序与模型预测高度一致。

原文摘要 · Abstract (English)

Concept Bottleneck Models (CBMs) provide an intrinsically interpretable alternative to post-hoc explanations. However, existing CBMs often rely on predefined concept vocabularies or supervised annotations, lack explicit concept grounding, and summarize each concept with a single image-level score -- discarding spatial recurrence and inter-concept dependencies. We propose a Graph-based Concept Bottleneck Model (G-CBM), an intrinsically interpretable framework that performs unsupervised concept discovery via Non-negative Matrix Factorization (NMF) and represents the discovered concepts as nodes in a per-image concept-graph representation. G-CBM matches region-level features to these concept nodes -- providing concept grounding and capturing concept recurrence across the image -- and applies a \emph{tunable concept filtering threshold} $τ$ to suppress weak region-level features. A Graph Attention Network (GAT) then performs concept-level reasoning by modeling nonlinear dependencies across nodes. Across ImageNet, HAM10000, PH2, and Derm7pt, G-CBM achieves an average relative AUC improvement of 3.7\% over a ResNet-50 baseline. Concept filtering frequently improves predictive performance while inducing selective concept use, achieving peak AUC of $0.96$ on PH2 with only 2 of 10 concepts and 0.92 on HAM10000 with 3.8 of 9 concepts. On dermoscopy benchmarks, G-CBM is competitive with supervised approaches requiring external annotations. Deletion/insertion analyses with random ablation controls show that the learned concept ranking faithfully reflects model predictions.

可解释性概念发现图神经网络无监督

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。