Skip to main content
Springer Nature Link
Log in

A Study of Human Sleep Stage Classification Based on Dual Channels of EEG Signal Using Machine Learning Techniques

  • Original Research
  • Published:
SN Computer Science Aims and scope Submit manuscript

Abstract

Sleep staging is one of the important methods for the diagnosis of the different types of sleep-related diseases. Manual inspection of sleep scoring is a very time-consuming process, labor-intensive, and requires more human interpretations, which may produce biased results. Therefore, in this paper, we propose an efficient automated sleep staging system to improve sleep staging accuracy. In this work, we extracted both linear and non-linear properties from the input signal. Next to that, a set of optimal features was selected from the extracted feature vector by using a feature reduction technique based on the ReliefF weight algorithm. Finally, the selected features were classified through four machine learning techniques like support vector machine, K-nearest neighbor, decision tree, and random forest. The proposed methodology performed using dual-channel EEG signals from the ISRUC-Sleep dataset under the AASM sleep scoring rules. The performance of the proposed methodology compared with the existing similar methods. In this work, we considered the 10-Fold cross validation strategy; our proposed methods reported the highest classification accuracy of 91.67% with the C4-A1 channel, and 93.8% with the O2-A1 channel using the Random forest classification model. The result of the proposed methodology outperformed the earlier contribution for two-class sleep states classification. The proposed dual-channel sleep staging method can be helpful for the clinicians during the sleep scoring and treatment for the different sleep-related diseases.

This is a preview of subscription content, log in via an institution to check access.

Access this article

Log in via an institution

Subscribe and save

Springer+ Basic
$34.99 /Month
  • Get 10 units per month
  • Download Article/Chapter or eBook
  • 1 Unit = 1 Article or 1 Chapter
  • Cancel anytime
Subscribe now

Buy Now

Price excludes VAT (USA)
Tax calculation will be finalised during checkout.

Instant access to the full article PDF.

Institutional subscriptions

Fig. 1
Fig. 2
Fig. 3
Fig. 4
Fig. 5
Fig. 6
Fig. 7

Similar content being viewed by others

Data availability statement

