A diagonal-structured-state-space-sequence-model based deep learning framework for effective diagnosis of mild cognitive impairment

Early diagnosis of mild cognitive impairment (MCI) may effectively prevent its development to Alzheimer’s disease. Function connectivity (FC) of the brain networks is a widely used biomarker for MCI detection. However, FC estimated by pre-defined metrics may unable to fully characterize the brain signals. The present study aims to develop a deep learning framework directly applied to the brain signals for improved MCI diagnosis. A resting-state functional magnetic resonance imaging (rs-fMRI) dataset containing normal controls (NC), early MCI (EMCI), and late MCI (LMCI) was used to develop and evaluate our model. Blood-oxygenation-level-dependent (BOLD) signals were measured by the fMRI. A 1-D pointwise convolution was employed to freely capture the spatial features, and a diagonal structured state space sequence (S4D) model was designed to extract the temporal features, particularly the long-term dependence of the BOLD signals. The proposed model was evaluated on three classification tasks, i.e., NC vs. EMCI, EMCI vs. LMCI, and NC vs. EMCI vs. LMCI, with repeated 10-fold cross validation. Accuracy, sensitivity, specificity, and area under the receiver operating characteristic curve were calculated as performance metrics. For the two binary classification tasks, our model achieves the best performance in all metrics among seven state-of-the-art (SOTA) methods. For the three-category classification, despite slightly lower sensitivity, our model produces an overall superior performance than other methods. Our results indicate that long-term dependence of the BOLD signals may contribute significantly to MCI detection, providing useful information for automated diagnosis of MCI.