arXiv:2501.11715cs.CVcs.AI2025-01被引 3

将CNN与可解释模型结合,实现阿尔茨海默病诊断的高精度与可解释性。

GL-ICNN: An End-To-End Interpretable Convolutional Neural Network for the Diagnosis and Prediction of Alzheimer's Disease

  • 用交替训练法融合CNN特征提取与EBM输出,构建端到端可解释模型。
  • 在ADNI数据集上分类AUC达0.956,预测轻度痴呆转为阿尔茨海默病的AUC为0.694。
  • 适合需要透明决策过程的临床医生和医疗AI研发者使用。

基于卷积神经网络(CNN)的深度学习方法在利用影像数据早期准确诊断阿尔茨海默病(AD)方面展现出巨大潜力。然而,这些方法尚未广泛应用于临床,可能因其深度学习模型的可解释性不足。解释性增强机器(EBM)是一种玻璃盒模型,但无法直接从输入影像数据中学习特征。本研究提出一种新型可解释模型,结合CNN与EBM,用于AD的诊断与预测。我们设计了一种创新的交替训练策略,交替训练CNN作为特征提取器、EBM作为输出模块,形成端到端模型。该模型以影像数据为输入,提供预测结果与可解释的特征重要性度量。我们在阿尔茨海默病神经影像倡议(ADNI)数据集和健康-RI帕雷尔诺尔神经退行性疾病生物库(PND)外部测试集上验证了该模型。在ADNI队列中,该模型对AD与对照的分类AUC达到0.956,对轻度认知障碍(MCI)转化为AD的预测AUC为0.694。所提模型为玻璃盒模型,性能与现有顶尖黑箱模型相当。代码已公开:https://anonymous.4open.science/r/GL-ICNN。

原文摘要 · Abstract (English)

Deep learning methods based on Convolutional Neural Networks (CNNs) have shown great potential to improve early and accurate diagnosis of Alzheimer's disease (AD) dementia based on imaging data. However, these methods have yet to be widely adopted in clinical practice, possibly due to the limited interpretability of deep learning models. The Explainable Boosting Machine (EBM) is a glass-box model but cannot learn features directly from input imaging data. In this study, we propose a novel interpretable model that combines CNNs and EBMs for the diagnosis and prediction of AD. We develop an innovative training strategy that alternatingly trains the CNN component as a feature extractor and the EBM component as the output block to form an end-to-end model. The model takes imaging data as input and provides both predictions and interpretable feature importance measures. We validated the proposed model on the Alzheimer's Disease Neuroimaging Initiative (ADNI) dataset and the Health-RI Parelsnoer Neurodegenerative Diseases Biobank (PND) as an external testing set. The proposed model achieved an area-under-the-curve (AUC) of 0.956 for AD and control classification, and 0.694 for the prediction of conversion of mild cognitive impairment (MCI) to AD on the ADNI cohort. The proposed model is a glass-box model that achieves a comparable performance with other state-of-the-art black-box models. Our code is publicly available at: https://anonymous.4open.science/r/GL-ICNN.

阿尔茨海默病可解释AI深度学习医学影像

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。