Exploring interpretable graph convolutional networks for autism spectrum disorder diagnosis.

Abstract

Finding the biomarkers associated with autism spectrum disorder (ASD) is helpful for understanding the underlying roots of the disorder and can lead to earlier diagnosis and more targeted treatments. In essence, we are faced with two challenges (i) how to learn a node representation and a clean graph structure from original graph data with high dimensionality and (ii) how to jointly model the procedure of node representation learning, structure learning and graph classification. We propose FSL-BrainNet, an interpretable graph convolution network (GCN) model for jointly Learning of node Features and clean Structures in brain networks for automatic brain network classification and interpretation. We formulate an end-to-end trainable and interpretable framework for graph classification and biomarkers (salient brain regions and potential subnetworks) identification. The experimental results on the ABIDE dataset show that our proposed methods not only achieve improved prediction performance compared with the state-of-the-art methods, but also find a compact set of highly suggestive biomarkers including relevant brain regions and subnetworks to ASD. Through node feature learning and structure learning, our model can simultaneously select important brain regions and identify subnetworks.