Integrating Similarity Awareness and Adaptive Calibration in Graph Convolution Network to Predict Disease

Significant memory concern (SMC) is the earlier stage of mild cognitive impairment (MCI), and its early treatment is quite vital to delay further disease-induced deterioration. To predict the deterioration, graph convolution network (GCN) with current adjacency matrix still suffers from limited prediction performance due to their subtle difference and obscure features. For this reason, we propose a similarity-aware adaptive calibrated GCN (SAC-GCN), which can combine functional and structural information to predict SMC and MCI. We utilize an adaptive calibration mechanism to construct a data-driven adjacency matrix. Specifically, we first design a similarity-aware graph using different receptive fields to consider the disease statuses. Namely, the labeled subjects are only connected with those subjects who have the same status in the convolution operation. Then we compute more accurate weights in graph edges from functional and structural scores. Current edge weights are used to construct an initial graph and pre-train the GCN. Based on the pre-trained GCN, the differences between scores replace the traditional correlation distances to evaluate edge weights. Lastly, we devise a calibration technique to fuse functional and structural information for edge weighting. The proposed method is tested on the Alzheimer’s Disease Neuroimaging Initiative (ADNI) dataset. The experimental results demonstrate that our proposed method is effective to predict disease-induced deterioration and superior over other related algorithm.

[1]  Daniel Rueckert,et al.  Random forest-based similarity measures for multi-modal classification of Alzheimer's disease , 2013, NeuroImage.

[2]  Ee-Leng Tan,et al.  Self-calibrated brain network estimation and joint non-convex multi-task learning for identification of early Alzheimer's disease , 2020, Medical Image Anal..

[3]  Daoqiang Zhang,et al.  Multi-modal AD classification via self-paced latent correlation analysis , 2019, Neurocomputing.

[4]  Daoqiang Zhang,et al.  Group-constrained sparse fMRI connectivity modeling for mild cognitive impairment identification , 2013, Brain Structure and Function.

[5]  K. Lovblad,et al.  Individual prediction of cognitive decline in mild cognitive impairment using support vector machine-based analysis of diffusion tensor imaging data. , 2010, Journal of Alzheimer's disease : JAD.

[6]  Hao Yang,et al.  Novel Effective Connectivity Inference Using Ultra-Group Constrained Orthogonal Forward Regression and Elastic Multilayer Perceptron Classifier for MCI Identification , 2019, IEEE Transactions on Medical Imaging.

[7]  Ben Glocker,et al.  Metric learning with spectral graph convolutions on brain connectivity networks , 2018, NeuroImage.

[8]  Dinggang Shen,et al.  Multimodal hyper‐connectivity of functional networks using functionally‐weighted LASSO for MCI classification , 2019, Medical Image Anal..

[9]  Yang Li,et al.  Deep Spatial-Temporal Feature Fusion From Adaptive Dynamic Functional Connectivity for MCI Identification , 2020, IEEE Transactions on Medical Imaging.

[10]  O. Abe,et al.  Diffeomorphic Anatomical Registration Through Exponentiated Lie Algebra provides reduced effect of scanner for cortex volumetry with atlas-based method in healthy subjects , 2013, Neuroradiology.

[11]  Ben Glocker,et al.  Disease prediction using graph convolutional networks: Application to Autism Spectrum Disorder and Alzheimer's disease , 2018, Medical Image Anal..

[12]  Sundaram Suresh,et al.  Meta-cognitive q-Gaussian RBF network for binary classification: Application to mild cognitive impairment (MCI) , 2013, The 2013 International Joint Conference on Neural Networks (IJCNN).

[13]  N. Tzourio-Mazoyer,et al.  Automated Anatomical Labeling of Activations in SPM Using a Macroscopic Anatomical Parcellation of the MNI MRI Single-Subject Brain , 2002, NeuroImage.

[14]  Dinggang Shen,et al.  Connectivity strength‐weighted sparse group representation‐based brain network construction for MCI classification , 2017, Human brain mapping.

[15]  Paul Thompson,et al.  Integrating Heterogeneous Brain Networks for Predicting Brain Disease Conditions , 2019, MICCAI.

[16]  Fan Zhang,et al.  Machine Learning Classification Combining Multiple Features of A Hyper-Network of fMRI Data in Alzheimer's Disease , 2017, Front. Neurosci..

[17]  Shadi Albarqouni,et al.  InceptionGCN: Receptive Field Aware Graph Convolutional Network for Disease Prediction , 2019, IPMI.

[18]  Dinggang Shen,et al.  Strength and similarity guided group-level brain functional network construction for MCI diagnosis , 2019, Pattern Recognit..

[19]  Dinggang Shen,et al.  Sparse temporally dynamic resting-state functional connectivity networks for early MCI identification , 2016, Brain Imaging and Behavior.