Abstract:Sleep stage classification is a necessary fundamental method for the diagnosis of sleep diseases, which has attracted extensive attention in recent years. Traditional methods for sleep stage classification, such as manual marking methods and machine learning algorithms, have the limitations of low efficiency and defective generalization. Recently, deep neural networks have shown improved results by the capability of learning complex pattern in the sleep data. However, these models ignore the intra-temporal sequential information and the correlation among all channels in each segment of the sleep data. To solve these problems, a hybrid attention temporal sequential network model is proposed in this paper, choosing recurrent neural network to replace traditional convolutional neural network, and extracting temporal features of polysomnography from the perspective of time. Furthermore, intra-temporal attention mechanism and channel attention mechanism are adopted to achieve the fusion of the intra-temporal representation and the fusion of channel-correlated representation. And then, based on recurrent neural network and inter-temporal attention mechanism, this model further realized the fusion of inter-temporal contextual representation. Finally, the end-to-end automatic sleep stage classification is accomplished according to the above hybrid representation. This paper evaluates the proposed model based on two public benchmark sleep datasets downloaded from open-source website, which include a number of polysomnography. Experimental results show that the proposed model could achieve better performance compared with ten state-of-the-art baselines. The overall accuracy of sleep stage classification could reach 0.801, 0.801 and 0.717, respectively. Meanwhile, the macro average F1-scores of the proposed model could reach 0.752, 0.728 and 0.700. All experimental results could demonstrate the effectiveness of the proposed model.

Convolution- and Attention-Based Neural Network for Automated Sleep Stage Classification

Automatic Classification of Sleep Stages Based on Raw Single-Channel EEG

CareSleepNet: A Hybrid Deep Learning Network for Automatic Sleep Staging

Application of Convolutional Neural Network-Based Biosensor and Electroencephalogram Signal in Sleep Staging

Automatic Sleep Stage Classification Based on Convolutional Neural Network and Fine-Grained Segments.

Learning a Convolutional Neural Network for Sleep Stage Classification.

A Novel Sleep Staging Network Based on Multi-Scale Dual Attention

Deep Convolutional Network Method For Automatic Sleep Stage Classification Based On Neurophysiological Signals

Time-Frequency Convolutional Neural Network for Automatic Sleep Stage Classification Based on Single-Channel EEG

A Compact Convolutional Neural Network for Sleep Stage Classification

MCASleepNet: Multimodal Channel Attention-Based Deep Neural Network for Automatic Sleep Staging

Convolutional Neural Network Based Sleep Stage Classification with Class Imbalance

Sleep Staging Based on Multi Scale Dual Attention Network

Computation-Efficient Multi-Model Deep Neural Network for Sleep Stage Classification.

Automatic Sleep Stage Scoring with Single-Channel EEG Using Convolutional Neural Networks

Dual-modal and Multi-Scale Deep Neural Networks for Sleep Staging Using EEG and ECG Signals.

Deep Learning for Sleep Stage Classification

A hybrid attention temporal sequential network for sleep stage classification

A Temporal-Spectral Fused and Attention-based Deep Model for Automatic Sleep Staging

Automated multi-model deep neural network for sleep stage scoring with unfiltered clinical data

Automatic and Accurate Sleep Stage Classification via a Convolutional Deep Neural Network and Nanomembrane Electrodes