Research Area:  Machine Learning

Abstract:

Sleep staging is an essential step in the diagnosis and treatment of sleep-related diseases. Currently, most supervised learning models face the problem of insufficient labeled data. In addition, most sleep staging models are based on multi-channel EEG, and the models are too complex to be suitable for home sleep monitoring scenarios. To tackle these problems, this study proposes a sleep staging method based on pseudo-label optimization and a single-channel sleep hybrid neural network called SHNN. In the SHNN model, we design a multi-scale convolutional neural network (CNN) to extract the features from the single-channel EEG and use a Bi-directional recurrent gating unit (Bi-GRU) to obtain temporal context information of sleep data sequences. Extensive experiments based on the single-channel EEG (FPz-Cz, Pz-Oz, and Cz-A1) of the Sleep-EDFx and the DREAMS-SUB datasets validate the effectiveness of the SHNN model and the pseudo-label optimization algorithm therein outperforming current single-channel methods regarding the accuracy, , and MF1 Score. Moreover, the pseudo-label optimization algorithm can achieve good results on other sleep staging methods.

Keywords:  
Models face
Neural network
Convolutional neural network
SHNN model
Optimization algorithm

Author(s) Name:  Yongqing Zhang, Wenpeng Cao, Lixiao Feng, Tianyu Geng

Publisher name:  Elsevier

DOI:  10.1016/j.eswa.2022.119288

Volume Information:  Volume 213

Paper Link:   https://www.sciencedirect.com/science/article/abs/pii/S0957417422023065