本研究では、3D MRI脳腫瘍分類のための解釈可能なモデルであるMAProtoNetを提案する。MAProtoNetは、既存のプロトタイプ部分ネットワークであるMProtoNetをベースとしており、以下の2つの改善点を持つ:
四重注意層の導入: 従来の三重注意層を3D MRIデータに適応させた四重注意層を導入し、オンラインCAM損失を強化することで、より精度の高い注意マップの生成を実現する。
マルチスケールモジュールの導入: 細粒度の特徴と粗粒度の特徴を融合することで、より精細な注意マップの生成を可能にする。また、マルチスケールマッピング損失を新たに定義し、アフィン変換に対するロバスト性を高める。
実験の結果、MAProtoNetは、BraTS 2018、2019、2020データセットにおいて、既存手法と比較して約2-4%の注意マップ精度の向上を達成した。一方で、分類性能や解釈性も維持できることを示した。
다른 언어로
소스 콘텐츠 기반
arxiv.org
더 깊은 질문