使用单通道 EEG 的自动睡眠分期:利用基于注意力的循环神经网络学习序列特征
Automatic Sleep Stage Classification Using Single-Channel EEG: Learning Sequential Features with Attention-Based Recurrent Neural Networks
基于摘要分析。本文研究单通道 EEG 自动睡眠分期,提出以深层双向 RNN 和注意力机制学习 EEG 片段内部的序列特征,再用线性 SVM 完成分类,并引入通过 DNN 判别式学习的滤波器组用于预处理。 核心流程是将一个 EEG epoch 分解为多个短帧,转换为逐帧特征向量序列,并以 sequence-to-label 方式训练带注意力机制的 RNN。作者预期,判别式训练能使注意力层之后的高层特征向量编码输入序列中的分类信息,因此将训练好的网络用作特征提取器,提取该向量后交由线性 SVM 分类。这里明确建模的是单个 epoch 内部的帧序列,摘要未说明是否利用相邻 epoch 间的睡眠阶段转移信息。 另一个方法组成是通过 DNN 判别式学习滤波器组,并在序列特征学习之前对逐帧特征向量进行滤波。作者报告,这一步进一步改善了分类性能,完整方法在 Sleep-EDF 上表现良好;但摘要未提供具体指标、数值、对照基线或评估划分,不能据此量化增益,也不能判断结果属于被试内还是跨被试协议。 复现时需核对 EEG 通道选择、epoch 与短帧划分、逐帧特征构造、滤波器组的训练目标、RNN 与注意力的具体实现,以及 SVM 的训练和调参方式。Sleep-EDF 的使用版本、被试范围、类别处理、数据划分、消融及统计信息尚未验证。本文的直接贡献面向 EEG 睡眠分期,不能仅凭该任务的结果推断其对其他 BCI 任务有效。
值得核对其片段内序列特征学习、注意力聚合与线性 SVM 分类的组合机制,以及可学习滤波器组的增益;摘要尚不足以判断其跨被试泛化能力。
来源:OpenReview · EEG · openreview.net