Temporal Graph Representation Learning for Autism spectrum disorder Brain Networks

Modeling spatio-temporal dynamics in functional brain networks is critical for underlying the functional mechanism of autism spectrum disorder (ASD). In our study, we propose an end-to-end framework called temporal graph representation learning for brain networks, which thoroughly captures spatio-temporal features in resting-state functional magnetic resonance imaging (rs-fMRI) data. Specifically, we first transform rs-fMRI time-series into temporal multi-graph using a sliding window technique. A temporal multi-graph clustering is then designed to eliminate the inconsistency of the temporal multi-graph series. Then, a graph structure aware LSTM (GSA-LSTM) is proposed to capture the spatio-temporal embedding for temporal graphs. The proposed GSA-LSTM can not only capture discriminative features for prediction but also impute the incomplete graphs for the temporal multi-graph series. Extensive experiments on autism brain imaging data exchange (ABIDE) dataset shows the effectiveness of our proposed framework. The results demonstrate that the proposed dynamic brain network embedding learning outperforms the state of-the-art brain network classification models. Furthermore, the obtained clustering results are consistent with the previous neuroimaging-derived evidence of biomarkers for autism spectrum disorder (ASD).