CBIR system using Capsule Networks and 3D CNN for Alzheimer's disease diagnosis

摘要

Alzheimer's disease (AD) is an irreversible disorder of the brain related to loss of memory, commonly seen in the elderly and aging population. Implementation of revolutionary computer aided diagnosis techniques with Content Based Image Retrieval (CBIR) has created new potentials in Magnetic resonance imaging (MRI) in relevant image retrieval and training for detection of progression of AD in early stages. This paper proposed a CBIR system using 3D Capsule Network, 3D-Convolutional Neural Network and pre-trained 3D-autoencoder technology for early detection of Alzheimer's. A 3D-Capsule Networks (CapsNets) is capable of fast learning, even for small datasets and can effectively handle robust image rotations and transitions. It was observed that, an ensemble method using 3D-CapsNets and convolution neural network (CNN) with 3D-autoencoder, increased the detection performances comparing to Deep-CNN alone. CBIR using the proposed model was found to be up to 98.42% accurate in AD classification. Moreover, we argue that CapsNet seems to be a promising new technique for image classification, and further experiments using more robust computation resources and refined CapsNet architectures may produce better outcomes.