An interpretable machine learning framework with data-informed imaging biomarkers for diagnosis and prediction of Alzheimer’s disease
Computerized Medical Imaging and Graphics, vol. 128, pp. 102722
Abstract
Machine learning methods based on imaging and other clinical data have shown great potential for improving the early and accurate diagnosis of Alzheimer's disease (AD). However, for most deep learning models, especially those including high-dimensional imaging data, the decision-making process remains largely opaque which limits clinical applicability. Explainable Boosting Machines (EBMs) are inherently interpretable machine learning models, but are typically applied to low-dimensional data. In this study, we propose an interpretable machine learning framework that integrates data-driven feature extraction based on Convolutional Neural Networks (CNNs) with the intrinsic transparency of EBMs for AD diagnosis and prediction. The framework enables interpretation at both the group-level and individual-level by identifying imaging biomarkers contributing to predictions. We validated the framework on the Alzheimer's Disease Neuroimaging Initiative (ADNI) cohort, achieving an area-under-the-curve (AUC) of 0.969 for AD vs. control classification and 0.750 for MCI conversion prediction. External validation was performed on an independent cohort, yielding AUCs of 0.871 for AD vs. subjective cognitive decline (SCD) classification and 0.666 for MCI conversion prediction. The proposed framework achieves performance comparable to state-of-the-art black-box models while offering transparent decision-making, a critical requirement for clinical translation. Our code is available at: https://gitlab.com/radiology/neuro/interpretable_ad_classification.
Authors 12
-
Wenjie Kang corresponding
Affiliation as printed
Biomedical Imaging Group Rotterdam, Department of Radiology & Nuclear Medicine, Erasmus MC, Rotterdam, The Netherlands. Electronic address: w.kang@erasmusmc.nl
-
Harvard University · Erasmus MC
Affiliation as printed
Biomedical Imaging Group Rotterdam, Department of Radiology & Nuclear Medicine, Erasmus MC, Rotterdam, The Netherlands; Harvard Medical School, Boston, MA, USA
-
Affiliation as printed
Department of Neurology, Erasmus MC, Rotterdam, The Netherlands
-
University Medical Center Groningen
Affiliation as printed
Department of Neurology & Alzheimer Center, University Medical Center Groningen, Groningen, The Netherlands
-
University Medical Center Utrecht
Affiliation as printed
Department of Geriatric Medicine, University Medical Center Utrecht, Utrecht, The Netherlands
-
University Medical Center Utrecht
Affiliation as printed
Department of Geriatric Medicine, University Medical Center Utrecht, Utrecht, The Netherlands
-
Radboud University Medical Center
Affiliation as printed
Radboud University Medical Center, Nijmegen, The Netherlands
-
Huub A.M. Middelkoop Aachen
Leiden University · Leiden University Medical Center
Affiliation as printed
Department of Neurology & Neuropsychology, Leiden University Medical Center, Leiden, The Netherlands; Institute of Psychology, Health, Medical and Neuropsychology Unit, Leiden University, The Netherlands
-
Amsterdam University Medical Centers
Affiliation as printed
Amsterdam University Medical Center, location VUmc, Amsterdam, The Netherlands
-
Maastricht University Medical Centre · Maastricht University
Affiliation as printed
Alzheimer Center Limburg, School for Mental Health and Neuroscience (MHeNS), Maastricht University Medical Center, Maastricht, The Netherlands
-
Affiliation as printed
Biomedical Imaging Group Rotterdam, Department of Radiology & Nuclear Medicine, Erasmus MC, Rotterdam, The Netherlands
-
Affiliation as printed
Biomedical Imaging Group Rotterdam, Department of Radiology & Nuclear Medicine, Erasmus MC, Rotterdam, The Netherlands
Cited by 7 stored of 6
7 results