arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2304.06258cs.CVcs.LGeess.IV

MProtoNet:基于案例的可解释三维多参数磁共振成像脑肿瘤分类模型

MProtoNet: A Case-Based Interpretable Model for Brain Tumor Classification with 3D Multi-parametric Magnetic Resonance Imaging

Southern University of Science and Technology(南方科技大学);The University of British Columbia(不列颠哥伦比亚大学)

Yuanyuan Wei, Roger Tam, Xiaoying Tang

AI总结:

本文提出首个医学原型网络MProtoNet,通过软掩码注意力模块和online-CAM损失将ProtoPNet扩展至三维mpMRI脑肿瘤分类,在无需人工标注的情况下显著提升可解释性。

摘要:

深度卷积神经网络在医学影像中的近期应用引发了对其可解释性的担忧。虽然大多数可解释深度学习应用使用事后方法(如GradCAM)生成特征归因图,但出现了一类新的基于案例的推理模型,即ProtoPNet及其变体,它们在训练过程中识别原型,并将输入图像块与这些原型进行比较。我们提出了首个医学原型网络(MProtoNet),将ProtoPNet扩展到使用三维多参数磁共振成像(mpMRI)数据进行脑肿瘤分类。为了应对二维自然图像与三维mpMRI之间的不同需求,尤其是在定位注意力区域方面,引入了一个带有软掩码和online-CAM损失的新注意力模块。软掩码有助于锐化注意力图,而online-CAM损失在训练注意力模块时直接利用图像级标签。与GradCAM和几种ProtoPNet变体相比,MProtoNet在训练过程中无需人工标注标签,在正确性和定位一致性方面的可解释性指标上取得了统计显著改进(最佳激活精度为$0.713\pm0.058$)。源代码可在https://github.com/aywi/mprotonet获取。

英文摘要:

Recent applications of deep convolutional neural networks in medical imaging raise concerns about their interpretability. While most explainable deep learning applications use post hoc methods (such as GradCAM) to generate feature attribution maps, there is a new type of case-based reasoning models, namely ProtoPNet and its variants, which identify prototypes during training and compare input image patches with those prototypes. We propose the first medical prototype network (MProtoNet) to extend ProtoPNet to brain tumor classification with 3D multi-parametric magnetic resonance imaging (mpMRI) data. To address different requirements between 2D natural images and 3D mpMRIs especially in terms of localizing attention regions, a new attention module with soft masking and online-CAM loss is introduced. Soft masking helps sharpen attention maps, while online-CAM loss directly utilizes image-level labels when training the attention module. MProtoNet achieves statistically significant improvements in interpretability metrics of both correctness and localization coherence (with a best activation precision of $0.713\pm0.058$) without human-annotated labels during training, when compared with GradCAM and several ProtoPNet variants. The source code is available at https://github.com/aywi/mprotonet.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