Back to Search Start Over

Graph convolution network with similarity awareness and adaptive calibration for disease-induced deterioration prediction.

Authors :
Song, Xuegang
Zhou, Feng
Frangi, Alejandro F
Cao, Jiuwen
Xiao, Xiaohua
Lei, Yi
Wang, Tianfu
Lei, Baiying
Source :
Medical Image Analysis. Apr2021, Vol. 69, pN.PAG-N.PAG. 1p.
Publication Year :
2021

Abstract

• We design a similarity-aware graph in GCN to consider disease statuses. • We propose an adaptive mechanism to estimate similarities. • We propose a calibration mechanism to fuse functional and structural information. • We obtain promising performance for disease deterioration prediction. Image, graphical abstract Graph convolution networks (GCN) have been successfully applied in disease prediction tasks as they capture interactions (i.e., edges and edge weights on the graph) between individual elements. The interactions in existing works are constructed by fusing similarity between imaging information and distance between non-imaging information, whereas disregarding the disease status of those individuals in the training set. Besides, the similarity is being evaluated by computing the correlation distance between feature vectors, which limits prediction performance, especially for predicting significant memory concern (SMC) and mild cognitive impairment (MCI). In this paper, we propose three mechanisms to improve GCN, namely similarity-aware adaptive calibrated GCN (SAC-GCN), for predicting SMC and MCI. First, we design a similarity-aware graph using different receptive fields to consider disease status. The labelled subjects on the graph are only connected with those labelled subjects with the same status. Second, we propose an adaptive mechanism to evaluate similarity. Specifically, we construct initial GCN with evaluating similarity by using traditional correlation distance, then pre-train the initial GCN by using training samples and use it to score all subjects. Then, the difference between these scores replaces correlation distance to update similarity. Last, we devise a calibration mechanism to fuse functional magnetic resonance imaging (fMRI) and diffusion tensor imaging (DTI) information into edges. The proposed method is tested on the Alzheimer's Disease Neuroimaging Initiative (ADNI) dataset. Experimental results demonstrate that our proposed method is useful to predict disease-induced deterioration and superior to other related algorithms, with a mean classification accuracy of 86.83% in our prediction tasks. [ABSTRACT FROM AUTHOR]

Details

Language :
English
ISSN :
13618415
Volume :
69
Database :
Academic Search Index
Journal :
Medical Image Analysis
Publication Type :
Academic Journal
Accession number :
148987667
Full Text :
https://doi.org/10.1016/j.media.2020.101947