All EEG files are available from the ISRUC-SLEEP database (https://sleeptight.isr.uc.pt/ISRUC_Sleep/).

References

  1. Panossian LA, Avidan AY. Review of sleep disorders. Med Clin N Am. 2009;93:407–25. https://doi.org/10.1016/j.mcna.2008.09.001.

    Article  Google Scholar 

  2. Smaldone A, Honig JC, Byrne MW. Sleepless in America: inadequate sleep and relationships to health and well-being of our nation’s children. Pediatrics. 2007;119:29–37.

    Google Scholar 

  3. Hassan AR, Hassan Bhuiyan MI. Automatic sleep scoring using statistical features in the EMD domain and ensemble methods. Biocybern Biomed Eng. 2016. https://doi.org/10.1016/j.bbe.2015.11.001.

    Article  Google Scholar 

  4. Aboalayon KA, Ocbagabir HT, Faezipour M (2014) Efficient sleep stage classification based on EEG signals. In: IEEE long island systems, applications and technology conference, LISAT 2014. https://doi.org/10.1109/LISAT.2014.6845193

  5. Obayya M, AbouChadi F. Automatic classification of sleep stages using EEG records based on Fuzzy c-means (FCM) algorithm. In: Radio science conference (NRSC); 2014. pp. 265–272.

  6. Alickovic E, Subasi A. Ensemble SVM method for automatic sleep stage classification. IEEE Trans Instrum Meas. 2018. https://doi.org/10.1109/TIM.2018.2799059.

    Article  Google Scholar 

  7. Abeyratne UR, Swarnkar V, Rathnayake SI, Hukins C. Sleep-stage and event de- pendency of brain asynchrony as manifested through surface EEG. In: Proceedings of the 29th IEEE annual international conference of the engineering in medicine and biology society 2007, 709–712.

  8. Rechtschaffen A, Kales AA. Manual of standardized terminology, techniques and scoring systems for sleep stages of human Subjects. Washington, DC: U.G.P. Office, Public Health Service; 1968.

  9. Iber C, Ancoli-Israel S, Chesson AL, Quan SF. The AASM manual for the scoring of sleep and associated events: rules, terminology and technical specification. Darien: American Academy of Sleep Medicine; 2007.

    Google Scholar 

  10. Ll C, Zhao Y, Zhang J, Zou JZ. Automatic detection of alertness/drowsiness from physiological signals using wavelet-based nonlinear features and machine learning. Expert Syst Appl. 2015;42:7344–55.

    Google Scholar 

  11. da Silveira TL, Kozakevicius AJ, Rodrigues CR. Single-channel EEG sleep stage classification based on a streamlined set of statistical features in wavelet domain. Med Biol Eng Comput. 2016;55(2):1–10.

    Google Scholar 

  12. Amin HU, Malik AS, Kamel N, Hussain M. A novel approach based on data redundancy for feature extraction of EEG signals. Brain Topogr. 2016;29:207–17.

    Google Scholar 

  13. Diykh M, Li Y, Wen P. EEG sleep stages classification based on time domain features and structural graph similarity. IEEE Trans Neural Syst Rehabil Eng. 2016;24:1159–68.

    Google Scholar 

  14. Acharya UR, Sree SV, Swapna G, Martis RJ, Suri JS. Automated EEG analysis of epilepsy: a review. Knowl-Based Syst. 2013;45:147–65.

    Google Scholar 

  15. Aydın S, Tunga MA, Yetkin S. Mutual information analysis of sleep EEG in detecting psycho-physiological insomnia. J Med Syst. 2015;39:1–10.

    Google Scholar 

  16. Rahman MM, Bhuiyan MIH, Hassan AR. Sleep stage classification using single-channel EOG. Comput Biol Med. 2018;102:211–20.

    Google Scholar 

  17. Diykh M, Li Y, Wen P, Li T. Complex networks approach for depth of anesthesia assessment. Measurement. 2018;119:178–89.

    Google Scholar 

  18. Ebrahimi F, Mikaeili M, Estrada E, Nazeran H. 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. IEEE; 2008, pp. 1151–54.

  19. Hsu YL, Yang YT, Wang JS, Hsu CY. Automatic sleep stage recurrent neural classifier using energy features of EEG signals. Neurocomputing. 2013;104:105–14.

    Google Scholar 

  20. Peker M. An efficient sleep scoring system based on EEG signal using complex-valued machine learning algorithms. Neurocomputing. 2016;207:165–77.

    Google Scholar 

  21. Bianchi MT. Sleep devices: wearables and nearables, informational and interventional. Consum Clin Metabol Clin Exp. 2017;84:99–108.

    Google Scholar 

  22. Mendez MO, Matteucci S, Cerutti et al. Sleep staging classification based on HRV: time-variant analysis. In: Int. Conf. IEEE Eng. Med. Biol. Soc.; 2009. pp. 9–12.

  23. Radha M, Garcia-Molina G, Poel M, et al. Comparison of feature and classifier algorithms for online automatic sleep staging based on a single EEG signal. Eng Med Biol Soc. 2014;1:1876–80.

    Google Scholar 

  24. Stochholm A, Mikkelsen K, Kidmose P. Automatic sleep stage classification using ear-EEG. Eng Med Biol Soc 2016.

  25. Chriskos P, Kaitalidou DS, Karakasis G, et al. Automatic sleep stage classification applying machine learning algorithms on EEG recordings. In: 2017 IEEE 30th international symposium on computer-based medical systems; 2017. pp. 435–439.

  26. Imtiaz SA, Villegas ER. A low computational cost algorithm for REM sleep detection using single channel EEG. Ann Biomed Eng. 2014;42(11):2344–59.

    Google Scholar 

  27. Imtiaz SA, Jiang ZE, Villegas ER. An ultralow power system on chip for automatic sleep staging. IEEE J Solidst Circ. 2017;52(3):822–33.

    Google Scholar 

  28. Chen C, Liu X, Ugon A. Symbolic fusion: a novel decision support algorithm for sleep staging application. Mobihealth. 2015;16(8):19–22.

    Google Scholar 

  29. Sen B, Peker M, Cavusoglu A, et al. A comparative study on classification of sleep stage based on EEG signals using feature selection and classification algorithms. J Med Syst. 2014;38:18–687.

    Google Scholar 

  30. Liu H, Zhang H, Zhao P, et al. Study on sleep staging based on EEG signals. Chin J Biomed Eng. 2015;34(6):693–700.

    Google Scholar 

  31. Chriskos P, Kaitalidou DS, Karakasis G. et al. Automatic sleep stage classification applying machine learning algorithms on EEG recordings. In: 2017 IEEE 30th international symposium on computer-based medical systems; 2017, pp. 435–39.

  32. Huang CS, Lin CL, Ko LW, et al. A hierarchical classification system for sleep stage scoring via forehead EEG signals. In: 2013 IEEE symposium on computational intelligence, cognitive algorithms, mind, and brain (CCMB), IEEE; 2013. pp. 1–5.

  33. Zhu G, Yan L, Peng W. Analysis and classification of sleep stages based on difference visibility graphs from a single-channel EEG signal. IEEE J Biomed Health Inform. 2014;18(6):1813–21.

    Google Scholar 

  34. See AR, CK Liang. A study on sleep EEG using sample entropy and power spectrum analysis. In: Defense science research conference and expo (DSR), 2011, IEEE; 2011, pp. 1–4.

  35. Yu S, Li P, Lin H. et al. Support vector machine based detection of drowsiness using minimum EEG features. In: International conference on social computing (SocialCom), 2013. IEEE; 2013. pp. 827–35.

  36. Nguyen HD, Wilkins BA, Cheng Q, et al. An online sleep apnea detection method based on recurrence quantification analysis. IEEE J Biomed Health Inform. 2014;18(4):1285–93.

    Google Scholar 

  37. Bandarabadi M, Teixeira CA, Rasekhi J, et al. Epileptic seizure prediction using relative spectral power features. Clin Neurophysiol. 2015;126(2):237–48.

    Google Scholar 

  38. Liang SF, Kuo CE, Hu YH, et al. A rule-based automatic sleep staging method. J Neurosci Methods. 2012;205(1):169–76.

    Google Scholar 

  39. Phan H, Do Q, Vu DL. Metric learning for automatic sleep stage classification. In: 35th annual international conference of the IEEE engineering in medicine and biology society (EMBC), 2013, IEEE; 2013. pp. 5025–28.

  40. Ronzhina M, Janoušek O, Kolárová J, et al. Sleep scoring using artificial neural networks. Sleep Med Rev. 2012;16(3):251–63.

    Google Scholar 

  41. Hassan AR, Bhuiyan MIH. Automated identification of sleep states from EEG signals by means of ensemble empirical mode decomposition and random under sampling boosting. Comput Methods Progr Biomed. 2017;140:201–10.

    Google Scholar 

  42. Hassan AR. A comparative study of various classifiers for automated sleep apnea screening based on single-lead electrocardiogram. In: 2015 international conference on electrical & electronic engineering (ICEEE). IEEE; 2015. pp. 45–8.

  43. Hassan AR. Computer-aided obstructive sleep apnea detection using normal inverse Gaussian parameters and adaptive boosting. Biomed Signal Process Control. 2016;29:22–30.

    Google Scholar 

  44. Hassan AR. A comparative study of various classifiers for automated sleep apnea screening based on single-lead electrocardiogram. In: 2015 international conference on electrical & electronic engineering (ICEEE), IEEE; 2015. pp. 45–8.

  45. Roebuck A, Monasterio V, Gederi E, Osipov M, Behar J, Malhotra A, Clifford GD. A review of signals used in sleep analysis. Physiol Meas. 2013;35(1):R1–57. https://doi.org/10.1088/0967-3334/35/1/r1.

    Article  Google Scholar 

  46. Radha M, Garcia-Molina G, Poel M, Tononi G. Comparison of feature and classifier algorithms for online automatic sleep staging based on a single EEG signal. In: Proceedings of the 36th IEEE annual international conference of engineering in medicine and biology society, Chicago, IL, USA, 26–30 August; 2014. pp. 1876–80.

  47. Savareh A, et al. Performance comparison of machine learning techniques in sleep scoring based on wavelet features and neighboring component analysis. PeerJ. 2018;6:e5247. https://doi.org/10.7717/peerj.5247.

    Article  Google Scholar 

  48. Ronzhina M, Janoušek O, Kolářová J, Nováková M, Honzík P, Provazník I. Sleep scoring using artificial neural networks. Sleep Med Rev. 2012;16(3):251–63. https://doi.org/10.1016/j.smrv.2011.06.003.

    Article  Google Scholar 

  49. Jo HG, Park JY, Lee CK, An SK, Yoo SK. Genetic fuzzy classifier for sleep stage identification. Comput Biol Med. 2010;40:629–34.

    Google Scholar 

  50. Fraiwan L, Lweesy K, Khasawneh N, Wenz H, Dickhaus H. Automated sleep stage identification system based on time-frequency analysis of a single EEG channel and random forest classifier. Comput Methods Programs Biomed. 2012;108(1):10–9. https://doi.org/10.1016/j.cmpb.2011.11.005.

    Article  Google Scholar 

  51. Braun ET, Kozakevicius ADJ, Da Silveira TLT, Rodrigues CR, Baratto G. Sleep stages classification using spectral based statistical moments as features. Rev Inform Teór E Appl. 2018;25(1):11. https://doi.org/10.22456/2175-2745.74030.

    Article  Google Scholar 

  52. Aboalayon KAI, Ocbagabir HT, Faezipour M. Efficient sleep stage classification based on EEG signals. In: IEEE long island systems, applications and technology (LISAT) conference 2014, Farmingdale, NY, USA, May 2014. pp. 1–6. https://doi.org/10.1109/LISAT.2014.6845193.

  53. Sousa T, Cruz A, Khalighi S, Pires G, Nunes U. A two-step automatic sleep stage classification method with dubious range detection. Comput Biol Med. 2015;59:42–53.

    Google Scholar 

  54. Heyat MBB, Lai D, Zhang FIKY. Sleep bruxism detection using decision tree method by the combination of C4-P4 and C4-A1 channels of Scalp EEG’. IEEE Access; 2019. vol. 1–1.

  55. Basha AJ, Balaji BS, Poornima S, Prathilothamai M, Venkatachalam K. Support vector machine and simple recurrent network based automatic sleep stage classification of fuzzy kernel. J Ambient Intell Humaniz Comput. 2020. https://doi.org/10.1007/s12652-020-02188-4.

    Article  Google Scholar 

  56. Sharma M, Patel S, Choudhary S, Acharya UR. Automated detection of sleep stages using energy-localized orthogonal wavelet filter banks. Arab J Sci Eng. 2019. https://doi.org/10.1007/s13369-019-04197-8.

    Article  Google Scholar 

  57. Tian P, Hu J, Qi J, Ye X, Che D, Ding Y, Peng YA. Hierarchical classification method for automatic sleep scoring using multiscale entropy features and proportion information of sleep architecture. Biocybern Biomed Eng. 2017;37(2):263–71. https://doi.org/10.1016/j.bbe.2017.01.005.

    Article  Google Scholar 

  58. Alickovic E, Subasi A. Ensemble SVM method for automatic sleep stage classification. IEEE Trans Instrum Meas. 2018;67(6):1258–65. https://doi.org/10.1109/TIM.2018.2799059.

    Article  Google Scholar 

  59. Khalighi S, Sousa T, Santos JM, Nunes U. ISRUC-Sleep: a comprehensive public dataset for sleep researchers. In: Computer methods and programs in biomedicine; 2016. vol. 124, pp. 180–192.

  60. Kononenko I. Estimating attributes: analysis and extensions of RELIEF. ECML’94; 1994. pp. 171–182.

  61. Cortes C, Vapnik V. Support vector networks. Mach Learn. 1995;20(3):273–97.

    MATH  Google Scholar 

  62. Kavzoglu T, Colkesen I. Classification of satellite images using decision trees: Kocaeli case. Electron J Map Technol. 2010;2(1):36–45.

    Google Scholar 

  63. Ocak H. A medical decision support system based on support vector machines and the genetic algorithm for the evaluation of fetal wellbeing. J Med Syst. 2013;37(2):1–9.

    Google Scholar 

  64. Ilhan HO, Bilgin G. Sleep stage classification via ensemble and conventional machine learning methods using single channel EEG signals. Int J Intell Syst Appl Eng. 2017;5(4):174–84.

    Google Scholar 

  65. Shang S, Li X, Zong M, et al. Learning k, forkNN classification. ACM Trans Intell Syst Technol. 2017;8(3):1–19.

    Google Scholar 

  66. Fraley C, Raftery AE. Model-based clustering, discriminant analysis, and density estimation. J Am Stat Assoc. 2002;97(458):611–31.

    MathSciNet  MATH  Google Scholar 

  67. Shabani F, Kumar L, Solhjouy-Fard S. Variances in the projections, resulting from CLIMEX, boosted regression trees and random forests techniques. Theor Appl Climatol. 2017;97(458):1–14.

    Google Scholar 

  68. Sanders TH, McCurry M, Clements MA. Sleep stage classification with cross frequency coupling. In: 36th Annu. Int. Conf. IEEE Eng. Med. Biol. (EMBC); 2014, pp. 4579–82.

  69. Bajaj V, Pachori R. Automatic classification of sleep stages based on the time-frequency image of EEG signals. Comput Methods Programs Biomed. 2013;112(3):320–8.

    Google Scholar 

  70. Zibrandtsen I, Kidmose P, Otto M, Ibsen J, Kjaer TW. Case comparison of sleep features from ear-EEG and scalp-EEG. Sleep Sci. 2016;9(2):69–72.

    Google Scholar 

  71. Berry RB, Brooks R, Gamaldo CE, Hardsim SM, Lloyd RM, Marcus CL, Vaughn BV. The AASM manual for the scoring of sleep and associated events: rules, terminology and technical specifications. Darien: American Academy of Sleep Medicine; 2014.

    Google Scholar 

  72. Khalighi S, Sousa T, Oliveira D, Pires G, Nunes U. Efficient feature selection for sleep staging based on maximal overlap discrete wavelet transform and SVM. In: Annual international conference of the IEEE engineering in medicine and biology society; 2011.

  73. Simões H, Pires G, Nunes U, Silva V. Feature extraction and selection for automatic sleep staging using EEG. In: Proceedings of the 7th international conference on informatics in control, automation and robotics, vol. 3, pp. 128–33.

  74. Khalighi S, Sousa T, Santos JM, Nunes U. ISRUC-Sleep: a comprehensive public dataset for sleep researchers. Comput Methods Programs Biomed. 2016;124:180–92.

    Google Scholar 

  75. Khalighi S, Sousa T, Pires G, Nunes U. Automatic sleep staging: a computer assisted approach for optimal combination of features and polysomnographic channels. Expert Syst Appl. 2013;40(17):7046–59.

    Google Scholar 

  76. Tzimourta KD, Tsilimbaris AK, Tzioukalia AT, Tzallas MG, Tsipouras LG. EEG-based automatic sleep stage classification. Biomed J Sci Tech Res. 2018;7(4):6032–6.

    Google Scholar 

  77. Najdi S, Gharbali AA, Fonseca JM. Feature Transformation based on stacked sparse auto encoders for sleep stage classification’. In: Technological innovation for smart systems, vol 499; 2017. p. 191–200

  78. Kalbkhani H, Ghasemzadeh PG, Shayesteh M. Sleep stages classification from EEG signal based on stockwell transform. IET Signal Process. 2018;13:242–52.

    Google Scholar 

  79. Hassan AR, Subasi A. A decision support system for automated identification of sleep stages from single-channel EEG signals. Knowl Based Syst. 2017;128:115–24. https://doi.org/10.1016/j.knosys.2017.05.005.

    Article  Google Scholar 

  80. Tripathy RK, Rajendra AU. Use of features from RR-time series and EEG signals for automated classification of sleep stages in deep neural network framework. Biocybern Biomed Eng. 2018. https://doi.org/10.1016/j.bbe.2018.05.005.

    Article  Google Scholar 

Download references

Funding

This study is not funded by any organization.

Author information

Authors and Affiliations

  1. Department of Computer Science and Engineering, Pondicherry Engineering College, Puducherry, India

    Santosh Kumar Satapathy & D. Loganathan

Authors
  1. Santosh Kumar Satapathy
    View author publications

    You can also search for this author inPubMed Google Scholar

  2. D. Loganathan
    View author publications

    You can also search for this author inPubMed Google Scholar

Corresponding author

Correspondence to Santosh Kumar Satapathy.

Ethics declarations

Conflict of interest

All the authors in the paper have no conflict of interest.

Ethical approval

All procedures performed in studies involving human participants were in accordance with the ethical standards of the institutional and/or national research committee and with the 1964 Helsinki declaration and its later amendments or comparable ethical standards.

Additional information

Publisher's Note

Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.

This article is part of the topical collection “Data Science and Communication” guest-edited by Kamesh Namudri, Naveen Chilamkurti, Sushma S J and S. Padmashree.

Rights and permissions

Reprints and permissions

About this article

Check for updates. Verify currency and authenticity via CrossMark

Cite this article

Satapathy, S.K., Loganathan, D. A Study of Human Sleep Stage Classification Based on Dual Channels of EEG Signal Using Machine Learning Techniques. SN COMPUT. SCI. 2, 157 (2021). https://doi.org/10.1007/s42979-021-00528-5

Download citation

  • Received:

  • Accepted:

  • Published:

  • DOI: https://doi.org/10.1007/s42979-021-00528-5

Keywords