Abstract
Machine learning methods have shown large potential for the automatic early diagnosis of Alzheimer’s Disease (AD). However, some machine learning methods based on imaging data have poor interpretability because it is usually unclear how they make their decisions. Explainable Boosting Machines (EBMs) are interpretable machine learning models based on the statistical framework of generalized additive modeling, but have so far only been used for tabular data. Therefore, we propose a framework that combines the strength of EBM with high-dimensional imaging data using deep learning-based feature extraction. The proposed framework is interpretable because it provides the importance of each feature. We validated the proposed framework on the Alzheimer’s Disease Neuroimaging Initiative (ADNI) dataset, achieving accuracy of 0.883 and area-under-the-curve (AUC) of 0.970 on AD and control classification. Furthermore, we validated the proposed framework on an external testing set, achieving accuracy of 0.778 and AUC of 0.887 on AD and subjective cognitive decline (SCD) classification. The proposed framework significantly outperformed an EBM model using volume biomarkers instead of deep learning-based features, as well as an end-to-end convolutional neural network (CNN) with optimized architecture.
Original language | English |
---|---|
Title of host publication | Medical Image Computing and Computer Assisted Intervention – MICCAI 2023 Workshops - ISIC 2023, Care-AI 2023, MedAGI 2023, DeCaF 2023, Held in Conjunction with MICCAI 2023, Proceedings |
Editors | M. Emre Celebi, Md Sirajus Salekin, Hyunwoo Kim, Shadi Albarqouni |
Publisher | Springer Verlag |
Pages | 69-78 |
Number of pages | 10 |
Volume | 14393 |
ISBN (Print) | 9783031474002 |
DOIs | |
Publication status | Published - 1 Jan 2023 |
Event | 26th International Conference on Medical Image Computing and Computer Assisted Intervention - Vancouver, Canada Duration: 8 Oct 2023 → 12 Oct 2023 Conference number: 26 https://conferences.miccai.org/2023/en/ |
Publication series
Series | Lecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics) |
---|---|
Volume | 14393 |
ISSN | 0302-9743 |
Conference
Conference | 26th International Conference on Medical Image Computing and Computer Assisted Intervention |
---|---|
Abbreviated title | MICCAI 2023 |
Country/Territory | Canada |
City | Vancouver |
Period | 8/10/23 → 12/10/23 |
Internet address |
Keywords
- Alzheimer’s disease
- Convolutional neural network
- Explainable boosting machine
- Interpretable AI
- MRI