Deep sequence modelling for Alzheimer's disease detection using MRI

Abstract

Background: Alzheimer's disease (AD) is one of the deadliest diseases in developed countries. Treatments following early AD detection can significantly delay institutionalisation and extend patients' independence. There has been a growing focus on early AD detection using artificial intelligence. Convolutional neural networks (CNNs) have proven revolutionary for image-based applications and have been applied to brain scans. In recent years, studies have utilised two-dimensional (2D) CNNs on magnetic resonance imaging (MRI) scans for AD detection. To apply a 2D CNN on three-dimensional (3D) MRI volumes, each MRI scan is split into 2D image slices. A CNN is trained over the image slices by calculating a loss function between each subject's label and each image slice's predicted output. Although 2D CNNs can discover spatial dependencies in an image slice, they cannot understand the temporal dependencies among 2D image slices in a 3D MRI volume. This study aims to resolve this issue by modelling the sequence of MRI features produced by a CNN with deep sequence-based networks for AD detection.

Method: The CNN utilised in this paper was ResNet-18 pre-trained on an ImageNet dataset. The employed sequence-based models were the temporal convolutional network (TCN) and different types of recurrent neural networks. Several deep sequence-based models and configurations were implemented and compared for AD detection.

Results: Our proposed TCN model achieved the best classification performance with 91.78% accuracy, 91.56% sensitivity and 92% specificity.

Conclusion: Our results show that applying sequence-based models can improve the classification accuracy of 2D and 3D CNNs for AD detection by up to 10%.

Keywords: Alzheimer's disease; Convolutional neural network; MRI; Recurrent neural network; Temporal convolutional network.