FC-HGNN: A heterogeneous graph neural network based on brain functional connectivity for mental disorder identification

Abstract

Rapid and accurate diagnosis of mental disorders has long been an essential challenge in clinical medicine. Due to the advantage in addressing non-Euclidean structures, graph neural networks have been increasingly used to study brain networks. Among the existing methods, the population graph models have achieved high predictive accuracy by considering intersubject relationships but weak interpretability limits its clinical applicability. The individual graph approach models functional brain networks and identifies abnormal brain regions that cause diseases but has poor accuracy. To address these issues, we propose a heterogeneous graph neural network based on brain functional connectivity (FC-HGNN), which is an end-to-end model with a two-stage process. In the first phase, the brain connectomic graph is used to extract individual brain features. An integrated intrahemispheric and interhemispheric convolutional graph layer is used to learn brain region features, and a local–global dual-channel pooling layer is used to identify biomarkers. In the second stage, a heterogeneous population graph is constructed based on sex and the fusion of imaging and non-imaging data from subjects. The feature embeddings of same-sex and opposite-sex neighbours are learned separately according to a hierarchical feature aggregation approach. Subsequently, they are adaptively fused to generate the final node embedding, which is then utilized for obtaining classification predictions. The cross-validation and transduction learning results show that FC-HGNN achieves state-of-the-art performance in classification prediction experiments using two public datasets. Moreover, FC-HGNN identifies crucial biomarker regions relevant for disease classification, aligning with existing studies and exhibiting outstanding predictive performance on actual clinical data. The code is available at https://github.com/Guuuu11/FC-HGNN_Pytorch.