通过非参数聚类学习语义独特部件,提升图像分类可解释性
Interpretable Image Classification via Non-parametric Part Prototype Learning
- 用非参数聚类从基础视觉模型中提取深度特征,学习每类的独特部件原型
- 在CUB-200-2011等数据集上,解释多样性与全面性得分优于现有方法
- 适合关注模型决策过程透明性的研究人员和实际应用中的可解释需求
图像分类的可解释决策过程是计算机视觉中的长期难题。近年来,原型部件网络因其能基于典型物体部件提供类人视觉推理解释而受到关注。然而,现有方法生成的解释质量仍有提升空间,因原型常聚焦于重复冗余概念。我们提出一种基于部件的可解释图像分类框架,为每类学习一组语义独特的物体部件,实现多样且全面的解释。核心在于通过聚类基础视觉模型提取的深度特征,以非参数方式学习部件原型,这些特征编码了鲁棒语义信息。为定量评估解释质量,引入区分度评分与全面性评分。在CUB-200-2011、Stanford Cars和Stanford Dogs数据集上的实验表明,本框架相较现有ProtoPNets在可解释性方面表现更优。代码已公开:https://github.com/zijizhu/proto-non-param。
原文摘要 · Abstract (English)
Classifying images with an interpretable decision-making process is a long-standing problem in computer vision. In recent years, Prototypical Part Networks has gained traction as an approach for self-explainable neural networks, due to their ability to mimic human visual reasoning by providing explanations based on prototypical object parts. However, the quality of the explanations generated by these methods leaves room for improvement, as the prototypes usually focus on repetitive and redundant concepts. Leveraging recent advances in prototype learning, we present a framework for part-based interpretable image classification that learns a set of semantically distinctive object parts for each class, and provides diverse and comprehensive explanations. The core of our method is to learn the part-prototypes in a non-parametric fashion, through clustering deep features extracted from foundation vision models that encode robust semantic information. To quantitatively evaluate the quality of explanations provided by ProtoPNets, we introduce Distinctiveness Score and Comprehensiveness Score. Through evaluation on CUB-200-2011, Stanford Cars and Stanford Dogs datasets, we show that our framework compares favourably against existing ProtoPNets while achieving better interpretability. Code is available at: https://github.com/zijizhu/proto-non-param.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。