Convolutional neural networks for Alzheimer's disease detection on MRI images

Abstract

Purpose: Detection of Alzheimer's disease (AD) on magnetic resonance imaging (MRI) using convolutional neural networks (CNNs), which is useful for detecting AD in its preliminary states. Approach: Our study implements and compares several deep models and configurations, including two-dimensional (2D) and three-dimensional (3D) CNNs and recurrent neural networks (RNNs). To use a 2D CNN on 3D MRI volumes, each MRI scan is split into 2D slices, neglecting the connection among 2D image slices in an MRI volume. Instead, a CNN model could be followed by an RNN in a way that the model of 2D CNN + RNN can understand the connection among sequences of 2D image slices for an MRI. The issue is that the feature extraction step in the 2D CNN is independent of classification in the RNN. To tackle this, 3D CNNs can be employed instead of 2D CNNs to make voxel-based decisions. Our study's main contribution is to introduce transfer learning from a dataset of 2D images to 3D CNNs. Results: The results on our MRI dataset indicate that sequence-based decisions improve the accuracy of slice-based decisions by 2% in classifying AD patients from healthy subjects. Also the 3D voxel-based method with transfer learning outperforms the other methods with 96.88% accuracy, 100% sensitivity, and 94.12% specificity. Conclusions: Several implementations and experiments using CNNs on MRI scans for AD detection demonstrated that the voxel-based method with transfer learning from ImageNet to MRI datasets using 3D CNNs considerably improved the results compared with the others.

Keywords: Alzheimer’s disease; convolutional neural network; deep learning; magnetic resonance imaging; transfer learning.