arXivDaily arXiv每日学术速递 周一至周五更新
稍安勿躁,官方尚未更新
arXiv 2201.11808cs.CVcs.LG
收藏

LAP:一种用于卷积神经网络中基于概念的自解释和知识注入的注意力模块

LAP: An Attention-Based Module for Concept Based Self-Interpretation and Knowledge Injection in Convolutional Neural Networks

Sharif University of Technology(谢里夫理工大学)

Rassa Ghavami Modegh, Ahmad Salimi, Alireza Dizaji, Hamid R. Rabiee

AI总结:

提出可插入已训练卷积神经网络的注意力池化模块LAP,通过弱监督方案学习决策区分特征,在不损失性能下实现基于概念的自解释和知识注入,提供比白盒方法更有效的解释。

摘要:

尽管深度卷积神经网络具有最先进的性能,但它们在未见过的场景中容易产生偏差和发生故障。此外,其推理背后的复杂计算无法被人类理解以建立信任。外部解释器方法试图以人类可理解的方式解释网络决策,但由于其假设和简化,被指责存在谬误。另一方面,模型固有的自解释器虽然对这些谬误更具鲁棒性,但可以应用于已经训练好的模型。在本研究中,我们提出了一种基于注意力的池化模块,称为 Local Attention Pooling (LAP),它在不损失性能的情况下实现了模型的可解释性以及知识注入的可能性。该模块可轻松插入到已经训练好的卷积神经网络中。我们定义了一种弱监督训练方案,用于学习决策中的区分特征,而不依赖专家的标注。我们通过在两个数据集(包括 ImageNet)上评估几个 LAP 扩展模型来验证了我们的主张。所提出的框架提供了比常用的白盒解释器方法更有效且人类可理解的模型内解释。

英文摘要:

Despite the state-of-the-art performance of deep convolutional neural networks, they are susceptible to bias and malfunction in unseen situations. Moreover, the complex computation behind their reasoning is not human-understandable to develop trust. External explainer methods have tried to interpret network decisions in a human-understandable way, but they are accused of fallacies due to their assumptions and simplifications. On the other side, the inherent self-interpretability of models, while being more robust to the mentioned fallacies, cannot be applied to the already trained models. In this work, we propose a new attention-based pooling layer, called Local Attention Pooling (LAP), that accomplishes self-interpretability and the possibility for knowledge injection without performance loss. The module is easily pluggable into any convolutional neural network, even the already trained ones. We have defined a weakly supervised training scheme to learn the distinguishing features in decision-making without depending on experts' annotations. We verified our claims by evaluating several LAP-extended models on two datasets, including ImageNet. The proposed framework offers more valid human-understandable and faithful-to-the-model interpretations than the commonly used white-box explainer methods.

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

↑