モデルの予測を理解することは、モデルの正確性の迅速なリアルタイム検証を容易にし、交絡変数を悪用するモデルの使用を防ぐために、ヘルスケアにおいて重要です。説明可能なモデルの必要性に動機付けられて、ボリューム医用画像における説明可能な複数の異常分類の挑戦的なタスクに対処します。クエリされた異常ごとに各ボリューム内の関連領域を強調表示する新しい注意メカニズム、HiResCAMを提案します。 HiResCAMと一般的なモデル説明方法Grad-CAMとの関係を調査し、HiResCAMが異常の位置特定でより優れたパフォーマンスをもたらし、基礎となるモデルにより忠実な説明を生成することを示します。最後に、HiResCAMを活用して、異常が発生している臓器のみに基づいてモデルが異常を予測することを要求するマスク損失を紹介します。私たちの革新により、説明の質が37%向上し、36,316CTボリュームのRAD-ChestCTデータセットにおける異常の最先端の弱く監視された臓器の位置特定が実現します。また、PASCAL VOC 2012で、自然画像上のHiResCAMとGrad-CAMのさまざまなプロパティを示します。全体として、この作業は、畳み込みニューラルネットワークの説明アプローチと体積医療画像における複数異常モデリングの臨床的適用性を向上させます。
Understanding model predictions is critical in healthcare, to facilitate rapid real-time verification of model correctness and to guard against the use of models that exploit confounding variables. Motivated by the need for explainable models, we address the challenging task of explainable multiple abnormality classification in volumetric medical images. We propose a novel attention mechanism, HiResCAM, that highlights relevant regions within each volume for each abnormality queried. We investigate the relationship between HiResCAM and the popular model explanation method Grad-CAM, and demonstrate that HiResCAM yields better performance on abnormality localization and produces explanations that are more faithful to the underlying model. Finally, we introduce a mask loss that leverages HiResCAM to require the model to predict abnormalities based on only the organs in which those abnormalities appear. Our innovations achieve a 37% improvement in explanation quality, resulting in state-of-the-art weakly supervised organ localization of abnormalities in the RAD-ChestCT data set of 36,316 CT volumes. We also demonstrate on PASCAL VOC 2012 the different properties of HiResCAM and Grad-CAM on natural images. Overall, this work advances convolutional neural network explanation approaches and the clinical applicability of multi-abnormality modeling in volumetric medical images.