arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2407.02418eess.IVcs.CV

AXIAL:基于注意力的可解释性方法,用于在3D MRI脑扫描上使用2D CNN进行可解释的阿尔茨海默病局部化诊断

AXIAL: Attention-based eXplainability for Interpretable Alzheimer's Localized Diagnosis using 2D CNNs on 3D MRI brain scans

University of Cassino and Southern Lazio(卡西诺和南拉齐奥大学);Radboud University Medical Center(拉德堡德大学医学中心)

Gabriele Lozupone, Alessandro Bria, Francesco Fontanella, Frederick J. A. Meijer, Claudio De Stefano

AI总结:

本文提出AXIAL方法,利用软注意力机制使2D CNN从3D MRI中提取体积表示并生成体素级可解释注意力图,在AD诊断和MCI预后任务上显著超越现有方法,并准确识别出海马体等AD相关脑区。

摘要:

本研究提出了一种利用3D MRI进行阿尔茨海默病诊断的创新方法,旨在增强模型决策的可解释性。我们的方法采用软注意力机制,使2D CNN能够提取体积表示。同时,模型学习每个切片在决策中的重要性,从而生成体素级注意力图,以产生可解释的MRI。为了测试我们的方法并确保结果的可重复性,我们选择了来自阿尔茨海默病神经影像学倡议(ADNI)的标准化MRI数据集合。在该数据集上,我们的方法在以下任务中显著优于最先进的方法:(i)区分阿尔茨海默病(AD)与认知正常(CN),准确率为0.856,马修斯相关系数(MCC)为0.712,分别比第二名提高了2.4%和5.3%;(ii)在区分稳定型与进展型轻度认知障碍(MCI)的预后任务中,准确率为0.725,MCC为0.443,分别比第二名提高了10.2%和20.5%。我们通过采用双重迁移学习策略取得了这一预后结果,该策略增强了对形态学变化的敏感性,并促进了早期AD检测。凭借体素级精度,我们的方法识别出了模型关注的具体区域,确定了这些主要脑区:海马体、杏仁核、海马旁回和下侧脑室。所有这些区域在临床上均与AD的发展相关。此外,我们的方法在不同的交叉验证折中一致地发现了相同的AD相关区域,证明了其在突出与已知病理标志物高度一致的区域方面的鲁棒性和精确性。

英文摘要:

This study presents an innovative method for Alzheimer's disease diagnosis using 3D MRI designed to enhance the explainability of model decisions. Our approach adopts a soft attention mechanism, enabling 2D CNNs to extract volumetric representations. At the same time, the importance of each slice in decision-making is learned, allowing the generation of a voxel-level attention map to produce an explainable MRI. To test our method and ensure the reproducibility of our results, we chose a standardized collection of MRI data from the Alzheimer's Disease Neuroimaging Initiative (ADNI). On this dataset, our method significantly outperforms state-of-the-art methods in (i) distinguishing AD from cognitive normal (CN) with an accuracy of 0.856 and Matthew's correlation coefficient (MCC) of 0.712, representing improvements of 2.4% and 5.3% respectively over the second-best, and (ii) in the prognostic task of discerning stable from progressive mild cognitive impairment (MCI) with an accuracy of 0.725 and MCC of 0.443, showing improvements of 10.2% and 20.5% respectively over the second-best. We achieved this prognostic result by adopting a double transfer learning strategy, which enhanced sensitivity to morphological changes and facilitated early-stage AD detection. With voxel-level precision, our method identified which specific areas are being paid attention to, identifying these predominant brain regions: the hippocampus, the amygdala, the parahippocampal, and the inferior lateral ventricles. All these areas are clinically associated with AD development. Furthermore, our approach consistently found the same AD-related areas across different cross-validation folds, proving its robustness and precision in highlighting areas that align closely with known pathological markers of the disease.

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

↑