用双曲几何捕捉抑郁脑网络的层次结构,提升癫痫识别准确率
SA-HGNN: Sample-Adaptive Hyperbolic Graph Neural Network for EEG-Based Depression Recognition

- 根据样本自适应构建脑网络拓扑,捕捉复杂空间关系
- 采用双曲图卷积,更精准表示脑网络的层级特征
- 注意力池化过滤噪声通道,增强对异常连接的识别能力
图神经网络(GNN)被广泛用于捕捉脑电图(EEG)中空间功能连接模式,以提升抑郁症识别性能。然而,抑郁症患者脑网络的功能连接具有内在层次结构,传统方法难以准确建模。为此,本文提出一种新型模型——样本自适应双曲图神经网络(SA-HGNN),旨在精确提取受抑郁影响的脑网络真实层次结构。该模型包含三个核心模块:首先,样本自适应图构建模块动态生成个性化脑网络拓扑,以捕捉脑网络内部更复杂的空间关系;其次,采用双曲图卷积克服欧氏空间的表示瓶颈,利用双曲几何精确捕捉脑网络中的潜在层级关系;最后,注意力池化模块自适应过滤脑电信号中高度冗余的噪声通道,有效缓解固有噪声对真实层级拓扑的干扰。在公开的EEG数据集上进行的大量实验表明,该方法在静息态和任务态范式下均表现优越,验证了其对噪声的鲁棒性以及捕捉抑郁症患者脑网络异常功能连接模式的有效性。
原文摘要 · Abstract (English)
Graph Neural Networks (GNNs) have been widely used to capture spatial functional connectivity patterns to improve electroencephalography (EEG)-based depression recognition performance. However, the functional connectivity of brain networks in patients with depression exhibits an inherent hierarchical structure, making it difficult to capture accurate connection patterns. To address these issues, this paper proposes a novel model named Sample-Adaptive Hyperbolic Graph Neural Network (SA-HGNN), which aims to accurately extract the authentic hierarchical structure of depression-affected brain networks. Specifically, the proposed model comprises three core modules. First, a Sample-Adaptive Graph Construction module dynamically constructs personalized brain network topologies to capture more complex spatial relationships within the brain network. Second, hyperbolic graph convolution is employed to overcome the representation bottlenecks of Euclidean space, leveraging hyperbolic geometry to precisely capture latent hierarchical relationships within the brain network. Finally, an Attention Pooling module adaptively filters out highly redundant noise channels in EEG signals, effectively mitigating the interference of inherent noise on the authentic hierarchical topology. Extensive experiments on public EEG datasets demonstrate the superior performance of our method across resting-state and task-related paradigms, validating its robustness to noise and efficacy in capturing abnormal functional connectivity patterns in brain networks of patients with depression.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。