HAGCN: A hybrid-order brain network-based graph convolution learning framework with multi-head attention for brain disorder classification

Xubin Wu,Xin Wen,Jie Sun,Yanqing Dong,Chenyu Wu,Bin Wang,Jie Xiang
DOI: https://doi.org/10.1016/j.bspc.2024.106944
IF: 5.1
2024-10-11
Biomedical Signal Processing and Control
Abstract:Brain networks provide essential insights in diagnosing many brain disorders, such as autism spectrum disorder (ASD) and attention-deficit/hyperactivity disorder (ADHD). However, current works often fail to fully capture the potential interaction characteristics of inter-regional functional connectivity within the brain network. To address this limitation, we propose a Hybrid-Attention Graph Convolution Network (HAGCN) for brain disorder classification. The learning framework designs a hybrid-order effective network (graph) for representing the lower-order and higher-order spatial interaction relations between brain regions, proposes a prior graph based on statistical tests to eliminate the noisy edges effectively, presents a dual-channel graph convolution module for simultaneously capturing information flow features of each brain region in out-degree and in-degree directions, and implement a multi-head attention module to focus and discern the significant features from the multidimensional interaction relations for improving the expressiveness of the model. Experiments on ADHD-200 and ABIDE-I datasets show the effectiveness and advantages of our proposed method, achieving the highest average accuracy values of 77.95 % and 78.17 %, respectively. The results demonstrate that integrating multidimensional and directional interaction characteristics can enhance the network's capacity and improve learning efficiency. Additionally, our approach identifies and analyzes several key brain regions linked to brain disorders.
engineering, biomedical
What problem does this paper attempt to address?