Feature Extraction From Single-Channel EEG Using Tsfresh and Stacked Ensemble Approach for Sleep Stage Classification

Abstract

The smart world under Industry 4.0 is witnessing a notable spurt in sleep disorders and sleep-related issues in patients. Artificial intelligence and IoT are taking a giant leap in connecting sleep patients remotely with healthcare providers. The contemporary single-channel-based monitoring devices play a tremendous role in predicting sleep quality and related issues. Handcrafted feature extraction is a time-consuming job in machine learning-based automatic sleep classification. The proposed single-channel work uses Tsfresh to extract features from both the EEG channels (Pz-oz and Fpz-Cz) of the SEDFEx database individually to realise a single-channel EEG. The adopted mRMR feature selection approach selected 55 features from the extracted 787 features. A stacking ensemble classifier achieved 95%, 94%, 91%, and 88% accuracy using stratified 5-fold validation in 2, 3, 4, and 5 class classification employing healthy subjects data. The outcome of the experiments indicates that Tsfresh is an excellent tool to extract standard features from EEG signals.