A Deep Learning-Based Method for Sleep Stage Classification Using Physiological Signal

  • Guanjie Huang
  • Chao-Hsien ChuEmail author
  • Xiaodan Wu
Conference paper
Part of the Lecture Notes in Computer Science book series (LNCS, volume 10983)


A huge number of people suffers from different types of sleep disorders, such as insomnia, narcolepsy, and apnea. A correct classification of their sleep stage is a prerequisite and essential step to effectively diagnose and treat their sleep disorders. Sleep stages are often scored by experts through manually inspecting the patients’ polysomnography which are usually needed to be collected in hospitals. It is very laborious for experts and discommodious for patients to go through the process. Accordingly, current studies focused on automatically identifying the sleep stages and nearly all of them need to use hand-crafted features to achieve a decent performance. However, the extraction and selection of these features are time-consuming and require domain knowledge. In this study, we adopt and present a deep learning approach for automatic sleep stage classification using physiological signal. Convolutional Neural Network (CNN) and Long Short-Term Memory (LSTM) of popular deep learning models are employed to automatically learn features from raw physiological signals and identify the sleep stages. Our experiments shown that the proposed deep learning-based method has better performance than previous work. Hence, it can be a promising tool for patients and doctors to monitor the sleep condition and diagnose the sleep disorder timely.


