An Invertible Dynamic Graph Convolutional Network for Multi-Center ASD Classification

Frontiers in Neuroscience

Jie Wen¹  Yueying Chen³  Xueyang Fu³  Xun Chen³  Aiping Liu³ 

[1] Division of Life Sciences and Medicine, Department of Radiology, The First Affiliated Hospital of USTC (Anhui Provincial Hospital), University of Science and Technology of China, Hefei, China;School of Information Science and Technology, University of Science and Technology of China, Hefei, China;USTC IAT-Huami Joint Laboratory for Brain-Machine Intelligence, Institute of Advanced Technology, University of Science and Technology of China, Hefei, China;

关键词: fMRI;    graph convolutional networks;    invertible networks;    brain connectivity networks;    autism spectrum disorder;    disease classification;   

DOI  :  10.3389/fnins.2021.828512

来源: DOAJ

【 摘 要 】

Autism Spectrum Disorder (ASD) is one common developmental disorder with great variations in symptoms and severity, making the diagnosis of ASD a challenging task. Existing deep learning models using brain connectivity features to classify ASD still suffer from degraded performance for multi-center data due to limited feature representation ability and insufficient interpretability. Given that Graph Convolutional Network (GCN) has demonstrated superiority in learning discriminative representations of brain connectivity networks, in this paper, we propose an invertible dynamic GCN model to identify ASD and investigate the alterations of connectivity patterns associated with the disease. In order to select explainable features from the model, invertible blocks are introduced in the whole network, and we are able to reconstruct the input dynamic features from the network's output. A pre-screening of connectivity features is adopted to reduce the redundancy of the input information, and a fully-connected layer is added to perform classification. The experimental results on 867 subjects show that our proposed method achieves superior disease classification performance. It provides an interpretable deep learning model for brain connectivity analysis and is of great potential in studying brain-related disorders.

【 授权许可】

Unknown