用图神经网络增强脑电模型,提升疾病诊断能力
GEFM: Graph-Enhanced EEG Foundation Model
- 融合图神经网络捕捉脑电通道间关系
- 在3个下游任务中均超越基线模型
- 适合脑电分析与医疗诊断研究者
脑电图(EEG)信号在疾病诊断和医疗健康领域具有重要意义,但标注数据稀缺。基础模型通过大规模无标签数据预训练,为解决此问题提供了前景。尽管时间动态和通道间关系对理解EEG都至关重要,现有模型主要关注前者而忽视后者。为此,我们提出图增强型脑电基础模型(GEFM),整合时序与通道间信息。该模型结合图神经网络(GNNs)与掩码自编码器,实现高效预训练。我们在三个下游任务上评估了方法,并测试了多种GNN架构。结果表明,采用优化配置的GCN架构时,本模型在所有任务中均持续优于基线方法。这些发现表明,该模型可作为脑电分析的强健基础模型。
原文摘要 · Abstract (English)
Electroencephalography (EEG) signals provide critical insights for applications in disease diagnosis and healthcare. However, the scarcity of labeled EEG data poses a significant challenge. Foundation models offer a promising solution by leveraging large-scale unlabeled data through pre-training, enabling strong performance across diverse tasks. While both temporal dynamics and inter-channel relationships are vital for understanding EEG signals, existing EEG foundation models primarily focus on the former, overlooking the latter. To address this limitation, we propose Graph-Enhanced EEG Foundation Model (GEFM), a novel foundation model for EEG that integrates both temporal and inter-channel information. Our architecture combines Graph Neural Networks (GNNs), which effectively capture relational structures, with a masked autoencoder to enable efficient pre-training. We evaluated our approach using three downstream tasks and experimented with various GNN architectures. The results demonstrate that our proposed model, particularly when employing the GCN architecture with optimized configurations, consistently outperformed baseline methods across all tasks. These findings suggest that our model serves as a robust foundation model for EEG analysis.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。