arXiv reaDer
HiResCAM:説明可能な3D医用画像分類のための視覚的注意における忠実な位置表現
HiResCAM: Faithful Location Representation in Visual Attention for Explainable 3D Medical Image Classification
モデルの予測を理解することは、モデルの正確性の迅速な検証を容易にし、交絡変数を悪用するモデルの使用を防ぐために、ヘルスケアにおいて重要です。ここでは、ボリューム医療画像の説明可能なマルチラベル分類の挑戦的な新しいタスクに対処します。最初に、一般的なモデル説明方法Grad-CAMのこれまで認識されていなかった制限について説明します。勾配平均化ステップの副作用として、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 verification of model correctness and to guard against the use of models that exploit confounding variables. Here we address the challenging new task of explainable multilabel classification of volumetric medical images. We first illustrate a previously unrecognized limitation of the popular model explanation method Grad-CAM: as a side effect of the gradient averaging step, Grad-CAM sometimes highlights the wrong location. To solve this problem, we propose HiResCAM, a novel label-specific attention mechanism that is guaranteed to highlight only the locations the model used to make each prediction. Next, we introduce a mask loss that leverages HiResCAM to encourage the model to predict abnormalities based only on the organs in which those abnormalities appear. Our innovations result in a 37% improvement in explanation quality, resulting in state-of-the-art weakly supervised organ localization of multiple 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 multiple abnormality modeling in volumetric medical images.
updated: Fri Mar 26 2021 00:07:20 GMT+0000 (UTC)
published: Tue Nov 17 2020 19:26:14 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アソシエイト