Machine learning with ensemble stacking model for automated sleep staging using dual-channel EEG signal

Abstract

Sleep staging is an important part of diagnosing the different types of sleep-related disorders because any discrepancies in the sleep scoring process may cause serious health problems such as misinterpretations of sleep patterns, medication errors, and improper diagnosis. The best way of analyzing sleep staging is visual interpretations of the polysomnography (PSG) signals recordings from the patients, which is a quite tedious task, requires more domain experts, and time-consuming process. This proposed study aims to develop a new automated sleep staging system using the brain EEG signals. Based on a new automated sleep staging system based on an ensemble learning stacking model that integrates Random Forest (RF) and eXtreme Gradient Boosting (XGBoosting). Additionally, this proposed methodology considers the subjects’ age, which helps analyze the S1 sleep stage properly. In this study, both linear (time and frequency) and non-linear features are extracted from the pre-processed signals. The most relevant features are selected using the ReliefF weight algorithm. Finally, the selected features are classified through the proposed two-layer stacking model. The proposed methodology performance is evaluated using the two most popular datasets, such as the Sleep-EDF dataset (S-EDF) and Sleep Expanded-EDF database (SE-EDF) under the Rechtschaffen & Kales (R&K) sleep scoring rules. The performance of the proposed method is also compared with the existing published sleep staging methods. The comparison results signify that the proposed sleep staging system has an excellent improvement in classification accuracy for the six-two sleep states classification. In the S-EDF dataset, the overall accuracy and Cohen’s kappa coefficient score obtained by the proposed model is (91.10%, 0.87) and (90.68%, 0.86) with inclusion and exclusion of age feature using the Fpz-Cz channel, respectively. Similarly, the Pz-Oz channel’s performance is (90.56%, 0.86) with age feature and (90.11%, 0.86) without age feature. The performed results with the SE-EDF dataset using Fpz-Cz channel is (81.32%, 0.77) and (81.06%, 0.76), using Pz-Oz channel with the inclusion and exclusion of the age feature, respectively. Similarly the model achieved an overall accuracy of 96.67% (CT-6), 96.60% (CT-5), 96.28% (CT-4),96.30% (CT-3) and 97.30% (CT-2) for with 16 selected features using S-EDF database. Similarly the model reported an overall accuracy of 85.85%, 84.98%, 85.51%, 85.37% and 87.40% for CT-6 to CT-2 with 18 selected features using SE-EDF database.