Advertisement

Automatic Sleep Staging Based on Deep Neural Network Using Single Channel EEG

  • Yongfeng Huang
  • Yujuan ZhangEmail author
  • Cairong Yan
Conference paper
Part of the Communications in Computer and Information Science book series (CCIS, volume 1027)

Abstract

Sleep staging is the first step for sleep research and sleep disorder diagnosis. The present study proposes an automatic sleep staging model, named ResSleepNet, using raw single-channel EEG signals. Most of the existing studies utilize hand-engineered features to identify sleep stages. These methods may ignore some important features of the signals, and then influence the effect of sleep stage classification. Instead of hand-engineering features, we combine feature extraction and classification into an algorithm based on residual network and bidirectional long short-term memory network. In the proposed method, we develop a 22-layer deep network to automatically learn features from the raw single-channel EEG and classify sleep stages. Residual network can learn time-invariant features, and bidirectional long short-term memory can add learned transition rules among sleep stages to the network. The model ResSleepNet is tested on the Sleep-EDF database. We perform 10 experiments and get average overall accuracy of 90.82% and 91.75% for 6-state and 5-state classification of sleep stages. Experimental results show the performance of our model is better than the state-of-the-art sleep staging methods, and it yields high detection accuracy for identifying sleep stage S1 and REM. In addition, our model is also suitable for extracting features from other signals (EOG, EMG) for sleep stage classification.

Keywords

Residual network Bidirectional long short-term memory Sleep stage classification Deep learning Single channel EEG 

