arXiv:2508.14086cs.LG2025-08被引 7

用生成式扩散模型学习脑电图表示,更高效准确。

EEGDM: EEG Representation Learning via Generative Diffusion Model

  • 基于扩散模型构建结构化状态空间,捕捉脑电信号动态
  • 在多个数据集上优于现有先进方法,显著提升癫痫检测性能
  • 适合需要低计算成本的脑电分析研究者使用

尽管脑电图(EEG)是监测大脑和诊断神经疾病(如癫痫)的关键工具,但受标注数据有限和信号高度变异影响,从原始脑电信号中学习有意义的表示仍具挑战。近年来,基于Transformer架构与自监督预训练(如掩码预测)的脑电基础模型(EEG FMs)展现出潜力,但在训练和推理中计算开销大,模型规模增大后性能提升有限。本文提出一种基于生成式扩散模型的脑电表示学习框架(EEGDM)。我们设计了用于扩散预训练的结构化状态空间模型(SSMDP),结合去噪扩散概率模型(DDPM)进行训练,以更好捕捉脑电信号的时间动态;随后通过提出的潜在融合变压器(LFT)将获得的潜在表示应用于下游分类任务。在涵盖间期癫痫样放电(TUEV)与发作(CHB-MIT)检测的多事件数据集上评估,结果表明该方法优于当前主流方法。这些发现表明,EEGDM为现有基础模型提供了一种有前景的替代方案。代码与模型检查点已公开:https://github.com/jhpuah/EEGDM。

原文摘要 · Abstract (English)

While electroencephalogram (EEG) has been a crucial tool for monitoring the brain and diagnosing neurological disorders (e.g., epilepsy), learning meaningful representations from raw EEG signals remains challenging due to limited annotations and high signal variability. Recently, EEG foundation models (FMs) have shown promising potential by adopting transformer architectures and self-supervised pre-training methods from large language models (e.g., masked prediction) to learn representations from diverse EEG data, followed by fine-tuning on specific EEG tasks. Nonetheless, these large models often incurred high computational costs during both training and inference, with only marginal performance improvements as the model size increases. In this work, we proposed an EEG representation learning framework building upon Generative Diffusion Model (EEGDM). Specifically, we developed a structured state-space model for diffusion pretraining (SSMDP) to better capture the temporal dynamics of EEG signals and trained it using Denoising Diffusion Probabilistic Model (DDPM) framework. Subsequently, the resulting latent EEG representations were then used for downstream classification tasks via our proposed latent fusion transformer (LFT). To evaluate our method, we used multi-event datasets covering both interictal epileptiform discharges (TUEV) and seizure (CHB-MIT) detection, and compared EEGDM with current state-of-the-art approaches, including EEG FMs. Empirical results showed that our method outperformed the existing methods. These findings suggested that EEGDM offered a promising alternative to current FMs. Our source code and checkpoint are available at: https://github.com/jhpuah/EEGDM.

脑电图扩散模型表征学习

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