4 个月前

NeuroNet:一种用于单通道EEG睡眠阶段分类的新型混合自监督学习框架

NeuroNet:一种用于单通道EEG睡眠阶段分类的新型混合自监督学习框架

摘要

睡眠阶段分类是诊断睡眠障碍和评估睡眠质量的关键方面。然而,传统的手动评分过程由临床医生进行,耗时且容易受到人为偏见的影响。近年来,深度学习的进展显著推动了睡眠阶段分类的自动化。尽管如此,仍存在一些挑战,包括对大量带标签数据集的需求以及人类生成注释中的固有偏见。本文介绍了一种名为NeuroNet的自监督学习(SSL)框架,该框架通过整合对比学习任务和掩码预测任务,有效利用未标记的单通道睡眠脑电图(EEG)信号。通过在三个多导睡眠图(PSG)数据集上进行的广泛实验,NeuroNet展示了优于现有自监督学习方法的性能。此外,本研究提出了一种基于Mamba的时间上下文模块,用于捕捉不同EEG时间段之间的关系。将NeuroNet与基于Mamba的时间上下文模块结合使用,即使在标签数据有限的情况下,也能达到甚至超过最新的监督学习方法的性能。本研究有望在睡眠阶段分类领域建立新的基准,为未来的睡眠分析研究和应用提供指导。

代码仓库

dlcjfgmlnasa/NeuroNet
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
sleep-stage-detection-on-isruc-sleepNeuroNet (C4-A1 only)
Accuracy: 77.05%
Macro-F1: 0.734
sleep-stage-detection-on-isruc-sleep-singleNeuroNet (C4-A1 only)
Accuracy: 77.05%
Macro-F1: 0.734
sleep-stage-detection-on-shhsNeuroNet (C4-A1 only)
Accuracy: 86.88%
Macro-F1: 0.812
sleep-stage-detection-on-shhs-single-channelNeuroNet (C4-A1 only)
Accuracy: 86.88%
Macro-F1: 0.812
sleep-stage-detection-on-sleep-edfxNeuroNet (Fpz-Cz only)
Accuracy: 85.24%
Macro-F1: 0.798
sleep-stage-detection-on-sleep-edfx-singleNeuroNet (Fpz-Cz only)
Accuracy: 85.24%
Macro-F1: 0.798

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
NeuroNet:一种用于单通道EEG睡眠阶段分类的新型混合自监督学习框架 | 论文 | HyperAI超神经