The research of sleep staging based on single-lead electrocardiogram and deep neural network

The polysomnogram (PSG) analysis is considered the golden standard for sleep staging under the clinical environment. The electroencephalogram (EEG) signal is the most important signal for classification of sleep stages. However, in-vivo signal recording and analysis of EEG signal presents us with a few technical challenges. Electrocardiogram signals on the other hand, are easier to record, and can provide an attractive alternative for home sleep monitoring. In this paper we describe a method based on deep neural network (DNN), which can be used for the classification of the sleep stages into Wake (W), rapid-eye-movement (REM) and non-rapid-eye-movement (NREM) sleep stage. We apply the sleep stage stacked autoencoder to constitute a 4-layer DNN model. In order to test the accuracy of our method, eighteen PSGs from the MIT-BIH Polysomnographic Database were used. A total of 11 features were extracted from each electrocardiogram recording The experimental design employs cross-validation across subjects, ensuring the independence of the training and the test data. We obtained an accuracy of 77% and a Cohen's kappa coefficient of about 0.56 for the classification of Wake, REM and NREM.