Classification of Autism Spectrum Disorder Using rs-fMRI data and Graph Convolutional Networks.

Abstract

Autism spectrum disorder (ASD) affects large number of children and adults in the US, and worldwide. Early and quick diagnosis of ASD can improve the quality of life significantly both for patients and their families. Prior research provides strong evidence that structural and functional magnetic resonance imaging (MRI) data collected from individuals with ASD exhibit distinguishing characteristics that differ in local and global, spatial and temporal neural patterns of the brain - and therefore can be used for diagnostic purposes for various mental disorders. However, the data from MRI are high-dimensional and advanced methods are needed to make sense out of these datasets. In this paper, we present a novel model based on graph convolutional network (GCN) that can utilize resting state fMRI (rs-fMRI) data to classify ASD subjects from health controls (HC). In addition to using the graph from traditional correlation matrices, our proposed GCN model incorporates graphlet topological counting as one of the training features. Our results show that graphlets can preserve the topological information of the graphs obtained from fMRI data. Combined with our GCN, the graphlets retain enough topological information to differentiate between the ASD and HC. Our proposed model gives an average accuracy of 64.27% on the whole ABIDE-I data sets (1035 subjects) and highest site-specific accuracy of 75.9%, which is comparable to other state-of-the-art methods - while potentially open to being more interpretable.