用自监督学习让可穿戴脑电设备少标注数据也能达到临床级睡眠分期准确率。
A Systematic Evaluation of Self-Supervised Learning for Label-Efficient Sleep Staging with Wearable EEG
- 设计专用自监督流水线,利用海量无标签可穿戴脑电信号提升模型性能。
- 仅用5%-10%标注数据就达80%以上准确率,比传统方法少用一半标签。
- 在真实家庭场景数据上验证,适合资源有限的睡眠监测系统落地。
可穿戴脑电设备正成为多导睡眠图(PSG)的有前景替代方案。由于成本低、可扩展性强,其广泛应用产生了大量无法由临床医生规模化分析的无标签数据。而近期深度学习在睡眠分期中的成功依赖于大规模标注数据集。自监督学习(SSL)为此提供了解决方案,能利用无标签信号缓解标注稀缺问题,减少人工标注负担。本文首次系统评估了可穿戴脑电设备上的自监督学习用于睡眠分期的效果。我们构建了一个结构化基准框架,涵盖多种SSL范式,并提出一种针对可穿戴脑电领域的专用流水线,在两个使用Ikon Sleep可穿戴头带采集的数据集上进行评估:BOAS(含共识标签的高质量数据集)和HOGAR(大规模居家自录无标签数据集)。定义了三种评估场景以研究标签效率、表征质量与跨数据集泛化能力。结果表明,SSL在分类性能上相较监督基线最高提升10%,尤其在标签稀缺时优势明显。使用5%-10%的标注数据即可实现超过80%的临床级准确率,而监督方法需两倍标签量。此外,所提出的领域专用SSL流水线在所有场景中均优于评估的通用脑电基础模型。研究证明,自监督学习可实现高效标注的睡眠分期,降低对人工标注的依赖,推动低成本睡眠监测系统发展。
原文摘要 · Abstract (English)
Wearable EEG devices have emerged as a promising alternative to polysomnography (PSG). As affordable and scalable solutions, their widespread adoption results in the collection of massive volumes of unlabeled data that cannot be analyzed by clinicians at scale. Meanwhile, the recent success of deep learning for sleep scoring has relied on large annotated datasets. Self-supervised learning (SSL) offers an opportunity to bridge this gap, leveraging unlabeled signals to address label scarcity and reduce annotation effort. In this paper, we present the first systematic evaluation of SSL for sleep staging using wearable EEG. We introduce a structured benchmarking framework encompassing a range of SSL paradigms and propose a specialized pipeline tailored to the wearable EEG domain, evaluating them on two sleep databases acquired with the Ikon Sleep wearable headband: BOAS, a high-quality benchmark containing PSG and wearable EEG recordings with consensus labels, and HOGAR, a large collection of home-based, self-recorded, and unlabeled recordings. Three evaluation scenarios are defined to study label efficiency, representation quality, and cross-dataset generalization. Results show that SSL consistently improves classification performance by up to 10% over supervised baselines, with gains particularly evident when labeled data is scarce. SSL achieves clinical-grade accuracy above 80% leveraging only 5% to 10% of labeled data, while the supervised approach requires twice the labels. Additionally, the proposed domain-specific SSL pipeline outperforms the evaluated general-purpose EEG foundation models across all scenarios. Our findings demonstrate the potential of SSL to enable label-efficient sleep staging with wearable EEG, reducing reliance on manual annotations and advancing the development of affordable sleep monitoring systems.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。