A Single Channel-Based Neonatal Sleep-Wake Classification using Hjorth Parameters and Improved Gradient Boosting

Muhammad Arslan,Muhammad Mubeen,Saadullah Farooq Abbasi,Muhammad Shahbaz Khan,Wadii Boulila,Jawad Ahmad
2024-08-15
Abstract:Sleep plays a crucial role in neonatal development. Monitoring the sleep patterns in neonates in a Neonatal Intensive Care Unit (NICU) is imperative for understanding the maturation process. While polysomnography (PSG) is considered the best practice for sleep classification, its expense and reliance on human annotation pose challenges. Existing research often relies on multichannel EEG signals; however, concerns arise regarding the vulnerability of neonates and the potential impact on their sleep quality. This paper introduces a novel approach to neonatal sleep stage classification using a single-channel gradient boosting algorithm with Hjorth features. The gradient boosting parameters are fine-tuned using random search cross-validation (randomsearchCV), achieving an accuracy of 82.35% for neonatal sleep-wake classification. Validation is conducted through 5-fold cross-validation. The proposed algorithm not only enhances existing neonatal sleep algorithms but also opens avenues for broader applications.
Machine Learning,Signal Processing
What problem does this paper attempt to address?
The problem that this paper attempts to solve is the classification of neonatal sleep - wakefulness using single - channel electroencephalogram (EEG) signals in the neonatal intensive care unit (NICU). Specifically, the paper aims to develop a method based on single - channel EEG signals and an improved gradient boosting algorithm to reduce the potential negative impacts of multi - channel EEG acquisition on neonates and improve the accuracy of classification. ### Background and Motivation - **Importance**: Sleep is crucial for the development of neonates, and monitoring the sleep patterns of neonates helps to understand their maturation process. - **Limitations of Existing Methods**: - **Polysomnography (PSG)**: Although considered the best practice, it is costly and relies on manual annotation, presenting time and economic challenges. - **Multi - channel EEG**: Although it can provide rich information, multi - channel acquisition may affect the sleep quality of neonates and the equipment is complex. ### Solution - **Method**: The paper proposes a new method based on single - channel EEG signals and an improved gradient boosting algorithm. - **Feature Extraction**: Thirteen significant time - domain and frequency - domain features, including Hjorth parameters (activity, mobility, and complexity), are extracted from the single - channel EEG signals. - **Model Optimization**: Random search cross - validation (randomsearchCV) is used to select the optimal parameters of the gradient boosting algorithm. - **Validation Method**: The overall data set is validated by 5 - fold cross - validation. ### Main Contributions - **Accuracy**: This method achieves a classification accuracy of 82.35% on single - channel EEG signals, with a Kappa value of 0.64. - **Low Computational Cost**: This model has a low computational cost and is suitable for real - time applications. - **Comparative Analysis**: Compared with existing machine learning and deep learning algorithms, this method performs well in single - channel classification tasks. ### Formulas - **Hjorth Parameters**: - **Activity**: \[ \text{Activity} = V(x(t)) \] where \( x(t) \) represents the signal and \( V \) represents the variance. - **Mobility**: \[ \text{Mobility} = \sqrt{\frac{V\left(\frac{dx(t)}{dt}\right)}{V(x(t))}} \] - **Complexity**: \[ \text{Complexity} = \frac{\sigma''}{\sigma'} \cdot \frac{\sigma'}{\sigma} \] - **Gradient Boosting**: - **Objective Function**: \[ F(x) = F_{\text{prev}}(x) + \sum_{m = 1}^{M} B_m h_m(x; \theta_m) \] - **Iterative Update**: \[ F_m(x) = F_{m - 1}(x) + B_m h_m(x; \theta_m) \] - **Minimize Loss Function**: \[ \min_{\beta_m, \theta_m} \sum_{i = 1}^{N} \psi(y_i, F_{m - 1}(x_i) + B_m h_m(x; \theta_m)) \] - **Performance Evaluation Metrics**: - **Sensitivity (Sensitivity)**: \[ \text{Se} = \frac{TP}{TP + FN} \times 100\% \] - **Specificity (Specificity)**: \[ \text{Sp} = \frac{TN}{TN + FP} \times 100\% \] - **Accuracy**