Sleep stage classification Deep learning Physiological signal Sleep disorders Feature extraction 


  1. 1.
    Abadi, M., et al.: TensorFlow: a system for large-scale machine learning. In: OSDI, pp. 265–283 (2016)Google Scholar
  2. 2.
    Ancoli-Israel, S., et al.: The AASM manual for the scoring of sleep and associated events: rules, terminology and technical specifications. American Academy of Sleep Medicine, Westchester (2007)Google Scholar
  3. 3.
    American Sleep Disorders Association, et al.: The international classification of sleep disorders: diagnostic and coding manual. American Sleep Disorders Association (1990)Google Scholar
  4. 4.
    Basunia, M., et al.: Relationship of symptoms with sleep-stage abnormalities in obstructive sleep apnea-hypopnea syndrome. J. Community Hosp. Intern. Med. Perspect. 6(4), 32170 (2016)CrossRefGoogle Scholar
  5. 5.
    Bianchi, M.T., et al.: Obstructive sleep apnea alters sleep stage transition dynamics. PLoS One 5(6), e11356 (2010)CrossRefGoogle Scholar
  6. 6.
    Carskadon, M.A., Rechtschaffen, A.: Monitoring and staging human sleep. Princ. Pract. Sleep Med. 3, 1197–1215 (2000)Google Scholar
  7. 7.
    Chapotot, F., Becq, G.: Automated sleep–wake staging combining robust feature extraction, artificial neural network classification, and flexible decision rules. Int. J. Adapt. Control. Signal Process. 24(5), 409–423 (2010)MathSciNetzbMATHGoogle Scholar
  8. 8.
    Chollet, F., et al.: Keras (2015)Google Scholar
  9. 9.
    Dong, H., et al.: Mixed neural network approach for temporal sleep stage classification. IEEE Trans. Neural Syst. Rehabil. Eng. 26(2), 324–333 (2018)CrossRefGoogle Scholar
  10. 10.
    Ebrahimi, F., et al.: Automatic sleep stage classification based on EEG signals by using neural networks and wavelet packet coefficients. In: 2008 30th Annual International Conference of the IEEE Engineering in Medicine and Biology Society, EMBS 2008, pp. 1151–1154 (2008)Google Scholar
  11. 11.
    Fonseca, P., et al.: Sleep stage classification with ECG and respiratory effort. Physiol. Meas. 36(10), 2027 (2015)CrossRefGoogle Scholar
  12. 12.
    Fraiwan, L., et al.: Time frequency analysis for automated sleep stage identification in fullterm and preterm neonates. J. Med. Syst. 35(4), 693–702 (2011)CrossRefGoogle Scholar
  13. 13.
    Goldberger, A.L., et al.: Physiobank, physiotoolkit, and physionet. Circulation 101(23), e215–e220 (2000)CrossRefGoogle Scholar
  14. 14.
    Goodfellow, I., et al.: Deep learning. In: ICML2013 Tutor, pp. 1–800 (2011)Google Scholar
  15. 15.
    Güneş, S., et al.: Efficient sleep stage recognition system based on EEG signal using k-means clustering based feature weighting. Expert Syst. Appl. 37(12), 7922–7928 (2010)CrossRefGoogle Scholar
  16. 16.
    Hassan, A.R., et al.: Automatic classification of sleep stages from single-channel electroencephalogram, pp. 1–6, November 2015Google Scholar
  17. 17.
    Himanen, S.-L., Hasan, J.: Limitations of rechtschaffen and kales. Sleep Med. Rev. 4(2), 149–167 (2000)CrossRefGoogle Scholar
  18. 18.
    Hobson, J.A.: A Manual of Standardized Terminology, Techniques and Scoring System for Sleep Stages of Human Subjects. In: Rechtschaffen, A., Kales, A. (eds.) 58 p. Public Health Service, US Government Printing Office, Washington, DC (1968)). (Electroencephalogr. Clin. Neurophysiol. 26(6), 644 (1969))Google Scholar
  19. 19.
    Hochreiter, S., Schmidhuber, J.: Long short-term memory. Neural Comput. 9(8), 1735–1780 (1997)CrossRefGoogle Scholar
  20. 20.
    Hsu, Y.-L., et al.: Automatic sleep stage recurrent neural classifier using energy features of EEG signals. Neurocomputing 104, 105–114 (2013)CrossRefGoogle Scholar
  21. 21.
    Huang, C.-S., et al.: Knowledge-based identification of sleep stages based on two forehead electroencephalogram channels. Front. Neurosci. 8, 263 (2014)Google Scholar
  22. 22.
    Humphrey, E.J., et al.: Feature learning and deep architectures: new directions for music informatics. J. Intell. Inf. Syst. 41(3), 461–481 (2013)CrossRefGoogle Scholar
  23. 23.
    Kingma, D.P., Ba, J.: Adam: a method for stochastic optimization. arXiv Prepr. arXiv:1412.6980 (2014)
  24. 24.
    Lajnef, T., et al.: Learning machines and sleeping brains: automatic sleep stage classification using decision-tree multi-class support vector machines. J. Neurosci. Methods 250, 94–105 (2015)CrossRefGoogle Scholar
  25. 25.
    Le, H., Borji, A.: What are the receptive, effective receptive, and projective fields of neurons in convolutional neural networks? arXiv Prepr. arXiv:1705.07049(2017)
  26. 26.
    Li, X., et al.: Hyclasss: a hybrid classifier for automatic sleep stage scoring. IEEE J. Biomed. Health Inform. 22(2), 375–385 (2018)CrossRefGoogle Scholar
  27. 27.
    Ng, A.K., Guan, C.: Impact of obstructive sleep apnea on sleep-wake stage ratio. In: 2012 Annual International Conference of the IEEE Engineering in Medicine and Biology Society (EMBC), pp. 4660–4663 (2012)Google Scholar
  28. 28.
    Radha, M., et al.: Comparison of feature and classifier algorithms for online automatic sleep staging based on a single EEG signal. In: 2014 36th Annual International Conference of the IEEE Engineering in Medicine and Biology Society (EMBC), pp. 1876–1880 (2014)Google Scholar
  29. 29.
    Roebuck, A., et al.: A review of signals used in sleep analysis. Physiol. Meas. 35(1), R1 (2013)MathSciNetCrossRefGoogle Scholar
  30. 30.
    Rossow, A.B., et al.: Automatic sleep staging using a single-channel EEG modeling by Kalman filter and HMM. In: Biosignals and Biorobotics Conference (BRC), 2011 ISSNIP, pp. 1–6 (2011)Google Scholar
  31. 31.
    Sanei, S., Chambers, J.A.: EEG Signal Processing. Wiley, Hoboken (2013)CrossRefGoogle Scholar
  32. 32.
    Schuster, M., Paliwal, K.K.: Bidirectional recurrent neural networks. IEEE Trans. Signal Process. 45(11), 2673–2681 (1997)CrossRefGoogle Scholar
  33. 33.
    Şen, B., et al.: A comparative study on classification of sleep stage based on EEG signals using feature selection and classification algorithms. J. Med. Syst. 38(3), 18 (2014)CrossRefGoogle Scholar
  34. 34.
    da Silveira, T.L., et al.: Single-channel EEG sleep stage classification based on a streamlined set of statistical features in wavelet domain. Med. Biol. Eng. Comput. 19, 19 (2016)Google Scholar
  35. 35.
    Supratak, A., et al.: 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
  36. 36.
    Tsinalis, O., et al.: Automatic sleep stage scoring with single-channel EEG using convolutional neural networks. arXiv Prepr. arXiv:1610.01683 (2016)
  37. 37.
    Wang, K., et al.: Research on healthy anomaly detection model based on deep learning from multiple time-series physiological signals. Sci. Program. 2016, 9 (2016)Google Scholar
  38. 38.
    Yilmaz, B., et al.: Sleep stage and obstructive apneaic epoch classification using single-lead ECG. Biomed. Eng. Online 9(1), 39 (2010)CrossRefGoogle Scholar
  39. 39.
    Zhang, X., et al.: Sleep stage classification based on multi-level feature learning and recurrent neural networks via wearable device. arXiv Prepr. arXiv:1711.00629 (2017)

Copyright information

© Springer Nature Switzerland AG 2018

Authors and Affiliations

  1. 1.Pennsylvania State UniversityUniversity ParkUSA
  2. 2.Hebei University of TechnologyTianjinPeople’s Republic of China

Personalised recommendations