arXiv:2503.10362q-bio.QMcs.LG2025-03被引 6

自监督学习的脑电模型,让小样本下精准诊断成为可能

BioSerenity-E1: a self-supervised EEG model for medical applications

  • 用频谱分块+掩码预测,从海量无标签脑电数据中学通用特征
  • 在癫痫检测、异常分类等任务上达顶尖水平,小样本下提升显著
  • 适合医疗场景,尤其适用于标注数据稀缺的临床研究与应用

脑电图(EEG)是神经科诊断的重要工具,但其人工解读耗时且依赖稀缺的专业人才。为解决这一问题,本文提出BioSerenity-E1,首个面向临床脑电的自监督基础模型。该模型采用两阶段自监督预训练:先通过基于Transformer的VQ-VAE压缩脑电信号,重建对数多锥形谱投影;再进行70%掩码块预测,强制学习脑电信号的复杂时空依赖性。在三个临床任务中表现优异:癫痫检测(AUROC=0.926,灵敏度=0.909),正常/异常分类(专有数据AUPRC=0.970,TUH-Abnormal数据AUPRC=0.910),以及不平衡数据下的多类病理区分(加权F1=0.730)。在低数据场景下,仅使用10%以下数据训练,其AUPRC仍提升2%至17%,验证了模型在小样本中的强适应性。

原文摘要 · Abstract (English)

Electroencephalography (EEG) serves as an essential diagnostic tool in neurology; however, its accurate manual interpretation is a time-intensive process that demands highly specialized expertise, which remains relatively scarce and not consistently accessible. To address these limitations, the implementation of automated pre-screening and analysis systems for EEG data holds considerable promise. Advances in self-supervised learning made it possible to pre-train complex deep learning architectures on large volumes of unlabeled EEG data to learn generalizable representations, that can later be used to enhance performance on multiple tasks while needing less downstream data. In the present paper, we introduce BioSerenity-E1, the first of a family of self-supervised foundation models for clinical EEG applications that combines spectral tokenization with masked prediction to achieve state-of-the-art performance across relevant diagnostic tasks. The two-phase self-supervised pretraining framework initially acquires compressed EEG representations via a transformer-based VQ-VAE architecture designed to reconstruct log-multitaper spectral projections, then implements extensive (70% block) masked token prediction to force the model to learn complex spatiotemporal dependencies in EEG signals. BioSerenity-E1 achieves strong performance across three clinical tasks, either in line or above state-of-the-art methods: seizure detection (AUROC = 0.926, Sensitivity = 0.909), normal/abnormal classification (AUPRC = 0.970 on proprietary data; 0.910 on TUH-Abnormal), and multiclass pathology differentiation on unbalanced data (Weighted F1 = 0.730). The utility of BioSerenity-E1 is further confirmed in low-data regimes scenarios, showing clear improvements in AUPRC (from +2% to 17%) when trained on less than 10% of the available data.

脑电分析自监督学习医疗AI

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