本論文では、MCPNetと呼ばれる新しい解釈可能な画像分類器を提案している。MCPNetは、画像の特徴を多層のコンセプトプロトタイプとして表現し、それらの分布を用いて画像を分類する。
具体的には以下の特徴がある:
Centered Kernel Alignment (CKA) lossを用いて、各層の特徴マップを互いに独立性の高い概念セグメントに分割する。これにより、各概念セグメントが異なる意味的特徴を表現するようになる。
各概念セグメントから、重み付きPCAを用いて概念プロトタイプを抽出する。これにより、各概念プロトタイプが画像の特徴を包括的に表現できるようになる。
各画像の概念プロトタイプ応答分布(MCP分布)を計算し、クラス固有のMCP分布との類似度を比較することで画像を分類する。これにより、最終層の特徴のみならず、中間層の特徴も分類に活用できる。
Class-aware Concept Distribution (CCD) lossを導入し、同一クラスの画像のMCP分布は近づき、異なるクラスのMCP分布は離れるよう学習する。これにより、MCP分布を用いた分類が効果的に行えるようになる。
実験の結果、MCPNetは従来手法と比べて同等以上の分類精度を達成しつつ、多層の解釈可能な説明を提供できることが示された。特に、少数サンプルでの分類タスクでも優れた性能を発揮した。
Para Outro Idioma
do conteúdo original
arxiv.org
Principais Insights Extraídos De
by Bor-Shiun Wa... às arxiv.org 04-16-2024
https://arxiv.org/pdf/2404.08968.pdfPerguntas Mais Profundas