提出量化模型关注阿尔茨海默病病灶区域的方法,提升可解释性。
A Quantitative Approach for Evaluating Disease Focus and Interpretability of Deep Learning Models for Alzheimer's Disease Classification
- 用显著图与脑区分割增强模型可解释性
- 提出疾病聚焦分数(DF)量化模型关注病灶区域程度
- 适合关注AI诊断可信度的临床研究者
深度学习(DL)模型在阿尔茨海默病(AD)分类中展现巨大潜力,但其理解和解释仍具挑战,限制了临床应用。尽管显著图等技术能提供视觉线索,但缺乏对模型是否聚焦于已知病理脑区的定量评估。本研究提出一种定量疾病聚焦策略:结合显著图与脑区分割,构建疾病聚焦(DF)分数,量化模型对临床已知的AD相关MRI病灶区域的关注程度。对比了3D ResNet基线模型、预训练MedicalNet及数据增强后的MedicalNet,在区分AD患者与认知正常人群的MRI数据上的表现。结果显示不同模型呈现特征性聚焦模式,尤其是预训练模型和数据增强后模型;该结果揭示了模型聚焦行为与其分类性能之间的关联。所提方法有助于提升深度学习模型在AD分类中的可解释性,推动其临床诊断应用。代码公开于https://github.com/Liang-lt/ADNI。
原文摘要 · Abstract (English)
Deep learning (DL) models have shown significant potential in Alzheimer's Disease (AD) classification. However, understanding and interpreting these models remains challenging, which hinders the adoption of these models in clinical practice. Techniques such as saliency maps have been proven effective in providing visual and empirical clues about how these models work, but there still remains a gap in understanding which specific brain regions DL models focus on and whether these brain regions are pathologically associated with AD. To bridge such gap, in this study, we developed a quantitative disease-focusing strategy to first enhance the interpretability of DL models using saliency maps and brain segmentations; then we propose a disease-focus (DF) score that quantifies how much a DL model focuses on brain areas relevant to AD pathology based on clinically known MRI-based pathological regions of AD. Using this strategy, we compared several state-of-the-art DL models, including a baseline 3D ResNet model, a pretrained MedicalNet model, and a MedicalNet with data augmentation to classify patients with AD vs. cognitive normal patients using MRI data; then we evaluated these models in terms of their abilities to focus on disease-relevant regions. Our results show interesting disease-focusing patterns with different models, particularly characteristic patterns with the pretrained models and data augmentation, and also provide insight into their classification performance. These results suggest that the approach we developed for quantitatively assessing the abilities of DL models to focus on disease-relevant regions may help improve interpretability of these models for AD classification and facilitate their adoption for AD diagnosis in clinical practice. The code is publicly available at https://github.com/Liang-lt/ADNI.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。