Abstract
Introduction
Manual sleep stage scoring is time consuming and subjective. Therefore, several studies focused on developing automated sleep scoring algorithms. The previously reported the automatic sleep scoring have been develop usually using small dataset, which less than 100 subjects. In this study, an automatic sleep scoring system based on ensemble convolutional neural network (ensemble-CNN) and spectrogram of sleep physiological signal was proposed and evaluated using a large dataset with sleep disorder.
Methods
The spectrograms were computed from each 30-s EEG and EOG of 994 subjects from PhysioNet 2018 challenge dataset, using the continuous wavelet transform, which were fed into an ensemble-CNN classification for training. The ensemble-CNN contained five pretrained models, ResNet-101, Inception-v4, DenseNet-201, Xception, and NASNet models, because these models’ architectures are different which can learn different features from the spectrograms to obtain high accuracy. The probabilities of five models were averaged to decide the sleep stage for each spectrogram. After classifying sleep stage, a smoothing process was used for sleep continuity. Moreover, the total 80% data from PhysioNet dataset were randomly assigned to the training set, and the remaining data were assigned to the testing set.
Results
To validate the robustness of the proposed system, the validation procedure was repeated five times. The performance measures were averaged over the five runs. The overall agreement and kappa coefficient of the proposed method are 82% and 0.73, respectively. The sensitivity of the sleep stages of Wake, N1, N2, N3, and REM are 90.0%, 48.6%, 84.9%, 84.2%, and 81.9%, respectively.
Conclusion
The performance of the proposed method was achieved expert level, and it was noted that the ensemble-CNN is a promising solution for automatic sleep stage scoring. This method can assist clinical staff in reducing the time required for sleep stage scoring in the future.
Support
This work was supported by the Ministry of Science and Technology, Taiwan. (MOST 106-2218-E-035-013-MY2, 108-2221-E-035-064, and 108-2634-F-006-012).