Machine learning-empowered sleep staging classification using multi-modality signals

Santosh Kumar Satapathy,Biswajit Brahma,Baidyanath Panda,Paolo Barsocchi,Akash Kumar Bhoi
DOI: https://doi.org/10.1186/s12911-024-02522-2
IF: 3.298
2024-05-08
BMC Medical Informatics and Decision Making
Abstract:The goal is to enhance an automated sleep staging system's performance by leveraging the diverse signals captured through multi-modal polysomnography recordings. Three modalities of PSG signals, namely electroencephalogram (EEG), electrooculogram (EOG), and electromyogram (EMG), were considered to obtain the optimal fusions of the PSG signals, where 63 features were extracted. These include frequency-based, time-based, statistical-based, entropy-based, and non-linear-based features. We adopted the ReliefF (ReF) feature selection algorithms to find the suitable parts for each signal and superposition of PSG signals. Twelve top features were selected while correlated with the extracted feature sets' sleep stages. The selected features were fed into the AdaBoost with Random Forest (ADB + RF) classifier to validate the chosen segments and classify the sleep stages. This study's experiments were investigated by obtaining two testing schemes: epoch-wise testing and subject-wise testing. The suggested research was conducted using three publicly available datasets: ISRUC-Sleep subgroup1 (ISRUC-SG1), sleep-EDF(S-EDF), Physio bank CAP sleep database (PB-CAPSDB), and S-EDF-78 respectively. This work demonstrated that the proposed fusion strategy overestimates the common individual usage of PSG signals.
medical informatics
What problem does this paper attempt to address?
The problem that this paper attempts to solve is to enhance the performance of the automatic sleep staging system by using multi - modal signals (i.e., electroencephalogram (EEG), electrooculogram (EOG) and electromyogram (EMG)). Specifically, the research objectives are: 1. **Improve the effect of multi - modal signal fusion**: Extract features from multi - modal polysomnography (PSG) signals and perform effective signal fusion to improve the performance of the classifier. 2. **Optimize feature selection**: Use the ReliefF algorithm to select the most relevant features, reduce the feature dimension, and improve the accuracy and efficiency of the classifier. 3. **Develop a high - precision sleep - stage classifier**: Adopt the method of combining AdaBoost and Random Forest to build a model that can accurately classify different sleep stages. 4. **Consider individual differences**: Introduce individual characteristics such as age to improve the personalized accuracy of sleep staging. 5. **Reduce the dependence on prior knowledge**: By introducing efficient adaptive signal analysis techniques, reduce the dependence on prior knowledge in the feature extraction stage. Through these methods, the paper aims to overcome the limitations in traditional sleep staging methods, such as inconsistent classification, poor model generalization ability, and high misclassification rates for specific sleep stages (such as N1 and REM), thereby improving the overall accuracy and reliability of sleep staging.