A deep learning algorithm based on 1D CNN-LSTM for automatic sleep staging

Abstract

BACKGROUND: Sleep staging is an important part of sleep research. Traditional automatic sleep staging based on machine learning requires extensive feature extraction and selection. OBJECTIVE: This paper proposed a deep learning algorithm without feature extraction based on one-dimensional convolutional neural network and long short-term memory. METHODS: The algorithm can automatically divide sleep into 5 phases including awake period, non-rapid eye movement sleep period (N1 ∼ N3) and rapid eye movement using the electroencephalogram signals. The raw signal was processed by the wavelet transform. Then, the processed signal was directly input into the deep learning algorithm to obtain the staging result. RESULTS: The accuracy of staging is 93.47% using the Fpz-Cz electroencephalogram signal. When using the Fpz-Cz and electroencephalogram signal, the algorithm can obtain the highest accuracy of 94.15%. CONCLUSION: These results show that this algorithm is suitable for different physiological signals and can realize end-to-end automatic sleep staging without any manual feature extraction.