Ensemble SVM Method for Automatic Sleep Stage Classification

Emina Alickovic, Abdulhamit Subasi
2018 IEEE Transactions on Instrumentation and Measurement  
Sleep scoring is used as a diagnostic technique in the diagnosis and treatment of sleep disorders. Automated sleep scoring is crucial, since the large volume of data should be analyzed visually by the sleep specialists which is burdensome, time-consuming tedious, subjective, and error-prone. Therefore, automated sleep stage classification is crucial step in sleep research and sleep disorder diagnosis. In the present article, a robust system, consisting of three modules, is proposed for
more » ... classification of sleep stages from single channel EEG. In the first module, signals taken from Pz-Oz electrode were denoised using multiscale principal component analysis. In the second module, the most informative features are extracted using discrete wavelet transform (DWT) and then, statistical values of DWT sub-bands are calculated. In the third module, extracted features were fed into an ensemble classifier, which can be called as rotational support vector machine (RotSVM). The proposed classifier combines advantages of the principal component analysis and SVM to improve classification performances of the traditional SVM. The sensitivity and accuracy values across all subjects were 84.46% and 91.1% respectively for five stage sleep classification with Cohen's kappa coefficient of 0.88. Obtained classification performance results indicate that, it is possible to have an efficient sleep monitoring system with a single channel EEG, and can be used effectively in medical and home-care application. Index Terms-Sleep stage classification, single-channel EEG, multiscale principal component analysis (MSPCA), discrete wavelet transform (DWT), rotational support vector machine (RotSVM).
doi:10.1109/tim.2018.2799059 fatcat:p4os5pi3tfehpldvwk2h6mhxnu