Hierarchical Encoding and Fusion of Brain Functions for Depression Subtype Classification

Mengjun Liu,Huifeng Zhang,Mianxin Liu,Dongdong Chen,Rubai Zhou,Wenxian Lu,Lichi Zhang,Dinggang Shen,Qian Wang,Daihui Peng
DOI: https://doi.org/10.1109/taffc.2024.3401251
IF: 13.99
2024-01-01
IEEE Transactions on Affective Computing
Abstract:Depression is a serious mental disorder with complex etiology, exhibiting strong heterogeneity in clinical manifestations such as various subtypes. Research on depression subtypes may deepen the understanding of the disease, contributing to the diagnosis and prognosis. While brain functional network and graph neural networks (GNNs) provide such a means, the task is still challenged by limited feature encoding from the informative fMRI data, ineffective information fusion of brain functional network, and small size of the recruited subjects. Therefore, we propose a hierarchical encoding and fusion framework of brain functions. First, we pre-train a model to extract the features from individual brain regions, which signify nodes in the brain functional network. Then, distinct graphs are constructed to link the nodes within each subject, resulting in multi-view graphs of the brain functional network. We further develop a graph fusion strategy to integrate the multi-view information, by referring to the local encoding of the nodes and their interactions across multiple graph instances. Finally, we attain the classification of depression subtypes based on the fused graph representation. The experimental results demonstrate that our method can superiorly distinguish major depression subtypes and outperform the state-of-the-art methods.
What problem does this paper attempt to address?