BrainTGL: A dynamic graph representation learning model for brain network analysis

Abstract

Modeling the dynamics characteristics in functional brain networks (FBNs) is important for understanding the functional mechanism of the human brain. However, the current works do not fully consider the potential complex spatial and temporal correlations in human brain. To solve this problem, we propose a temporal graph representation learning framework for brain networks (BrainTGL). The framework involves a temporal graph pooling for eliminating the noisy edges as well as data inconsistency, and a dual temporal graph learning for capturing the spatio-temporal features of the temporal graphs. The proposed method has been evaluated in both tasks of brain disease (ASD, MDD and BD) diagnosis/gender classification (classification task) and subtype identification (clustering task) on the four datasets: Human Connectome Project (HCP), Autism Brain Imaging Data Exchange (ABIDE), NMU-MDD and NMU-BD. A large improvement is achieved for the ASD diagnosis. Specifically, our model outperforms the GroupINN and ST-GCN by an average increase of 4.2% and 8.6% on accuracy, respectively, demonstrating its advantages in comparison to the state-of-the-art methods based on functional connectivity features or learned spatio-temporal features. The results demonstrate that learning the spatial–temporal brain network representation for modeling dynamics characteristics in FBNs can improve the model’s performance on both disease diagnosis and subtype identification tasks for multiple disorders. Apart from performance, the improvements of computational efficiency and convergence speed reduce training costs.