A Hierarchical Graph Convolutional Network With Infomax-Guided Graph Embedding for Population-Based ASD Detection.

Journal: IEEE journal of biomedical and health informatics
Published Date:

Abstract

Recently, functional magnetic resonance imaging (fMRI)-based brain networks have been shown to be an effective diagnostic tool with great potential for accurately detecting autism spectrum disorders (ASD). Meanwhile, the successful use of graph convolution networks (GCNs) methods based on fMRI information has improved the classification accuracy of ASD. However, many graph convolution-based methods do not fully utilize the topological information of the brain functional connectivity network (BFCN) or ignore the effect of non-imaging information. Therefore, we propose a hierarchical graph embedding model that leverage both the topological information of the BFCN and the non-imaging information of the subjects to improve the classification accuracy. Specifically, our model first use the Infomax Module to automatically identify embedded features in regions of interests (ROIs) in the brain. Then, these features, along with non-imaging information, is used to construct a population graph model. Finally, we design a graph convolution framework to propagate and aggregate the node features and obtain the results for ASD detection. Our model takes into account both the significance of the BFCN to individual subjects and relationships between subjects in the population graph. The model performed autism detection using the Autism Brain Imaging Data Exchange (ABIDE) dataset and obtained an average accuracy of 77.2% and an AUC of 87.2%. These results exceed those of the baseline approach. Through extensive experiments, we demonstrate the competitiveness, robustness and effectiveness of our model in aiding ASD diagnosis.

Authors

  • Xiaoke Hao
    School of Artificial Intelligence, Hebei University of Technology, Tianjin 300401, China.
  • Mingming Ma
    Department of Diagnostic Ultrasound and Echocardiography, Sir Run Run Shaw Hospital, Zhejiang University College of Medicine, Hangzhou, People's Republic of China.
  • Jiaqing Tao
  • Jiahui Cao
  • Jing Qin
    School of Nursing, The Hong Kong Polytechnic University, Hong Kong, China.
  • Feng Liu
    Department of Vascular and Endovascular Surgery, The First Medical Center of Chinese PLA General Hospital, 100853 Beijing, China.
  • Daoqiang Zhang
    College of Computer Science and Technology, Nanjing University of Aeronautics and Astronautics, Nanjing, China.
  • Dong Ming
    Institute of Medical Engineering and Translational Medicine, Tianjin University, Tianjin, China.