arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2102.07332cs.LGcs.AI
收藏

归因掩码:通过递归地将注意力集中在DNN的输入上来过滤掉不相关的特征

Attribution Mask: Filtering Out Irrelevant Features By Recursively Focusing Attention on Inputs of DNNs

Hanyang University(汉阳大学)

Jae-Hong Lee, Joon-Hyuk Chang

摘要:

归因方法通过突出显示输入特征的重要部分来计算属性,从而直观地解释深度神经网络(DNN)的预测。特别是,基于梯度的属性(GBA)方法被广泛使用,因为它们可以很容易地通过自动微分实现。在这项研究中,我们使用过滤掉输入特征中不相关部分的属性,然后通过测量预训练DNN的分类准确性来验证这种方法的有效性。这是通过计算并将\textit{attribution mask}应用于输入特征并随后将掩蔽的特征引入DNN来实现的,为此,掩码被设计为递归地将注意力集中在与目标标签相关的输入部分上。在一定条件下提高了精度,\textit{no implicit bias},它可以基于我们对将DNN压缩成单层神经网络的理论见解而导出。我们还提供梯度\,*\,输入符号(GxSI)来获得进一步提高准确性的属性掩码。作为一个例子,在CIFAR-10上,使用从GxSI获得的属性掩码进行修改,我们在没有额外训练的情况下实现了99.8%到99.9%的准确率。

英文摘要:

Attribution methods calculate attributions that visually explain the predictions of deep neural networks (DNNs) by highlighting important parts of the input features. In particular, gradient-based attribution (GBA) methods are widely used because they can be easily implemented through automatic differentiation. In this study, we use the attributions that filter out irrelevant parts of the input features and then verify the effectiveness of this approach by measuring the classification accuracy of a pre-trained DNN. This is achieved by calculating and applying an \textit{attribution mask} to the input features and subsequently introducing the masked features to the DNN, for which the mask is designed to recursively focus attention on the parts of the input related to the target label. The accuracy is enhanced under a certain condition, i.e., \textit{no implicit bias}, which can be derived based on our theoretical insight into compressing the DNN into a single-layer neural network. We also provide Gradient\,*\,Sign-of-Input (GxSI) to obtain the attribution mask that further improves the accuracy. As an example, on CIFAR-10 that is modified using the attribution mask obtained from GxSI, we achieve the accuracy ranging from 99.8\% to 99.9\% without additional training.

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

↑