モデルの予測を理解することは、モデルの正確性の迅速な検証を促進し、交絡変数を悪用するモデルの使用を防ぐために、ヘルスケアにおいて重要です。モデルが各異常を予測するために使用される領域を示さなければならない、体積医療画像における説明可能な複数の異常分類の挑戦的な新しいタスクを紹介します。このタスクを解決するために、各異常のトップ スライスの識別を可能にする複数インスタンス学習畳み込みニューラル ネットワーク、AxialNet を提案します。次に、注目メカニズムである HiResCAM を組み込み、サブスライス領域を識別します。 AxialNet の場合、HiResCAM の説明はモデルが使用した場所を反映することが保証されていることを証明します。Grad-CAM は無関係な場所を強調表示することがあります。忠実な説明を生成するモデルを武器に、HiResCAM と 3D 許可領域を活用する新しいマスク損失を通じてモデルの学習を改善し、異常が現れる臓器のみに基づいてモデルが異常を予測することを目指します。 3D 許可領域は、放射線レポートから抽出された位置情報と、形態学的画像処理によって取得された臓器セグメンテーション マップを組み合わせた新しいアプローチ PARTITION によって自動的に取得されます。全体として、体積医療画像における説明可能な複数の異常を予測するための最初のモデルを提案し、マスク損失を使用して、状態を表す 36,316 スキャンの RAD-ChestCT データセットで複数の異常の臓器局在化を 33% 改善しました。アートの。この作業は、胸部 CT ボリュームにおける複数の異常モデリングの臨床適用性を向上させます。
Understanding model predictions is critical in healthcare, to facilitate rapid verification of model correctness and to guard against use of models that exploit confounding variables. We introduce the challenging new task of explainable multiple abnormality classification in volumetric medical images, in which a model must indicate the regions used to predict each abnormality. To solve this task, we propose a multiple instance learning convolutional neural network, AxialNet, that allows identification of top slices for each abnormality. Next we incorporate HiResCAM, an attention mechanism, to identify sub-slice regions. We prove that for AxialNet, HiResCAM explanations are guaranteed to reflect the locations the model used, unlike Grad-CAM which sometimes highlights irrelevant locations. Armed with a model that produces faithful explanations, we then aim to improve the model's learning through a novel mask loss that leverages HiResCAM and 3D allowed regions to encourage the model to predict abnormalities based only on the organs in which those abnormalities appear. The 3D allowed regions are obtained automatically through a new approach, PARTITION, that combines location information extracted from radiology reports with organ segmentation maps obtained through morphological image processing. Overall, we propose the first model for explainable multi-abnormality prediction in volumetric medical images, and then use the mask loss to achieve a 33% improvement in organ localization of multiple abnormalities in the RAD-ChestCT data set of 36,316 scans, representing the state of the art. This work advances the clinical applicability of multiple abnormality modeling in chest CT volumes.