文献类型：期刊文献

英文题名：EEG data augmentation: towards class imbalance problem in sleep staging tasks

作者：Fan, Jiahao[1,2];Sun, Chenglu[1];Chen, Chen[1];Jiang, Xinyu[1];Liu, Xiangyu[3];Zhao, Xian[1];Meng, Long[1];Dai, Chenyun[1];Chen, Wei[1,2]

机构：[1]Fudan Univ, Sch Informat Sci & Technol, Dept Elect Engn, Ctr Intelligent Med Elect, Shanghai 200433, Peoples R China;[2]Fudan Univ, Human Phenome Inst, 825 Zhangheng Rd, Shanghai 201203, Peoples R China;[3]East China Univ Sci & Technol, Sch Art Design & Media, Shanghai 200237, Peoples R China

年份：2020

卷号：17

期号：5

外文期刊名：JOURNAL OF NEURAL ENGINEERING

收录：;EI(收录号：20204509444320);WOS:【SCI-EXPANDED(收录号:WOS:000577213900001)】；

基金：This research was supported by National Key R&D Program of China (Grant No. 2017YFE0112000), Shanghai Municipal Science and Technology Major Project (Grant No. 2017SHZDZX01), and China Postdoctoral Science Foundation Grant (Grant Nos. 2018T110346 and 2018M632019).

语种：英文

外文关键词：sleep stage classification; class imbalance problem; data augmentation; generative adversarial network

摘要：Objective.Automatic sleep staging models suffer from an inherent class imbalance problem (CIP), which hinders the classifiers from achieving a better performance. To address this issue, we systematically studied sleep electroencephalogram data augmentation (DA) approaches. Furthermore, we modified and transferred novel DA approaches from related research fields, yielding new efficient ways to enhance sleep datasets.Approach.This study covers five DA methods, including repeating minority classes, morphological change, signal segmentation and recombination, dataset-to-dataset transfer, as well as generative adversarial network (GAN). We evaluated these mentioned DA methods by a sleep staging model on two datasets, the Montreal archive of sleep studies (MASS) and Sleep-EDF. We used a classification model with a typical convolutional neural network architecture to evaluate the effectiveness of the mentioned DA approaches. We also conducted a comprehensive analysis of these methods.Main results.The classification results showed that DA methods, especially DA by GAN, significantly improved the total classification performance in comparison with the baseline. The improvement of accuracy, F1 score and Cohen Kappa coefficient range from 0.90% to 3.79%, 0.73% to 3.48%, 2.61% to 5.43% on MASS and 1.36% to 4.79%, 1.47% to 4.23%, 2.22% to 4.04% on Sleep-EDF, respectively. DA methods improved the classification performance in most cases, whereas the performance of class N1 showed a subtle degradation in the F1 scores.Significance.Overall, our study proved that DA approaches are efficient in alleviating CIP lying in sleep staging tasks. Meanwhile, this study provided avenues for further improving the sleep staging accuracy using DA methods.