arXiv reaDer
MDM: ニューラル ネットワークを視覚的に説明するための複数の動的マスク
MDM: Multiple Dynamic Masks for Visual Explanation of Neural Networks
ニューラル ネットワークのクラス アクティベーション マップ (CAM) ルックアップは、ニューラル ネットワークが決定を下すときに焦点を当てている領域を示します。これまで、CAM の検索方法は、ネットワークの特定の内部モジュールに依存していました。ニューラルネットワークの構造には特定の制約があります。 CAMの検索に汎用性と高性能を持たせる。学習ベースのアルゴリズム、つまり複数の動的マスク (MDM) を提案します。これは、分類に関連する画像のアクティブな特徴のみがニューラル ネットワークの分類結果に影響を与え、その他の特徴はネットワークの分類結果にほとんど影響を与えないという一般的な認識に基づいています。 MDM によって生成されるマスクは、上記の認識に準拠しています。マスク値を制約し、一貫性を有効にすることで、さまざまなサイズのマスク ベクトルをトレーニングします。次に、さまざまなスケールのスタック マスクを使用して、空間情報とセマンティック情報のバランスを取ることができる CAM を生成します。 MDM の結果を最近の高度な CAM 検索方法の結果と比較すると、MDM のパフォーマンスは最先端の結果に達しています。 MDM法を解釈可能なニューラルネットワークProtoPNetとXProtoNetに適用し、説明可能なプロトタイプ検索におけるモデルのパフォーマンスを改善しました。最後に、さまざまなアーキテクチャのニューラル ネットワークに対する MDM の CAM 生成効果を視覚化し、MDM 法の一般性を検証しました。
The Class Activation Map (CAM) lookup of a neural network tells us to which regions the neural network focuses when it makes a decision. In the past, the CAM search method was dependent upon a specific internal module of the network. It has specific constraints on the structure of the neural network. To make the search of CAM have generality and high performance. We propose a learning-based algorithm, namely Multiple Dynamic Masks (MDM). It is based on a public cognition that only active features of a picture related to classification will affect the classification results of the neural network, and other features will hardly affect the classification results of the network. The mask generated by MDM conforms to the above cognition. It trains mask vectors of different sizes by constraining mask values and activating consistency, then it uses stacking masks of different scale to generate CAM that can balance spatial information and semantic information. Comparing the results of MDM with those of the recent advanced CAM search method, the performance of MDM has reached the state of the art results. We applied the MDM method to the interpretable neural networks ProtoPNet and XProtoNet, which improved the performance of model in the explainable prototype search. Finally, we visualized the CAM generation effect of MDM on neural networks of different architectures, verifying the generality of the MDM method.
updated: Tue Aug 09 2022 14:30:56 GMT+0000 (UTC)
published: Sun Jul 17 2022 00:25:16 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト