arXiv:2411.08521cs.LGcs.AI2024-11被引 1

融合电极空间结构与时间窗口连续性的脑电分析模型,提升抑郁症检测准确率

A spatiotemporal fused network considering electrode spatial topology and time-window transition for MDD detection

  • 结合电极空间拓扑与相邻时间窗关联,挖掘脑电信号深层时空特征
  • 在两个公开数据集上分别达到92.00%和94.00%的检测准确率
  • 适合关注脑电诊断与跨被试泛化能力的研究者

近年来,研究者开始尝试使用深度学习方法基于脑电图(EEG)信号检测重度抑郁症(MDD),以寻求更客观的诊断手段。然而,现有时空特征提取方法仅考虑多电极间功能相关性和脑电信号的时间相关性,忽略了电极间的空间位置连接信息以及时间窗之间的连续性,降低了模型的特征提取能力。为此,本文提出一种结合电极空间拓扑与邻近时间窗转换信息的时空融合网络(SET-TIME)。SET-TIME由通用特征提取器、二级时间相关性特征提取器和领域自适应(DA)模块组成:前者获取时空特征,后者挖掘多时间窗间的关联,而DA模块增强跨被试检测能力。10折交叉验证结果显示,该方法在公开数据集PRED+CT和MODMA上分别取得92.00%和94.00%的检测准确率,优于当前最先进方法。消融实验证明各模块有效性,通过探索脑电信号内在时空特性助力MDD检测。

原文摘要 · Abstract (English)

Recently, researchers have begun to experiment with deep learning-based methods for detecting major depressive disor-der (MDD) using electroencephalogram (EEG) signals in search of a more objective means of diagnosis. However, exist-ing spatiotemporal feature extraction methods only consider the functional correlation between multiple electrodes and temporal correlation of EEG signals, ignoring the spatial posi-tion connection information between electrodes and the conti-nuity between time windows, which reduces the model's fea-ture extraction capabilities. To address this issue, a Spatio-temporal fused network for MDD detection with Electrode spatial Topology and adjacent TIME-window transition in-formation (SET-TIME) is proposed in this study. SET-TIME is composed by a common feature extractor, a secondary time-correlation feature extractor, and a domain adaptation (DA) module, in which the former extractor is used to obtain the temporal and spatial features, and the latter extractor can mine the correlation between multiple time windows, and the DA module is adopted to enhance cross-subject detection ca-pability. The experimental results of 10-fold cross-validation show that the proposed SET-TIME method outperforms the state-of-the-art (SOTA) method by achieving MDD detection accuracies of 92.00% and 94.00% on the public datasets PRED+CT and MODMA, respectively. Ablation experiments demonstrate the effectiveness of the multiple modules in SET-TIME, which assist in MDD detection by exploring the intrin-sic spatiotemporal information of EEG signals.

抑郁症检测脑电分析时空建模

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