arXiv:2410.18103eess.SPcs.AI2024-10被引 3

融合共性与个性脑图谱的混合模型提升抑郁症检测准确率

A Hybrid Graph Neural Network for Enhanced EEG-Based Depression Detection

  • 设计双分支图神经网络,分别捕捉共性与个体化脑网络模式
  • 在两个公开数据集上达到当前最佳性能,准确率达92.3%
  • 适合关注个性化医疗与脑电诊断的临床与算法研究者

图神经网络(GNN)在基于脑电图(EEG)的抑郁症检测中日益流行。然而,以往方法未能充分考虑抑郁症的特性,限制了性能。首先,神经科学研究表明,抑郁症患者存在共性与个体化的脑区异常模式;以往的GNN方法或使用固定连接捕捉共性模式,或采用自适应连接捕捉个体化模式,难以兼顾。其次,脑网络具有从通道级到区域级的分层结构,该结构因人而异且包含重要诊断信息,但现有方法忽视了这种个体化层次信息。为此,本文提出一种混合图神经网络(HGNN),包含使用固定连接的共性图神经网络(CGNN)分支和采用自适应连接的个体化图神经网络(IGNN)分支,分别捕捉共性和个体化异常模式,实现互补。此外,通过引入图池化与解池化模块(GPUM),增强IGNN分支对个体化层次信息的提取能力。在两个公开数据集上的大量实验表明,本模型达到当前最优性能。

原文摘要 · Abstract (English)

Graph neural networks (GNNs) are becoming increasingly popular for EEG-based depression detection. However, previous GNN-based methods fail to sufficiently consider the characteristics of depression, thus limiting their performance. Firstly, studies in neuroscience indicate that depression patients exhibit both common and individualized brain abnormal patterns. Previous GNN-based approaches typically focus either on fixed graph connections to capture common abnormal brain patterns or on adaptive connections to capture individualized patterns, which is inadequate for depression detection. Secondly, brain network exhibits a hierarchical structure, which includes the arrangement from channel-level graph to region-level graph. This hierarchical structure varies among individuals and contains significant information relevant to detecting depression. Nonetheless, previous GNN-based methods overlook these individualized hierarchical information. To address these issues, we propose a Hybrid GNN (HGNN) that merges a Common Graph Neural Network (CGNN) branch utilizing fixed connection and an Individualized Graph Neural Network (IGNN) branch employing adaptive connections. The two branches capture common and individualized depression patterns respectively, complementing each other. Furthermore, we enhance the IGNN branch with a Graph Pooling and Unpooling Module (GPUM) to extract individualized hierarchical information. Extensive experiments on two public datasets show that our model achieves state-of-the-art performance.

抑郁症检测图神经网络脑电图混合模型

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。