Denoise Enhanced Neural Network with Efficient Data Generation for Automatic Sleep Stage Classification of Class Imbalance.

Le Yu,Peiwang Tang, Zhiguo Jiang,Xianchao Zhang

IJCNN(2023)

引用 0|浏览3
暂无评分
摘要
Signal-channel electroencephalogram (EEG) based automatic sleep stage classification with machine learning is widely used in the study of sleep quality and analysis of sleep disorders. While due to the inevitable class imbalance problem, relatively poor accuracy in the detection of the first stage of the Non-Rapid-Eye-Movement (N1) is found. In this work, we propose to use a generative adversarial network (GAN) based on transformer encoder to reduce the class imbalance problem. In order to ensure the quality of the synthetic signal, the percentage form of the Kullback-Leibler divergence (KLP) index is designed to measure the similarity of synthetic signals generated by GAN and real ones. Meanwhile, we design a Residual Shrinkage Sequence Network for Sleep Staging (RsSleepNet) as the baseline to compare other resolutions of class imbalance with ours. The performance of the new method with the combination of GAN and RsSleepNet is effectively verified on two public datasets from PhysioNet, in which the accuracy of the N1 stage can be improved by more than 10% as compared to the current state-of-the-art approaches, largely alleviating the class imbalance problem in automatic sleep stage classification.
更多
查看译文
关键词
Sleep Staging,Class Imbalance,Data Generation
AI 理解论文
溯源树
样例
生成溯源树,研究论文发展脉络
Chat Paper
正在生成论文摘要