A Robust Deep Model for Improved Classification of AD/MCI Patients

Accurate classification of Alzheimer's disease (AD) and its prodromal stage, mild cognitive impairment (MCI), plays a critical role in possibly preventing progression of memory impairment and improving quality of life for AD patients. Among many research tasks, it is of a particular interest to identify noninvasive imaging biomarkers for AD diagnosis. In this paper, we present a robust deep learning system to identify different progression stages of AD patients based on MRI and PET scans. We utilized the dropout technique to improve classical deep learning by preventing its weight coadaptation, which is a typical cause of overfitting in deep learning. In addition, we incorporated stability selection, an adaptive learning factor, and a multitask learning strategy into the deep learning framework. We applied the proposed method to the ADNI dataset, and conducted experiments for AD and MCI conversion diagnosis. Experimental results showed that the dropout technique is very effective in AD diagnosis, improving the classification accuracies by 5.9% on average as compared to the classical deep learning methods.

[1]  Thomas Serre,et al.  Categorization by Learning and Combining Object Parts , 2001, NIPS.

[2]  N. Meinshausen,et al.  Stability selection , 2008, 0809.2932.

[3]  M. Greicius,et al.  Default-mode network activity distinguishes Alzheimer's disease from healthy aging: Evidence from functional MRI , 2004, Proc. Natl. Acad. Sci. USA.

[4]  Dinggang Shen,et al.  Robust Deep Learning for Improved Classification of AD/MCI Patients , 2014, MLMI.

[5]  Rich Caruana,et al.  Multitask Learning , 1997, Machine Learning.

[6]  B. Långström,et al.  The use of PET in Alzheimer disease , 2010, Nature Reviews Neurology.

[7]  Vikas Singh,et al.  Predictive markers for AD in a multi-modality framework: An analysis of MCI progression in the ADNI population , 2011, NeuroImage.

[8]  J. Trojanowski,et al.  Prediction of MCI to AD conversion, via MRI, CSF biomarkers, and pattern classification , 2011, Neurobiology of Aging.

[9]  Daoqiang Zhang,et al.  Multimodal classification of Alzheimer's disease and mild cognitive impairment , 2011, NeuroImage.

[10]  Tso-Jung Yen,et al.  Discussion on "Stability Selection" by Meinshausen and Buhlmann , 2010 .

[11]  E. Kolibáš,et al.  P03.443 ADAS-COG (Alzheimer's Disease Assessment Scale-Cognitive subscale)-validation of the Slovak version , 2000, European Psychiatry.

[12]  Juhan Nam,et al.  Multimodal Deep Learning , 2011, ICML.

[13]  Nitish Srivastava,et al.  Dropout: a simple way to prevent neural networks from overfitting , 2014, J. Mach. Learn. Res..

[14]  Yee Whye Teh,et al.  A Fast Learning Algorithm for Deep Belief Nets , 2006, Neural Computation.

[15]  Nitish Srivastava,et al.  Improving neural networks by preventing co-adaptation of feature detectors , 2012, ArXiv.

[16]  R. Tibshirani Regression Shrinkage and Selection via the Lasso , 1996 .

[17]  L. Guse,et al.  An examination of psychometric properties of the mini-mental state examination and the standardized mini-mental state examination: implications for clinical practice. , 2000, Applied nursing research : ANR.

[18]  I. Jolliffe Principal Component Analysis , 2002 .

[19]  Rich Caruana,et al.  Multitask Learning: A Knowledge-Based Source of Inductive Bias , 1993, ICML.

[20]  Corinna Cortes,et al.  Support-Vector Networks , 1995, Machine Learning.

[21]  Yoshua. Bengio,et al.  Learning Deep Architectures for AI , 2007, Found. Trends Mach. Learn..

[22]  Harvey M. Wagner,et al.  Global Sensitivity Analysis , 1995, Oper. Res..

[23]  Jieping Ye,et al.  An accelerated gradient method for trace norm minimization , 2009, ICML '09.

[24]  Dinggang Shen,et al.  Deep Learning-Based Feature Representation for AD/MCI Classification , 2013, MICCAI.

[25]  Pascal Vincent,et al.  Representation Learning: A Review and New Perspectives , 2012, IEEE Transactions on Pattern Analysis and Machine Intelligence.

[26]  Yoshua Bengio,et al.  Why Does Unsupervised Pre-training Help Deep Learning? , 2010, AISTATS.

[27]  Saltelli Andrea,et al.  Global Sensitivity Analysis: The Primer , 2008 .

[28]  W. Thies,et al.  2008 Alzheimer’s disease facts and figures , 2008, Alzheimer's & Dementia.

[29]  Alan C. Evans,et al.  3D Anatomical Atlas of the Human Brain , 1998, NeuroImage.

[30]  J. Weuve,et al.  2016 Alzheimer's disease facts and figures , 2016 .

[31]  Lawrence Carin,et al.  Multi-Task Learning for Classification with Dirichlet Process Priors , 2007, J. Mach. Learn. Res..

[32]  Tapani Raiko,et al.  Improved Learning of Gaussian-Bernoulli Restricted Boltzmann Machines , 2011, ICANN.