arXiv reaDer
解釈可能な画像分類のための学習サポートと自明なプロトタイプ
Learning Support and Trivial Prototypes for Interpretable Image Classification
プロトタイプ パーツ ネットワーク (ProtoPNet) メソッドは、予測をトレーニング プロトタイプのセットに関連付けることで、解釈可能な分類を実現するように設計されています。トレーニング プロトタイプは、分類から遠く離れた場所にあるようにトレーニングされているため、単純な (学習しやすい) プロトタイプと呼ばれます。特徴空間の境界。 ProtoPNet とサポート ベクター マシン (SVM) の分類は、一連のトレーニング ポイント (つまり、ProtoPNet では自明なプロトタイプ、SVM ではサポート ベクター) との類似性を計算することに依存しているため、ProtoPNet とサポート ベクター マシン (SVM) を類似させることができることに注意してください。ただし、自明なプロトタイプは分類境界から遠く離れた位置にありますが、サポート ベクターはこの境界の近くに位置しており、十分に確立された SVM 理論とのこの不一致により、分類精度が最適ではない ProtoPNet モデルになる可能性があると主張します。この論文では、SVM理論によって提案されているように、特徴空間の分類境界の近くにあるサポートプロトタイプを学習する新しい方法でProtoPNetの分類精度を向上させることを目指しています。さらに、ST-ProtoPNet という名前の新しいモデルを使用して、分類の解釈可能性の向上を目指しています。このモデルは、サポート プロトタイプと自明なプロトタイプを活用して、解釈可能性の補完的な情報を提供します。 CUB-200-2011、Stanford Cars、および Stanford Dogs のデータセットに関する実験結果は、提案された方法が最先端の分類精度を達成し、より視覚的に意味のある多様なプロトタイプを生成することを示しています。
Prototypical part network (ProtoPNet) methods have been designed to achieve interpretable classification by associating predictions with a set of training prototypes, which we refer to as trivial (i.e., easy-to-learn) prototypes because they are trained to lie far from the classification boundary in the feature space. Note that it is possible to make an analogy between ProtoPNet and support vector machine (SVM) given that the classification from both methods relies on computing similarity with a set of training points (i.e., trivial prototypes in ProtoPNet, and support vectors in SVM). However, while trivial prototypes are located far from the classification boundary, support vectors are located close to this boundary, and we argue that this discrepancy with the well-established SVM theory can result in ProtoPNet models with suboptimal classification accuracy. In this paper, we aim to improve the classification accuracy of ProtoPNet with a new method to learn support prototypes that lie near the classification boundary in the feature space, as suggested by the SVM theory. In addition, we target the improvement of classification interpretability with a new model, named ST-ProtoPNet, which exploits our support prototypes and the trivial prototypes to provide complementary interpretability information. Experimental results on CUB-200-2011, Stanford Cars, and Stanford Dogs datasets demonstrate that the proposed method achieves state-of-the-art classification accuracy and produces more visually meaningful and diverse prototypes.
updated: Sun Jan 08 2023 09:27:41 GMT+0000 (UTC)
published: Sun Jan 08 2023 09:27:41 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アソシエイト