Towards Disease-Aware Self-Supervised Dynamic Brain Network Learning For Mental Diagnosis

Zhiyong Jin, Guangqi Wen, Peng Cao, Lingwen Liu, Jinzhu Yang, Xinrong Zhu, Osmar R. Zaïane, Fei Wang

Abstract: The dynamic brain network learning methods ignored the separation of redundant disease-irrelevant information, resulting in the model only achieving suboptimal diagnosis results. Meanwhile, the supervised learning scheme inevitably suffers from poor generalization due to the limited data. To address these problems, we propose a Self-supervised Dynamic Brain network Disentangled representation learning framework named SDBD, which incorporates 1) a dynamic topology-aware encoder for capturing diverse topological information, 2) a cross decoder for reconstructing the graph structure and 3) a spatio-temporal learning model based on the multi-head self-attention mechanism for classification. To disentangle the disease-related information from the dynamic brain networks, we design a temporal contrastive loss and a structure reconstruction loss. We evaluate our model on three real-world mental diseases: Autism Spectrum Disorder (ASD), Major Depressive Disorder (MDD), and Bipolar Disorder (BD). The results indicate significant improvements in our SDBD over the state-of-the-art methods owing to the disentangled disease-related information. Moreover, our method can identify the biomarkers associated with the diseases, which is consistent with the previous studies. To the best of our knowledge, our work is the first attempt to disentangle the disease-related information for the dynamic brain network analysis. The code is available at https://github.com/IntelliDAL/Graph/tree/main/SDBD.