arXiv:2510.00773cs.CVcs.AI2025-10中稿 · KDD

提升图像分类可解释性,同时量化预测不确定性

Uncertainty-Aware Concept Bottleneck Models with Enhanced Interpretability

  • 用类别原型距离判断分类结果与置信度
  • 在概念层实现高可解释性且鲁棒的分类决策
  • 适合需要可信推理的医疗、金融等场景

在图像分类中,概念瓶颈模型(CBMs)先将图像映射为人类可理解的概念,再通过可解释的分类器基于这些中间表示预测标签。尽管CBMs具有语义清晰、可解释性强的优点,但其预测性能常低于端到端卷积神经网络。此外,概念预测中的不确定性如何传递至最终标签决策仍未被充分研究。本文提出一种新型的不确定性感知、可解释的第二阶段分类器。该方法学习一组二值类别级概念原型,利用预测概念向量与各原型之间的距离作为分类得分和不确定性度量。这些原型同时充当可解释的分类规则,指示某类预测所需的特征组合。该框架通过基于输入与原型偏离程度的符合性预测,增强了模型对不确定或异常输入的鲁棒性与可解释性。

原文摘要 · Abstract (English)

In the context of image classification, Concept Bottleneck Models (CBMs) first embed images into a set of human-understandable concepts, followed by an intrinsically interpretable classifier that predicts labels based on these intermediate representations. While CBMs offer a semantically meaningful and interpretable classification pipeline, they often sacrifice predictive performance compared to end-to-end convolutional neural networks. Moreover, the propagation of uncertainty from concept predictions to final label decisions remains underexplored. In this paper, we propose a novel uncertainty-aware and interpretable classifier for the second stage of CBMs. Our method learns a set of binary class-level concept prototypes and uses the distances between predicted concept vectors and each class prototype as both a classification score and a measure of uncertainty. These prototypes also serve as interpretable classification rules, indicating which concepts should be present in an image to justify a specific class prediction. The proposed framework enhances both interpretability and robustness by enabling conformal prediction for uncertain or outlier inputs based on their deviation from the learned binary class-level concept prototypes.

可解释性不确定性图像分类

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