A Single-Channel EEG Signal Based Sleep Staging with Global-Context-Modeling Feature Fusion

Yongjie Feng,Xiaoyu Zou,Jie‐Nan Pan
DOI: https://doi.org/10.1109/CAC59555.2023.10450427
2023-11-17
Abstract:Sleep staging is important for the detection and treatment of sleep disorders. For sleep specialists, manual sleep stage scoring is a time-consuming task, which is also limited by inter-rater reliability. Therefore, it is important to implement automatic sleep staging. However, existing methods still face the challenges of poor feature quality and unbalanced classes. In this paper, we design a single-channel EEG signal based sleep staging method with more integrated feature fusion that employs global context modeling. To improve the feature quality, the proposed model utilizes global context network (GCNet) to implement feature fusion. The module improves feature quality by modeling input features globally through a self-attentive mechanism, which allows the model to capture long-term dependencies between features. The presence of the class imbalance problem makes the model perform poorly in minority classes. Therefore, the proposed model increases the focus of the model on minority classes of samples through gradient harmonizing mechanism (GHM), which improves the overall performance of the model. The proposed model achieves accuracy rates of 84.6% and 81.5% on two public data sets of Sleep-EDF-20 and Sleep-EDF -78, respectively, which is better than the state-of-the-art.
Computer Science,Medicine
What problem does this paper attempt to address?