arXiv:2607.11578cs.LGcs.AI2026-07

用自监督扩散模型在极少标注下实现高精度癫痫检测。

DiffEEG: A Self-Supervised Denoising Diffusion Model for Learning EEG Generic Representations

论文配图:DiffEEG: A Self-Supervised Denoising Diffusion Model for Learning EEG Generic Representations
图 1 · 摘自论文原文
  • 通过去噪扩散预训练学习脑电通用表征
  • 在极端不平衡数据中达85%加权F1
  • 适合临床部署的癫痫监测系统

基于深度学习的脑电图癫痫检测面临严重标注稀缺和类别极度不平衡问题,发作期事件占比不足10%。我们提出DiffEEG,一个960万参数的自监督基础模型,通过去噪扩散预训练和强化学习微调解决这两个问题。在Temple University Hospital Seizure Corpus(TUHSZ)的130万未标注片段上预训练,采用一维U-Net结合多头自注意力机制学习通用神经表征。下游任务中,强化决策层使用策略梯度优化直接最大化F1分数,优先提升对罕见发作事件的敏感性。在严格的患者级评估(279名患者,留一折交叉验证)下,四类发作亚型分类达到61%准确率和59% F1,二分类检测达81%准确率与85%加权F1,尽管发作前验仅6.7%,仍保持59%的临床可接受发作检出率。段级评估显示最高可达97.6%准确率,证明模型强大容量。结果表明,基于扩散的预训练结合度量感知强化学习,可在极低标注需求下实现临床可用的癫痫监测。

原文摘要 · Abstract (English)

Deep learning for EEG-based seizure detection faces critical challenges: severe annotation scarcity and extreme class imbalance, where ictal events comprise less than 10\% of clinical recordings. We present DiffEEG, a 9.6M-parameter self-supervised foundation model that addresses both limitations through denoising diffusion pre-training and reinforcement learning (RL)-based fine-tuning. Pre-trained on 1.3M unlabeled segments from the Temple University Hospital Seizure Corpus (TUHSZ), DiffEEG learns generic neural representations via a 1D U-Net with multi-head self-attention. For downstream adaptation, a reinforced decision layer employs policy gradient optimization to directly maximize F1-score, prioritizing sensitivity to rare seizure events over overall accuracy. Under strict patient-wise evaluation (279 patients, Leave-One-Fold-Out), DiffEEG achieves 61\% accuracy and 59\% F1 for 4-class seizure subtyping, and 81\% accuracy with 85\% weighted F1 for binary detection, maintaining clinically viable seizure recall (59\%) despite extreme imbalance (6.7\% prevalence). Segment-level evaluation establishes an upper bound of 97.6\% accuracy, confirming strong architectural capacity. DiffEEG demonstrates that diffusion-based pre-training combined with metric-aware reinforcement learning enables clinically deployable seizure monitoring with minimal labeled data requirements.

脑电分析自监督学习扩散模型癫痫检测

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