Graph Reasoning Module for Alzheimer’s Disease Diagnosis: A Plug-and-Play Method

Abstract

Recent advances in deep learning have led to increased adoption of convolutional neural networks (CNN) for structural magnetic resonance imaging (sMRI)-based Alzheimer’s disease (AD) detection. AD results in widespread damage to neurons in different brain regions and destroys their connections. However, current CNN-based methods struggle to relate spatially distant information effectively. To solve this problem, we propose a graph reasoning module (GRM), which can be directly incorporated into CNN-based AD detection models to simulate the underlying relationship between different brain regions and boost AD diagnosis performance. Specifically, in GRM, an adaptive graph Transformer (AGT) block is designed to adaptively construct a graph representation based on the feature map given by CNN, a graph convolutional network (GCN) block is adopted to update the graph representation, and a feature map reconstruction (FMR) block is built to convert the learned graph representation to a feature map. Experimental results demonstrate that the insertion of the GRM in the existing AD classification model can increase its balanced accuracy by more than 4.3%. The GRM-embedded model achieves state-of-the-art performance compared with current deep learning-based AD diagnosis methods, with a balanced accuracy of 86.2%.

Keywords

• Alzheimer’s disease diagnosis
• graph convolution network
• plug-and-play
• structural magnetic resonance imaging