アトリビューションマップの視覚化は、畳み込みニューラルネットワークの基礎となる推論プロセスを理解するための最も効果的な手法の1つとして生まれました。このタスクの目標は、最終的なネットワーク出力への寄与に関連する各画像ピクセルのスコアを計算することです。この論文では、ReLUネットワークの区分的線形性を利用してモデル関数をさまざまな線形マッピングに分解する新しい勾配ベースの方法であるDisentangled Masked Backpropagation(DMBP)を紹介します。この分解は、バックプロパゲーション中に各フィルターの寄与をマスクする変数のセットを学習することにより、属性マップから正、負、および迷惑な要因を解きほぐすことを目的としています。標準アーキテクチャ(ResNet50およびVGG16)およびベンチマークデータセット(PASCAL VOCおよびImageNet)に対する徹底的な評価は、DMBPが以前のアプローチよりも視覚的に解釈可能なアトリビューションマップを生成することを示しています。さらに、私たちの方法で作成されたマップが、最終的なネットワーク出力への各ピクセルの真の寄与とより一致していることを定量的に示します。
Attribution map visualization has arisen as one of the most effective techniques to understand the underlying inference process of Convolutional Neural Networks. In this task, the goal is to compute an score for each image pixel related with its contribution to the final network output. In this paper, we introduce Disentangled Masked Backpropagation (DMBP), a novel gradient-based method that leverages on the piecewise linear nature of ReLU networks to decompose the model function into different linear mappings. This decomposition aims to disentangle the positive, negative and nuisance factors from the attribution maps by learning a set of variables masking the contribution of each filter during back-propagation. A thorough evaluation over standard architectures (ResNet50 and VGG16) and benchmark datasets (PASCAL VOC and ImageNet) demonstrates that DMBP generates more visually interpretable attribution maps than previous approaches. Additionally, we quantitatively show that the maps produced by our method are more consistent with the true contribution of each pixel to the final network output.