Single-channel EEG sleep staging based on data augmentation and cross-subject discrepancy alleviation

Abstract

Automatic sleep stage classification is an effective technology compared to conventional artificial visual inspection in the field of sleep staging. Numerous algorithms based on machine learning and deep learning on single-channel electroencephalogram (EEG) have been proposed in recent years, however, category imbalance and cross-subject discrepancy are still the main factors restricting the accuracy of existing methods. This study proposed an innovative end-to-end neural network to solve these problems, specifically, four data augmentation methods were designed to eliminate category imbalance, and domain adaptation modules were designed for the alignment of marginal distribution, conditional distribution, and channel and spatial level distribution of feature maps, as well as the capture of transferable regions on the feature maps using a transfer attention mechanism. We conducted experiments on two publicly available datasets (Sleep-EDF Database Expanded, 2013 and 2018 version), Cohen's kappa coefficient (k) of 0.77 (Fpz-Cz) and 0.73 (Pz-Oz) were realized on the Sleep-EDF-2013 dataset, and a k of 0.75 (Fpz-Cz) and 0.68 (Pz-Oz) were realized on the Sleep-EDF-2018 dataset. An experiment was also conducted on the dataset drawn from the 2018 Physionet challenge, which containing people with sleep disorders, and a performance improvement was still found. Our comparative experiments with similar studies showed that our model was superior to most other studies, indicating our proposed EEG data augmentation and domain adaptation based cross-subject discrepancy alleviation approach is effective to improve the performance of automatic sleep staging.