中文

利用亚 epoch 特征的 epoch 内与 epoch 间时间上下文网络 (IITNet) 用于原始单通道 EEG 的自动睡眠分期

机器学习 2020-06-11 v2 信号处理 机器学习

摘要

我们提出了一种名为 IITNet 的深度学习模型,用于从原始单通道 EEG 中学习 epoch 内与 epoch 间时间上下文以实现自动睡眠分期。为了从称为 epoch 的半分钟 EEG 分类睡眠阶段,睡眠专家会考察睡眠相关事件并考虑所发现事件之间的转换规则。类似地,IITNet 通过残差神经网络在亚 epoch 级别提取代表性特征,并经由双向 LSTM 从该特征序列中捕获 epoch 内与 epoch 间时间上下文。针对三个数据集,研究了序列长度 (L) 从 1 增加到 10 时的性能。IITNet 取得了与其他 SOTA 结果相当的性能。最佳准确率、MF1 与 Cohen's kappa (κ\kappa) 在 SleepEDF (L=10) 上分别为 83.9%、77.6%、0.78,在 MASS (L=9) 上分别为 86.5%、80.7%、0.80,在 SHHS (L=10) 上分别为 86.7%、79.8%、0.81。即使使用四个 epoch,性能仍具可比性。与使用单个 epoch 相比,平均准确率与 MF1 分别提升了 2.48%p 与 4.90%p,N1、N2 与 REM 的 F1 分别提升了 16.1%p、1.50%p 与 6.42%p。超过四个 epoch 后,性能提升不显著。结果支持考虑最近两分钟原始单通道 EEG 可作为通过深度神经网络进行高效可靠睡眠分期的合理选择。此外,与基线的实验表明,引入基于深度残差网络的 epoch 内时间上下文学习有助于整体性能提升,并与 epoch 间时间上下文学习具有正向协同效应。

关键词

引用

@article{arxiv.1902.06562,
  title  = {Intra- and Inter-epoch Temporal Context Network (IITNet) Using Sub-epoch Features for Automatic Sleep Scoring on Raw Single-channel EEG},
  author = {Hogeon Seo and Seunghyeok Back and Seongju Lee and Deokhwan Park and Tae Kim and Kyoobin Lee},
  journal= {arXiv preprint arXiv:1902.06562},
  year   = {2020}
}

备注

First three authors contributed equally to this work; Accepted manuscript for Biomedical Signal Processing and Control (BSPC); 12 pages, 6 figures;