Optimization of Sleep Stage Classification using Single-Channel EEG Signals

Classification of various stages of sleep is mandatory for the diagnosis and treatment of sleep disorders. Manual scoring is a time-consuming and tedious task as well as it requires sleep specialists. Therefore, automatic sleep stage classification is necessary. In this paper, we have utilized state-of-the-art signal processing and machine learning techniques for sleep stage classification using single-channel EEG signal. Three cases of sleep classification have been done using support vector classifier (SVC), Decision tree (DT), Random forest (RF) and XGBoost (XGB). The features extracted from pre-procesed EEG have been applied to Spectral Regression dimensionality reduction technique to reduce the model complexity. The Bayesian Optimization (BO) technique is applied to optimize the hyper-parameters of the classifiers. Our proposed classification techniques provide the minimum error of 25.52%, 14.03%, and 4.93% for case I, case II and case III, respectively.

[1]  Long Chen,et al.  Short-Term Load Forecasting Using EMD-LSTM Neural Networks with a Xgboost Algorithm for Feature Importance Evaluation , 2017 .

[2]  G. Lightbody,et al.  Grading hypoxic–ischemic encephalopathy severity in neonatal EEG using GMM supervectors and the support vector machine , 2016, Clinical Neurophysiology.

[3]  Esther Rodríguez-Villegas,et al.  Recommendations for performance assessment of automatic sleep staging algorithms , 2014, 2014 36th Annual International Conference of the IEEE Engineering in Medicine and Biology Society.

[4]  B. Koley,et al.  An ensemble system for automatic sleep stage classification using single channel EEG signal , 2012, Comput. Biol. Medicine.

[5]  Jiawei Han,et al.  SRDA: An Efficient Algorithm for Large-Scale Discriminant Analysis , 2008, IEEE Transactions on Knowledge and Data Engineering.

[6]  Miad Faezipour,et al.  Sleep Stage Classification Using EEG Signal Analysis: A Comprehensive Survey and New Investigation , 2016, Entropy.

[7]  José Luis Rodríguez-Sotelo,et al.  Automatic Sleep Stages Classification Using EEG Entropy Features and Unsupervised Pattern Analysis Techniques , 2014, Entropy.

[8]  G. Clark,et al.  Prevalence and impact of sleep disorders and sleep habits in the United States , 2010, Sleep and Breathing.

[9]  Licheng Zhang,et al.  Machine Learning in Rock Facies Classification: An Application of XGBoost , 2017 .

[10]  Mohammed Imamul Hassan Bhuiyan,et al.  Automatic sleep scoring using statistical features in the EMD domain and ensemble methods , 2016 .

[11]  Yu Ma,et al.  Robust sleep stage classification with single-channel EEG signals using multimodal decomposition and HMM-based refinement , 2019, Expert Syst. Appl..

[12]  Jasper Snoek,et al.  Practical Bayesian Optimization of Machine Learning Algorithms , 2012, NIPS.

[13]  Geraldine B. Boylan,et al.  NEURAL: quantitative features for newborn EEG using Matlab , 2017, 1704.05694.

[14]  Aeilko H. Zwinderman,et al.  Analysis of a sleep-dependent neuronal feedback loop: the slow-wave microcontinuity of the EEG , 2000, IEEE Transactions on Biomedical Engineering.

[15]  Abdulhamit Subasi,et al.  A decision support system for automated identification of sleep stages from single-channel EEG signals , 2017, Knowl. Based Syst..

[16]  Abdullah Al Nahid,et al.  Effective Intrusion Detection System Using XGBoost , 2018, Inf..

[17]  Senén Barro,et al.  Do we need hundreds of classifiers to solve real world classification problems? , 2014, J. Mach. Learn. Res..

[18]  Mohammed Imamul Hassan Bhuiyan,et al.  Automatic sleep stage classification , 2015, 2015 2nd International Conference on Electrical Information and Communication Technologies (EICT).