Abstract
Sleep is important part for human health and quality of life in the daily routine basis. However, numerous individuals face sleep problems due to rapid changes occurred in both social and professional lifestyles. These problems can lead to several neurological and physical disorder diseases, and therefore, decrease their overall life quality. Machine learning methods for automated sleep stage classification (ASSC) are a fundamental approach to evaluate and treat this public health challenge. The main objective of this study is to propose a high-effective and high-accuracy based multiple sleep staging classification model based on single-channel electroencephalogram (EEG) signals using machine learning (ML) model. The proposed automated sleep staging system followed four basic stages: signal preprocessing, feature extraction and screening, classification algorithms, and performance evaluation. In this research work, a novel method is applied for signal preprocessing, feature screening and classification models. In signal preprocessing we obtain the wiener filter techniques for removing the different types of artifacts from input sleep recordings. In feature extraction, we obtain a total of 28 features based on both time and frequency domain features and non-linear features. The relevant features are screened through ReliefF weight feature selection algorithm, and eliminating the redundant features using Pearson correlation coefficients. The important contribution of this research work is establishes two layers an ensembling learning stacking model for classifying the multiple sleep stages. Three different subgroups of ISRUC-Sleep (SG-I/SG-II/SG-III) subjects sleep recordings having different health condition obtained for our proposed experimental work. Comparing with the recent contributions on sleep staging performances using single-channel EEG signals, it has found that our proposed ensemble learning stacking model was reported excellent in sleep staging classification accuracy performance for five sleep stages classification task (CT-5). The overall classification accuracy reported as 99.34%, 90.8%, and 98.50% for SG-I, SG-II, and SG-III dataset, respectively.
Similar content being viewed by others
Data availability
All EEG files are available from the ISRUC-SLEEP database (https://sleeptight.isr.uc.pt/ISRUC_Sleep/).
References
Aboalayon K, Faezipour M, Almuhammadi W, Moslehpour S (2016) Sleep stage classification using EEG signal analysis: a comprehensive survey and new investigation. Entropy 18(9):272. https://doi.org/10.3390/e18090272
Acharya UR et al (2015) Nonlinear dynamics measures for automated EEG-based sleep stage detection. Eur Neurol 74(5–6):268–287
Ahmad W, Ahmad A, Lu C et al (2018) A novel hybrid decision support system for thyroid disease forecasting. Soft Comput 22:5377–5383. https://doi.org/10.1007/s00500-018-3045-9
Al Ghayab HR, Li Y, Siuly S et al (2019) Epileptic seizures detection in EEGs blending frequency domain with information gain technique. Soft Comput 23:227–239. https://doi.org/10.1007/s00500-018-3487-0
Alickovic E, Subasi A (2018) Ensemble SVM method for automatic sleep stage classification. IEEE Trans Instrum Meas 67(6):1258–1265. https://doi.org/10.1109/TIM.2018.2799059
Al-Janabi S, Mohammad M, Al-Sultan A (2019) A new method for prediction of air pollution based on intelligent computation. Soft Comput. https://doi.org/10.1007/s00500-019-04495-1
Bajaj V, Pachori RB (2013) Automatic classification of sleep stages based on the time-frequency image of EEG signals. Comput Methods Programs Biomed 112(3):320–328. https://doi.org/10.1016/j.cmpb.2013.07.006
Berry RB, Brooks R, Gamaldo CE, Hardsim SM, Lloyd RM, Marcus CL, Vaughn BV (2014) The AASM manual for the scoring of sleep and associated events: rules, terminology and technical specifications. American Academy of Sleep Medicine
Borowicz A (2017) Correction of eye-blink artifacts in EEG recordings using Wiener filtering. In: 2017 signal processing: algorithms, architectures, arrangements, and applications (SPA), Poznan, pp 172–177. https://doi.org/10.23919/SPA.2017.8166859
Braun ET, Kozakevicius ADJ, Da Silveira TLT, Rodrigues CR, Baratto G (2018) Sleep stages classification using spectral based statistical moments as features. Rev Inf Teórica Apl 25(1):11. https://doi.org/10.22456/2175-2745.74030
Carskadon MA, Dement WC (2017) Normal human sleep: an overview. In: Kryger M, Roth T, Dement WC (eds.), Principles and Practice of sleep medicine, 6th edn., Elsevier, Amsterdam, pp. 15–24. https://doi.org/10.1016/B978-0-323-24288-2.00002-7
Chung MH, Kuo TB, Hsu N, Chu H, Chou KR, Yang CC (2009) Sleep and autonomic nervous system changes—enhanced cardiac sympathetic modulations during sleep in permanent night shift nurses. Scand J Work Environ Health 35(3):180–187. https://doi.org/10.5271/sjweh.1324
Da Silveira TLT, Kozakevicius AJ, Rodrigues CR (2016) Single-channel EEG sleep stage classification based on a streamlined set of statistical features in wavelet domain. Med Biol Eng Comput 55(2):343–352. https://doi.org/10.1007/s11517-016-1519-4
Dhok S, Pimpalkhute V, Chandurkar A, Bhurane AA, Sharma M, Acharya UR (2020) Automated phase classification in cyclic alternating patterns in sleep stages using Wigner-Ville distribution based features. Comput Biol Med 119:103691. https://doi.org/10.1016/j.compbiomed.2020.103691
Diykh M et al (2016) EEG sleep stages classification based on time domain features and structural graph similarity. IEEE Trans Neural Syst Rehabil Eng 24(11):1159–1168. https://doi.org/10.1109/tnsre.2016.2552539
Gunnarsdottir KM, Gamaldo CE, Salas RME, Ewen JB, Allen RP, Sarma SV (2018) A novel sleep stage scoring system: combining expert-based rules with a decision tree classifier. In: 2018 40th annual international conference of the IEEE engineering in medicine and biology society (EMBC). https://doi.org/10.1109/embc.2018.8513039
Gupta A, Agrawal RK, Kaur B (2015) Performance enhancement of mental task classification using EEG signal: a study of multivariate feature selection methods. Soft Comput 19:2799–2812. https://doi.org/10.1007/s00500-014-1443-1
Hassan AR, Bhuiyan MIH (2016a) Computer-aided sleep staging using complete ensemble empirical mode decomposition with adaptive noise and bootstrap aggregating. Biomed Signal Process Control 24:1–10. https://doi.org/10.1016/j.bspc.2015.09.002
Hassan AR, Bhuiyan MI (2016b) A decision support system for automatic sleep staging from EEG signals using tunable Q-factor wavelet transform and spectral features. J Neurosci Methods 15(271):107–118. https://doi.org/10.1016/j.jneumeth.2016.07.012
Hassan AR, Subasi A (2017) A decision support system for automated identification of sleep stages from single-channel EEG signals. Knowl Based Syst 128:115–124. https://doi.org/10.1016/j.knosys.2017.05.005
Hassan AR, Bashar SK, Bhuiyan MIH (2015) Automatic classification of sleep stages from single-channel electroencephalogram. In: 2015 annual IEEE India conference (INDICON)
He G, Zhao W, Xia X et al (2019) An ensemble of shapelet-based classifiers on inter-class and intra-class imbalanced multivariate time series at the early stage. Soft Comput 23:6097–6114. https://doi.org/10.1007/s00500-018-3261-3
Heyat MBB, Lai D, Khan FI, Zhang Y (2019) Sleep bruxism detection using decision tree method by the combination of C4–P4 and C4–A1 channels of scalp EEG. IEEE Access 7:102542–102553. https://doi.org/10.1109/ACCESS.2019.2928020
Holland JV, Dement WC, Raynal DM (1974) Polysomnography: a response to a need for improved communication. In: Presented at the 14th annual meeting of the association of psychophysiological study of sleep
Hsu YL, Yang Y-TC, Wang J-S, Hsu CY (2013) Automatic sleep stage recurrent neural classifier using energy features of EEG signals. Neurocomputing 104:105–114. https://doi.org/10.1016/j.neucom.2012.11.003
Huang W, Guo B, Shen Y, Tang X, Zhang T, Li D, Jiang Z (2019) Sleep staging algorithm based on multichannel data adding and multifeature screening. Comput Methods Programs Biomed. https://doi.org/10.1016/j.cmpb.2019.105253
Iber C (2007) The AASM manual for the scoring of sleep and associated events: rules, terminology and technical specifications: American Academy of Sleep Medicine
Kalbkhani H, Ghasemzadeh P, Shayesteh MG (2018) Sleep stages classification from EEG signal based on Stockwell transform. In: IET Signal Processing 2018. https://ukdiss.com/examples/sleep-stages-classification-eeg-signal.php?vref=1
Khalighi S, Sousa T, Oliveira D, Pires G, Nunes U (2011) Efficient feature selection for sleep staging based on maximal overlap discrete wavelet transform and SVM. Annu Int Conf IEEE Eng Med Biol Soc 2011:3306–3309. https://doi.org/10.1109/IEMBS.2011.6090897
Khalighi S, Sousa T, Pires G, Nunes U (2013) Automatic sleep staging: a computer assisted approach for optimal combination of features and polysomnographic channels. Expert Syst Appl 40(17):7046–7059. https://doi.org/10.1016/j.eswa.2013.06.023
Khalighi S, Sousa T, Santos JM, Nunes U (2016) ISRUC-Sleep: a comprehensive public dataset for sleep researchers. Comput Methods Programs Biomed 124:180–192. https://doi.org/10.1016/j.cmpb.2015.10.013
Kim T, Kim J, Lee K (2018) Detection of sleep disordered breathing severity using acoustic biomarker and machine learning techniques. BioMed Eng OnLine 17:16. https://doi.org/10.1186/s12938-018-0448-x
Lai D, Heyat MBB, Khan FI, Zhang Y (2019) Prognosis of sleep bruxism using power spectral density approach applied on EEG signal of both EMG1–EMG2 and ECG1–ECG2 channels. IEEE Access 7:82553–82562. https://doi.org/10.1109/ACCESS.2019.2924181
Memar P, Faradji F (2018) A novel multi-class EEG-based sleep stage classification system. IEEE Trans Neural Syst Rehabil Eng 26(1):84–95. https://doi.org/10.1109/TNSRE.2017.2776149
Mendez MO, Chouvarda I, Alba A, Bianchi AM, Grassi A, Arce-Santana E, Milioli G, Terzano MG, Parrino L (2016) Analysis of A-phase transitions during the cyclic alternating pattern under normal sleep. Med Biol Eng Comput 54(1):133–148. https://doi.org/10.1007/s11517-015-1349-9
Nagabushanam P, Thomas George S, Radha S (2019) EEG signal classification using LSTM and improved neural network algorithms. Soft Comput. https://doi.org/10.1007/s00500-019-04515-0
Najdi S, Gharbali AA, Fonseca JM (2017) Feature transformation based on stacked sparse autoencoders for sleep stage classification. In: Technological innovation for smart systems 2017, pp 191–200. https://doi.org/10.1007/978-3-319-56077-9_18
O’Reilly C, Nielsen T (2014) Assessing EEG sleep spindle propagation. Part 1: theory and proposed methodology. J Neurosci Methods 221:202–214. https://doi.org/10.1016/j.jneumeth.2013.08.013
Obayya M, Abou-Chadi F (2014) Automatic classification of sleep stages using EEG records based on Fuzzy c-means (FCM) algorithm. In: Radio science conference (NRSC), 2014 31st National, pp 265–272. https://doi.org/10.1109/NRSC.2014.6835085.
Rahman T, Farook O, Heyat BB, Siddiqui MM (2016) An overview of narcolepsy. IARJSET 3:85–87
Raza H, Cecotti H, Li Y et al (2016) Adaptive learning with covariate shift-detection for motor imagery-based brain–computer interface. Soft Comput 20:3085–3096. https://doi.org/10.1007/s00500-015-1937-5
Rechtschaffen A (1968) A manual for standardized terminology techniques and scoring system for sleep stages in human subjects, Brain Inf. Serv
Robnik-Šikonja M, Kononenko I (2003) Theoretical and empirical analysis of ReliefF and ReliefF. Mach Learn 53:23–69. https://doi.org/10.1023/A:1025667309714
Ronzhina M, Janoušek O, Kolářová J, Nováková M, Honzík P, Provazník I (2012) Sleep scoring using artificial neural networks. Sleep Med Rev 16(3):251–263. https://doi.org/10.1016/j.smrv.2011.06.003
Santaji S, Desai V (2020) Analysis of EEG signal to classify sleep stages using machine learning. Sleep Vigil. https://doi.org/10.1007/s41782-020-00101-9
Schetinin V, Schult J (2006) Learning polynomial networks for classification of clinical electroencephalograms. Soft Comput 10:397–403. https://doi.org/10.1007/s00500-005-0499-3
Şen B, Peker M, Çavuşoğlu A et al (2014) A comparative study on classification of sleep stage based on EEG signals using feature selection and classification algorithms. J Med Syst 38:18. https://doi.org/10.1007/s10916-014-0018-0
Sharma R, Pachori RB, Upadhyay A (2017) Automatic sleep stages classification based on iterative filtering of electroencephalogram signals. Neural Comput Appl 28:2959–2978. https://doi.org/10.1007/s00521-017-2919-6
Sharma M, Tan R-S, Acharya UR (2019a) Automated heartbeat classification and detection of arrhythmia using optimal orthogonal wavelet filters. Inf Med Unlocked. https://doi.org/10.1016/j.imu.2019.100221
Sharma M, Patel S, Choudhary S, Acharya UR (2019b) Automated detection of sleep stages using energy-localized orthogonal wavelet filter banks. Arab J Sci Eng. https://doi.org/10.1007/s13369-019-04197-8
Siddiqui MM, Srivastava G, Saeed SH (2016) Diagnosis of insomnia sleep disorder using short time frequency analysis of PSD approach applied on EEG signal using channel ROC-LOC. Sleep Sci 9(3):186–191. https://doi.org/10.1016/j.slsci.2016.07.002
Simões H, Pires G, Nunes U, Silva V (2010) 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–133. https://doi.org/10.5220/0002950601280133
Souri A, Ghafour MY, Ahmed AM et al (2020) A new machine learning-based healthcare monitoring model for student’s condition diagnosis in Internet of Things environment. Soft Comput 24:17111–17121. https://doi.org/10.1007/s00500-020-05003-6
Sousa T, Cruz A, Khalighi S, Pires G, Nunes U (2015) A two-step automatic sleep stage classification method with dubious range detection. Comput Biol Med 59:42–53. https://doi.org/10.1016/j.compbiomed.2015.01.017
Spriggs WH (2014) Essentials of Polysomnography. World Headquarter Jones & Bartlett Publishers, MA, United States
Sriraam N, Padma Shri TK, Maheshwari U (2016) Recognition of wake-sleep stage 1 multichannel EEG patterns using spectral entropy features for drowsiness detection. Australas Phys Eng Sci Med 39(3):797–806. https://doi.org/10.1007/s13246-016-0472-8
The Kappa statistic in reliability studies: use, interpretation, and sample size requirements (2005) Physical Therapy. https://doi.org/10.1093/ptj/85.3.257
Tian P, Hu J, Qi J, Ye X, Che D, Ding Y, Peng Y (2017) A hierarchical classification method for automatic sleep scoring using multiscale entropy features and proportion information of sleep architecture. Biocybern Biomed Eng 37(2):263–271. https://doi.org/10.1016/j.bbe.2017.01.005
Tzimourta KD, Tsimbaris A, Tzioukalia K, Tzallas AT, Tsipouras MG, Asktrakas LG, Giannakeas N (2018) EEG based automatic sleep stage classifcation. Biomed J Sci Tech Res. https://doi.org/10.26717/BJSTR.2018.07.001535
Wang Q, Zhao D, Wang Y, Hou X (2019) Ensemble learning algorithm based on multi-parameters for sleep staging. Med Biol Eng Comput. https://doi.org/10.1007/s11517-019-01978-z
Yasoda K, Ponmagal RS, Bhuvaneshwari KS et al (2020) Automatic detection and classification of EEG artifacts using fuzzy kernel SVM and wavelet ICA (WICA). Soft Comput 24:16011–16019. https://doi.org/10.1007/s00500-020-04920-w
Zhang B, Lei T, Liu H, Cai H (2018) EEG-based automatic sleep staging using ontology and weighting feature analysis. Comput Math Methods Med 2018:1–16. https://doi.org/10.1155/2018/6534041
Zhu G, Li Y, Wen P (2014) Analysis and classification of sleep stages based on difference visibility graphs from a single-channel EEG signal. IEEE J Biomed Health Inform 18(6):1813–1821. https://doi.org/10.1109/JBHI.2014.2303991
Zibrandtsen I, Kidmose P, Otto M, Ibsen J, Kjaer TW (2016) Case comparison of sleep features from ear-EEG and scalp-EEG. Sleep Sci 9(2):69–72
Funding
This study is not funded by any organization.
Author information
Authors and Affiliations
Corresponding author
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.
Rights and permissions
About this article
Cite this article
Satapathy, S.K., Loganathan, D. Prognosis of automated sleep staging based on two-layer ensemble learning stacking model using single-channel EEG signal. Soft Comput 25, 15445–15462 (2021). https://doi.org/10.1007/s00500-021-06218-x
Accepted:
Published:
Issue Date:
DOI: https://doi.org/10.1007/s00500-021-06218-x