3D Convolutional Neural Networks for Classification of Alzheimer’s and Parkinson’s Disease with T1-Weighted Brain MRI

Parkinson’s disease (PD) and Alzheimer’s disease (AD) are progressive neurodegenerative disorders that affect millions of people worldwide. In this work, we propose a deep learning approach to classify these diseases based on 3D T1-weighted brain MRI. We analyzed several datasets including the Parkinson’s Progression Markers Initiative (PPMI), an independent dataset from the University of Pennsylvania School of Medicine (UPenn), the Alzheimer’s Disease Neuroimaging Initiative (ADNI), and the Open Access Series of Imaging Studies (OASIS) dataset. The UPenn and OASIS datasets were used as independent test sets to evaluate the model performance during inference. We also implemented a random forest classifier as a baseline model by extracting key radiomics features from the same T1-weighted MRI scans. The proposed 3D convolutional neural network (CNN) model was trained from scratch for the classification tasks. For AD classification, the 3D CNN model achieved an ROC-AUC of 0.878 on the ADNI test set and an average ROC-AUC of 0.789 on the OASIS dataset. For PD classification, the proposed 3D CNN model achieved an ROC-AUC of 0.667 on the PPMI test set and an average ROC-AUC of 0.743 on the UPenn dataset. Model performance was largely maintained when using only 25% of the training dataset. The 3D CNN outperformed the random forest classifier for both the PD and AD tasks. The 3D CNN also generalized better on unseen MRI data from different imaging centers. These approaches show promise for screening of PD and AD patients using only T1-weighted brain MRI, which is relatively widely available. This model with additional validation could also be used to help differentiate between challenging cases of AD and PD when they present with similarly subtle motor and non-motor symptoms.

[1]  Jerry L Prince,et al.  Unsupervised MR harmonization by learning disentangled representations using information bottleneck theory , 2021, NeuroImage.

[2]  Joanna K. Bright,et al.  International Multicenter Analysis of Brain Structure Across Clinical Stages of Parkinson's Disease , 2021, Movement disorders : official journal of the Movement Disorder Society.

[3]  Paul M. Thompson,et al.  Membership Inference Attacks on Deep Regression Models for Neuroimaging , 2021, MIDL.

[4]  N. Jahanshad,et al.  Style Transfer Using Generative Adversarial Networks for Multi-Site MRI Harmonization , 2021, bioRxiv.

[5]  Greg Ver Steeg,et al.  Improved Brain Age Estimation With Slice-Based Set Networks , 2021, 2021 IEEE 18th International Symposium on Biomedical Imaging (ISBI).

[6]  Paul M. Thompson,et al.  3D Grid-Attention Networks for Interpretable Age and Alzheimer's Disease Prediction from Structural MRI. , 2020, 2011.09115.

[7]  Paul M. Thompson,et al.  A Practical Alzheimer Disease Classifier via Brain Imaging-Based Deep Learning on 85,721 Samples , 2020, bioRxiv.

[8]  Paul M. Thompson,et al.  Accurate brain age prediction using recurrent slice-based networks , 2020, bioRxiv.

[9]  Joseph Paul Cohen,et al.  Uniformizing Techniques to Process CT scans with 3D CNNs for Tuberculosis Prediction , 2020, PRIME@MICCAI.

[10]  Joanna K. Bright,et al.  An International Multicenter Analysis of Brain Structure across Clinical Stages of Parkinson's Disease: The ENIGMA-Parkinson's Study , 2020, medRxiv.

[11]  Andrea Vedaldi,et al.  Accurate brain age prediction with lightweight deep neural networks , 2019, bioRxiv.

[12]  Irene Cheng,et al.  Assessing the Capability of Deep-Learning Models in Parkinson's Disease Diagnosis , 2019, ICSM.

[13]  Luca Citi,et al.  Generalization Performance of Deep Learning Models in Neurodegenerative Disease Classification , 2019, 2019 IEEE International Conference on Bioinformatics and Biomedicine (BIBM).

[14]  C. M. Sujatha,et al.  Deep learning based diagnosis of Parkinson’s disease using convolutional neural network , 2019, Multimedia Tools and Applications.

[15]  David Bonekamp,et al.  Automated brain extraction of multisequence MRI using artificial neural networks , 2019, Human brain mapping.

[16]  Jyoti Islam,et al.  A Novel Deep Learning Based Multi-class Classification Method for Alzheimer's Disease Detection Using Brain MRI Data , 2017, BI.

[17]  Andriy Fedorov,et al.  Computational Radiomics System to Decode the Radiographic Phenotype. , 2017, Cancer research.

[18]  Nick C Fox,et al.  Magnetic resonance imaging in Alzheimer's Disease Neuroimaging Initiative 2 , 2015, Alzheimer's & Dementia.

[19]  Jimmy Ba,et al.  Adam: A Method for Stochastic Optimization , 2014, ICLR.

[20]  Arno Klein,et al.  Large-scale evaluation of ANTs and FreeSurfer cortical thickness measurements , 2014, NeuroImage.

[21]  Brian B. Avants,et al.  N4ITK: Improved N3 Bias Correction , 2010, IEEE Transactions on Medical Imaging.

[22]  André L. V. Coelho,et al.  Classification with Imbalanced Data , 2015 .

[23]  P. Rabins,et al.  Dementia , 2008, Annals of Internal Medicine.

[24]  W. Youden,et al.  Index for rating diagnostic tests , 1950, Cancer.