EEG Sleep Staging Model Using Wavelet Transform and Bidirectional Long Short-Term Memory Network

WANG Tianyu; CHEN Han; WANG Gang; WU Ning

doi:10.7652/xjtuxb202209011

您当前的位置：

首页 >

文章列表页 >

EEG Sleep Staging Model Using Wavelet Transform and Bidirectional Long Short-Term Memory Network

更新时间：2025-07-09

- EEG Sleep Staging Model Using Wavelet Transform and Bidirectional Long Short-Term Memory Network
- Vol. 56, Issue 9, Pages: 104-111(2022)
- 作者机构：
  
  1. 西安交通大学电子与信息学部,西安,710049
  2. 西安交通大学生物医学信息工程教育部重点实验室,西安,710049
  3. 西安交通大学生命科学与技术学院,西安,710049
- 作者简介：
- 基金信息：
- DOI：10.7652/xjtuxb202209011
  CLC： TN911.6
- Online First：10 September 2022，
  
  Published：2022
- 稿件说明：
移动端阅览
WANG Tianyu, CHEN Han, WANG Gang, et al. EEG Sleep Staging Model Using Wavelet Transform and Bidirectional Long Short-Term Memory Network[J]. 2022, 56(9): 104-111.
DOI：

WANG Tianyu, CHEN Han, WANG Gang, et al. EEG Sleep Staging Model Using Wavelet Transform and Bidirectional Long Short-Term Memory Network[J]. 2022, 56(9): 104-111. DOI： 10.7652/xjtuxb202209011.

摘要

针对睡眠生理信号采集难度大、睡眠分期精度低的问题

提出一种采用小波变换和双向长短期记忆网络的脑电睡眠分期模型。首先使用连续小波变换提取睡眠脑电的时频图; 然后使用卷积神经网络从脑电信号的时频图中提取睡眠相关的脑电特征

作为单个睡眠片段的分期依据

再使用双向长短期记忆网络进一步提取睡眠片段之间的状态转换规则; 最后利用深度学习方法建立特征、规则与睡眠阶段的映射

使用数据扩充和两步训练法训练模型

削弱数据不均衡的影响

完成连续片段的睡眠分期。采用SHHS公开数据库的5 793名被试者的睡眠脑电数据对该模型进行验证

实验结果表明

睡眠分期准确率达到85.82%

整体F1达到78.39

Kappa系数达到0.799

和现有方法相比性能明显提升。

Abstract

For the difficulty in acquisition of physiological signals and low accuracy of sleep staging

an EEG sleep staging model using wavelet transform and bidirectional long short-term memory network is proposed. First

the time-frequency map of raw sleep EEG is extracted by using continuous wavelet transform. Then

the sleep-related EEG features are extracted from the time-frequency map through the VGG convolutional network as the staging basis for a single sleep segment

and the transition rules of sleep state are further extracted by the bidirectional long short-term memory network. Finally

the deep learning method is used to establish the mapping of features

rules and sleep stages

and the data augmentation and two-step training methods are used to train the model to weaken the influence of data imbalance and complete the continuous sleep staging work. The model is verified by the sleep EEG data of 5 793 subjects from the public database SHHS. The experimental results show that the sleep staging accuracy of the model reaches 85.82%

the F1 reaches 78.39

and the Kappa index reaches 0.799. Compared with existing methods

the performance of the proposed model is significantly improved.

关键词

Keywords

references

赵忠新. 睡眠医学 [M]. 北京: 人民卫生出版社, 2016.

MOSER D, ANDERER P, GRUBER G, et al. Sleep classification according to AASM and Rechtschaffen Kales: effects on sleep scoring parameters [J]. Sleep, 2009, 32(2): 139-149.

徐富献, 王抒伟, 钱镶钰, 等. 睡眠自动分期方法综述 [J]. 生物物理学, 2019, 7(3): 34-48.

XU Fuxian, WANG Shuwei, QIAN Xiangyu, et al. A review of automatic sleep staging [J]. Biophysics, 2019, 7(3): 34-48.

KUO C E, CHEN Guanting. Automatic sleep staging based on a hybrid stacked LSTM neural network: verification using large-scale dataset [J]. IEEE Access, 2020, 8: 111837-111849.

PATANAIK A, ONG J L, GOOLEY J J, et al. An end-to-end framework for real-time automatic sleep stage classification [J]. Sleep, 2018, 41(5): zsy041.

PHAN H, ANDREOTTI F, COORAY N, et al. SeqSleepNet: end-to-end hierarchical recurrent neural network for sequence-to-sequence automatic sleep staging [J]. IEEE Transactions on Neural Systems and Rehabilitation Engineering, 2019, 27(3): 400-410.

PERSLEV M, JENSEN M H, DARKNER S, et al. U-Time: a fully convolutional network for time series segmentation applied to sleep staging [C]∥Proceedings of the 33rd International Conference on Neural Information Processing Systems. Red Hook, NY, USA: Curran Associates Inc., 2019: 4415-4426.

JIA Ziyu, LIN Youfang, WANG Jing, et al. SalientSleepNet: multimodal salient wave detection network for sleep staging [C]∥Proceedings of the 30th International Joint Conference on Artificial Intelligence. San Mateo, CA, USA: Morgan Kaufmann, 2021: 2614-2620.

WEI Yuhui, QI Xia, WANG Huaning, et al. A multi-class automatic sleep staging method based on long short-term memory network using single-lead electrocardiogram signals [J]. IEEE Access, 2019, 7: 85959-85970.

