An efficient 3D deep convolutional network for Alzheimer's disease diagnosis using MR images

Automatic extraction of features from MRI brain scans and diagnosis of Alzheimer's Disease (AD) remain a challenging task. In this paper, we propose an efficient and simple three-dimensional convolutional network (3D ConvNet) architecture that is able to achieve high performance for detection of AD on a relatively large dataset. The proposed 3D ConvNet consists of five convolutional layers for feature extraction, followed by three fully-connected layers for AD/NC classification. The main contributions of the paper include: (a) propose a novel and effective 3D ConvNet architecture; (b) study the impact of hyper-parameter selection on the performance of AD classification; (c) study the impact of pre-processing; (d) study the impact of data partitioning; (e) study the impact of dataset size. Experiments conducted on an ADNI dataset containing 340 subjects and 1198 MRI brain scans have resulted good performance (with the test accuracy of 98.74%, 100% AD detection rate and 2,4% false alarm). Comparisons with 7 existing state-of-the-art methods have provided strong support to the robustness of the proposed method.

[1]  Daniel Rueckert,et al.  Multiple instance learning for classification of dementia in brain MRI , 2013, Medical Image Anal..

[2]  Giovanni Montana,et al.  Predicting Alzheimer's disease: a neuroimaging study with 3D convolutional neural networks , 2015, ICPRAM 2015.

[3]  Y. Yao,et al.  On Early Stopping in Gradient Descent Learning , 2007 .

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

[5]  T. Chan,et al.  Independent component analysis-based classification of Alzheimer's disease MRI data. , 2011, Journal of Alzheimer's disease : JAD.

[6]  Sidong Liu,et al.  Early diagnosis of Alzheimer's disease with deep learning , 2014, 2014 IEEE 11th International Symposium on Biomedical Imaging (ISBI).

[7]  M. Chong,et al.  Preclinical Alzheimer's disease: diagnosis and prediction of progression , 2005, The Lancet Neurology.

[8]  Yu Liu,et al.  3D object understanding with 3D Convolutional Neural Networks , 2016, Inf. Sci..

[9]  Yoshua Bengio,et al.  Deep Sparse Rectifier Neural Networks , 2011, AISTATS.

[10]  Anthony Maida,et al.  Natural Image Bases to Represent Neuroimaging Data , 2013, ICML.

[11]  Yoram Singer,et al.  Adaptive Subgradient Methods for Online Learning and Stochastic Optimization , 2011, J. Mach. Learn. Res..

[12]  Ghassem Tofighi,et al.  DeepAD: Alzheimer’s Disease Classification via Deep Convolutional Neural Networks using MRI and fMRI , 2016, bioRxiv.

[13]  Ayman El-Baz,et al.  Alzheimer's disease diagnostics by adaptation of 3D convolutional network , 2016, 2016 IEEE International Conference on Image Processing (ICIP).