References

  1. 1.
    Hassan, A.R., Bhuiyan, M.I.H.: Computer-aided sleep staging using complete ensemble empirical mode decomposition with adaptive noise and boot strap aggregating. Biomed. Signal Process. Control 24, 1–10 (2016)CrossRefGoogle Scholar
  2. 2.
    Hsu, Y.L., Yang, Y.T., Wang, J.S., Hsu, C.Y.: Automatic sleep stage recurrent neural classifier using energy features of EEG signals. Neurocomputing 104, 105–114 (2013)CrossRefGoogle Scholar
  3. 3.
    Berry, R.B.: Rules for scoring respiratory events in sleep. J. Clin. Sleep Med. JCSM Off. Publ. Am. Acad. Sleep Med. 8(5), 597–619 (2012)Google Scholar
  4. 4.
    Osorioforero, A.: Automatic sleep stages classification using EEG entropy features and unsupervised pattern analysis techniques. Entropy 16(12), 6573–6589 (2014)CrossRefGoogle Scholar
  5. 5.
    Lan, K.C., Chang, D.W., Kuo, C.E., Wei, M.Z.: Using off-the-shelf lossy compression for wireless home sleep staging. J. Neurosci. Methods 246, 142–152 (2015)CrossRefGoogle Scholar
  6. 6.
    Hassan, A.R., Bhuiyan, M.I.H.: Automatic sleep scoring using statistical features in the EMD domain and ensemble methods. Biocybern. Biomed. Eng. 36(1), 248–255 (2016)CrossRefGoogle Scholar
  7. 7.
    Imtiaz, S.A., Rodriguez-Villegas, E.: A low computational cost algorithm for REM sleep detection using single channel EEG. Ann. Biomed. Eng. 42(11), 2344–2359 (2014)CrossRefGoogle Scholar
  8. 8.
    Liang, S.: Automatic stage scoring of single-channel sleep eeg by using multiscale entropy and autoregressive models. IEEE Trans. Instrum. Meas. 61(6), 1649–1657 (2012)CrossRefGoogle Scholar
  9. 9.
    Aboalayon, K.A.I., Faezipour, M., Almuhammadi, W.S., Moslehpour, S.: Sleep stage classification using EEG signal analysis: a comprehensive survey and new investigation. Entropy 18(9), 272 (2016)CrossRefGoogle Scholar
  10. 10.
    Hassan, A.R., Subasi, A.: A decision support system for automated identification of sleep stages from single-channel EEG signals. Knowl.-Based Syst. 128, 115–124 (2017)CrossRefGoogle Scholar
  11. 11.
    Zhu, G., Li, Y., Wen, P.P.: Analysis and classification of sleep stages based on difference visibility graphs from a single-channel eeg signal. IEEE J. Biomed. Health Inf. 18(6), 1813–1821 (2014)CrossRefGoogle Scholar
  12. 12.
    Jia, X., Shen, L., Zhou, X., Yu, S.: Deep convolutional neural network based HEp-2 cell classification. In: International Conference on Pattern Recognition, pp. 77–80 (2017)Google Scholar
  13. 13.
    Xiong, W., Droppo, J., Huang, X., Seide, F., Seltzer, M., Stolcke, A.: Achieving human parity in conversational speech recognition. IEEE/ACM Trans. Audio Speech Lang. Process. PP(99)Google Scholar
  14. 14.
    Tsinalis, O., Matthews, P.M., Guo, Y., Zafeiriou, S.: Automatic sleep stage scoring with single-channel EEG using convolutional neural networksGoogle Scholar
  15. 15.
    He, K., Zhang, X., Ren, S., Sun, J.: Deep residual learning for image recognition. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 770–778 (2015)Google Scholar
  16. 16.
    Simonyan, K., Zisserman, A.: Very deep convolutional networks for large-scale image recognition. Comput. Sci.Google Scholar
  17. 17.
    Szegedy, C., Liu, W., Jia, Y., Sermanet, P., Reed, S.: Going deeper with convolutions. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 1–9 (2014)Google Scholar
  18. 18.
    Rajpurkar, P., Hannun, A.Y., Haghpanahi, M., Bourn, C., Ng, A.Y.: Cardiologist-level arrhythmia detection with convolutional neural networksGoogle Scholar
  19. 19.
    Supratak, A., Dong, H., Wu, C., Guo, Y.: DeepSleepNet: a model for automatic sleep stage scoring based on raw single-channel EEG. IEEE Trans. Neural Syst. Rehabil. Eng. 25(11), 1998–2008 (2017)CrossRefGoogle Scholar
  20. 20.
    Glorot, X., Bengio, Y.: Understanding the difficulty of training deep feed-forward neural networks. J. Mach. Learn. Res. 9, 249–256 (2010)Google Scholar
  21. 21.
    Ioffe, S., Szegedy, C.: Batch normalization: accelerating deep network training by reducing internal covariate shift, pp. 448–456 (2015)Google Scholar
  22. 22.
    Hinton, G.E., Srivastava, N., Krizhevsky, A., Sutskever, I., Salakhutdinov, R.R.: Improving neural networks by preventing co-adaptation of feature detectors. Comput. Sci. 3(4), 212–223 (2012)Google Scholar
  23. 23.
    Krizhevsky, A., Sutskever, I., Hinton, G.E.: ImageNet classification with deep convolutional neural networks. In: International Conference on Neural Information Processing Systems, pp. 1097–1105 (2012)Google Scholar
  24. 24.
    Goldberger, A.L., Amaral, L.A., Glass, L., Hausdorff, J.M., Ivanov, P.C.: PhysioBank, PhysioToolkit, and PhysioNet: components of a new research resource for complex physiologic signals. Circulation 101(23), E215 (2000)CrossRefGoogle Scholar
  25. 25.
    Hassan, A.R., Bashar, S.K., Bhuiyan, M.I.H.: On the classification of sleep states by means of statistical and spectral features from single channel electroencephalogram. In: International Conference on Advances in Computing, Communications and Informatics (2015)Google Scholar
  26. 26.
    Hassan, A.R., Bashar, S.K., Bhuiyan, M.I.H.: Automatic classification of sleep stages from single-channel electroencephalogram. In: India Conference (2016)Google Scholar
  27. 27.
    Tsinalis, O., Matthews, P.M., Guo, Y.: Automatic sleep stage scoring using time-frequency analysis and stacked sparse autoencoders. Ann. Biomed. Eng. 44(5), 1587–1597 (2015)CrossRefGoogle Scholar
  28. 28.
    Vilamala, A., Madsen, K.H., Hansen, L.K.: Deep convolutional neural networks for interpretable analysis of EEG sleep stage scoring, pp. 1–6 (2017)Google Scholar

Copyright information

© Springer Nature Switzerland AG 2019

Authors and Affiliations

  1. 1.School of Computer Science and TechnologyDonghua UniversityShanghaiChina

Personalised recommendations