SUPRATAK A, DONG Hao, WU Chao, et al. DeepSleepNet: a model for automatic sleep stage scoring based on raw single-channel EEG [J]. IEEE Transactions on Neural Systems and Rehabilitation Engineering, 2017, 25(11): 1998-2008.

SUPRATAK A, GUO Yike. TinySleepNet: an efficient deep learning model for sleep stage scoring based on raw single-channel EEG [C]∥Proceedings of the 2020 42nd Annual International Conference of the IEEE Engineering in Medicine Biology Society(EMBC). Piscataway, NJ, USA: IEEE, 2020: 641-644.

KORKALAINEN H, AAKKO J, NIKKONEN S, et al. Accurate deep learning-based sleep staging in a clinical population with suspected obstructive sleep apnea [J]. IEEE Journal of Biomedical and Health Informatics, 2020, 24(7): 2073-2081.

QU Wei, WANG Zhiyong, HONG Hong, et al. A residual based attention model for EEG based sleep staging [J]. IEEE Journal of Biomedical and Health Informatics, 2020, 24(10): 2833-2843.

ZHOU Jinjin, WANG Guangsheng, LIU Junbiao, et al. Automatic sleep stage classification with single channel EEG signal based on two-layer stacked ensemble model [J]. IEEE Access, 2020, 8: 57283-57297.

JIANG Dihong, LU Yanan, MA Yu, et al. Robust sleep stage classification with single-channel EEG signals using multimodal decomposition and HMM-based refinement [J]. Expert Systems with Applications, 2019, 121: 188-203.

WACKER W, WITTE H. Time-frequency techniques in biomedical signal analysis [J]. Methods of Information in Medicine, 2013, 52(4): 279-296.

胡勤伟, 陶庆, 王妮妮, 等. 用于稳态视觉诱发电位目标识别的多尺度特征融合卷积神经网络方法 [J]. 西安交通大学学报, 2022, 56(4): 185-193, 202.

HU Qinwei, TAO Qing, WANG Nini, et al. A multi-scale feature fusion convolutional neural network approach for steady-state visual evoked potential target recognition [J]. Journal of Xi'an Jiaotong University, 2022, 56(4): 185-193, 202.

邵舒羽, 吴锦涛, 周前祥, 等. 睡眠-觉醒节律紊乱下注意诱发脑电特征分析与识别 [J]. 航天医学与医学工程, 2021, 34(6): 439-447.

SHAO Shuyu, WU Jintao, ZHOU Qianxiang, et al. EEG characteristics analysis and recognition induced by attention under sleep-wake rhythm disorder [J]. Space Medicine Medical Engineering, 2021, 34(6): 439-447.

SIMONYAN K, ZISSERMAN A. Very Deep convolutional networks for large-scale image recognition [C]∥Proceedings of the 3rd International Conference on Learning Representations. Amherst, Massachusetts, USA: OpenReview, 2015: 1-14.

BERRY R B. Fundamentals of sleep medicine [M]. Philadelphia, USA: Elsevier/Saunders, 2012.

陈保家, 陈学力, 沈保明, 等. CNN-LSTM深度神经网络在滚动轴承故障诊断中的应用 [J]. 西安交通大学学报, 2021, 55(6): 28-36.

CHEN Baojia, CHEN Xueli, SHEN Baoming, et al. An application of convolution neural network and long short-term memory in rolling bearing fault diagnosis [J]. Journal of Xi'an Jiaotong University, 2021, 55(6): 28-36.

SCHUSTER M, PALIWAL K K. Bidirectional recurrent neural networks [J]. IEEE Transactions on Signal Processing, 1997, 45(11): 2673-2681.

ZHANG Guoqiang, CUI Licong, MUELLER R, et al. The national sleep research resource: towards a sleep data commons [J]. Journal of the American Medical Informatics Association, 2018, 25(10): 1351-1358.

QUAN S F, HOWARD B V, IBER C, et al. The sleep heart health study: design, rationale, and methods [J]. Sleep, 1997, 20(12): 1077-1085.

ROY Y, BANVILLE H, ALBUQUERQUE I, et al. Deep learning-based electroencephalography analysis: a systematic review [J]. Journal of Neural Engineering, 2019, 16(5): 051001.

SUN Chenglu, FAN Jiahao, CHEN Chen, et al. A two-stage neural network for sleep stage classification based on feature learning, sequence learning, and data augmentation [J]. IEEE Access, 2019, 7: 109386-109397.

Views

下载量

CSCD

Alert me when the article has been cited

提交

Tools

Publicity Resources

Interpretable Mechanical Fault Diagnosis Method Based on Dual-Tree Complex Wavelet Transform

DDoS Attacks Detecting Algorithm Based on Continuous Wavelet Transforms with Reliability Evaluation

Pruning Based on Separable Convolutions for Object Tracking

Spatial Attention Mechanism with Global Characteristics

SPAM: Spatially Partitioned Attention Module in Deep Convolutional Neural Networks for Image Classification

Related Author

ZHANG Long

YU Jianping

SONG Yizhe

PAN Shihao

LI Gongzhun

PAN Yuanming

LIU Jiayan

杨新宇

Related Institution

School of Mechanical, Electrical and Vehicle Engineering, East China Jiaotong University

Department of Computer Science and Technology, Xi'an Jiaotong University

Xi’an Microelectronics Technology Institute

School of Information and Communications Engineering, Xi’an Jiaotong University

(School of Information and Communication Engineering, Xi'an Jiaotong University,,)

⁰