arXiv reaDer
本質的に解釈可能な CNN とビジョン トランスフォーマーの B-cos アライメント
B-cos Alignment for Inherently Interpretable CNNs and Vision Transformers
私たちは、トレーニング中に重みと入力の調整を促進することでディープ ニューラル ネットワーク (DNN) の解釈可能性を高めるための新しい方向性を提案します。このため、DNN の線形変換を新しい B-cos 変換で置き換えることを提案します。示すように、このような変換のシーケンス (ネットワーク) は、完全なモデルの計算を忠実に要約する単一の線形変換を引き起こします。さらに、B-cos 変換は、最適化中に重みが関連信号と一致するように設計されています。その結果、これらの誘導された線形変換は高度に解釈可能になり、タスクに関連した特徴が強調表示されます。重要なのは、B-cos 変換は既存のアーキテクチャと互換性があるように設計されており、コンピュータ ビジョン用の事実上すべての最新のモデル (ResNets、DenseNets、ConvNext モデルなど) に簡単に統合できることを示しています。ビジョン トランスフォーマー - ImageNet 上で同様の精度を維持しながら、B-cos ベースの説明を正規化レイヤーとアテンション レイヤーと組み合わせることで実現します。最後に、結果として得られる説明は視覚的な品質が高く、定量的な解釈可能性の指標の下で良好に機能することを示します。
We present a new direction for increasing the interpretability of deep neural networks (DNNs) by promoting weight-input alignment during training. For this, we propose to replace the linear transformations in DNNs by our novel B-cos transformation. As we show, a sequence (network) of such transformations induces a single linear transformation that faithfully summarises the full model computations. Moreover, the B-cos transformation is designed such that the weights align with relevant signals during optimisation. As a result, those induced linear transformations become highly interpretable and highlight task-relevant features. Importantly, the B-cos transformation is designed to be compatible with existing architectures and we show that it can easily be integrated into virtually all of the latest state of the art models for computer vision - e.g. ResNets, DenseNets, ConvNext models, as well as Vision Transformers - by combining the B-cos-based explanations with normalisation and attention layers, all whilst maintaining similar accuracy on ImageNet. Finally, we show that the resulting explanations are of high visual quality and perform well under quantitative interpretability metrics.
updated: Mon Jan 15 2024 09:13:05 GMT+0000 (UTC)
published: Mon Jun 19 2023 12:54:28 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アソシエイト