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 のデータセットに関する実験結果は、ST-ProtoPNet が最先端の分類精度と解釈可能性の結果を達成することを示しています。また、提案されたサポート プロトタイプは、背景領域よりも関心のあるオブジェクトによりよくローカライズされる傾向があることも示しています。
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 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 inferior classification accuracy. In this paper, we aim to improve the classification 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 results with a new model, named ST-ProtoPNet, which exploits our support prototypes and the trivial prototypes to provide more effective classification. Experimental results on CUB-200-2011, Stanford Cars, and Stanford Dogs datasets demonstrate that ST-ProtoPNet achieves state-of-the-art classification accuracy and interpretability results. We also show that the proposed support prototypes tend to be better localised in the object of interest rather than in the background region.
updated: Mon Mar 13 2023 06:37:31 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アソシエイト