arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2405.03301cs.LGcs.CV
收藏

可解释网络可视化:一种用于CNN图像分类事后可解释性的人机协同方法

Interpretable Network Visualizations: A Human-in-the-Loop Approach for Post-hoc Explainability of CNN-based Image Classification

Politecnico di Milano(米兰理工大学)

Matteo Bianchi, Antonio De Santis, Andrea Tocchetti, Marco Brambilla

AI总结:

提出一种事后可解释性方法,通过逐层特征显著性图、泛化Grad-CAM权重及Sentence-BERT处理的众包文本标签,完整解释CNN特征提取过程并支持全局解释。

摘要:

图像分类的透明度和可解释性对于建立对机器学习模型的信任以及检测偏差和错误至关重要。最先进的可解释性方法生成显著性图来显示特定类别被识别的位置,而没有提供对模型决策过程的详细解释。为了解决这一需求,我们引入了一种事后方法,解释卷积神经网络的整个特征提取过程。这些解释包括模型从输入中提取特征的逐层表示。这些特征表示为由聚类和合并相似特征图生成的显著性图,我们为其关联了一个通过为所提方法泛化Grad-CAM而推导出的权重。为了进一步增强这些解释,我们包含了一组通过游戏化众包活动收集并使用NLP技术和Sentence-BERT处理的文本标签。最后,我们展示了一种通过聚合多张图像的标签来生成全局解释的方法。

英文摘要:

Transparency and explainability in image classification are essential for establishing trust in machine learning models and detecting biases and errors. State-of-the-art explainability methods generate saliency maps to show where a specific class is identified, without providing a detailed explanation of the model's decision process. Striving to address such a need, we introduce a post-hoc method that explains the entire feature extraction process of a Convolutional Neural Network. These explanations include a layer-wise representation of the features the model extracts from the input. Such features are represented as saliency maps generated by clustering and merging similar feature maps, to which we associate a weight derived by generalizing Grad-CAM for the proposed methodology. To further enhance these explanations, we include a set of textual labels collected through a gamified crowdsourcing activity and processed using NLP techniques and Sentence-BERT. Finally, we show an approach to generate global explanations by aggregating labels across multiple images.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