SLG-NET: Subgraph Neural Network with Local-Global Braingraph Feature Extraction Modules and a Novel Subgraph Generation Algorithm for Automated Identification of Major Depressive Disorder

Yan Zhang,Xin Liu,Panrui Tang,Zuping Zhang
DOI: https://doi.org/10.1007/978-981-99-8070-3_3
2024-01-01
Abstract:Major depressive disorder (MDD) is a severe mental illness that poses significant challenges to both society and families. Recently, several graph neural network (GNN)-based methods have been proposed for MDD diagnosis and achieved promising results. However, these methods encode entire braingraph directly, have overlooked the subgraph structure of braingraph, which leads to poor specificity to braingraphs. Additionally, the GNN framework they used is rudimentary, resulting in insufficient feature extraction capabilities. In light of the two shortcomings mentioned above, this paper designed a novel depression diagnosis framework named SLG-NET based on subgraph neural network. To the best of our knowledge, this study is the first attempt to apply subgraph neural network to the field of depression diagnosis. In order to enhance the specificity of our model to braingraphs, we propose a novel subgraph generation algorithm based on sub-structure information of brain. To improve feature extraction capabilities, a local and global braingraph feature extraction modules are proposed to extract braingraph properties at both local and global levels. Comprehensive experiments performed on rest-metamdd dataset show that the performance of proposed SLG-NET significantly surpasses many state-of-the-art methods, which show that the SLG-NET has the potential for auxiliary diagnosis of depression in clinical scenarios.
What problem does this paper attempt to address?