A1 Journal article (refereed)
Alleviating Class Imbalance Problem in Automatic Sleep Stage Classification (2022)

Zhou, D., Xu, Q., Wang, J., Xu, H., Kettunen, L., Chang, Z., & Cong, F. (2022). Alleviating Class Imbalance Problem in Automatic Sleep Stage Classification. IEEE Transactions on Instrumentation and Measurement, 71, Article 4006612. https://doi.org/10.1109/TIM.2022.3191710

JYU authors or editors

Publication details

All authors or editors: Zhou, Dongdong; Xu, Qi; Wang, Jian; Xu, Hongming; Kettunen, Lauri; Chang, Zheng; Cong, Fengyu

Journal or series: IEEE Transactions on Instrumentation and Measurement

ISSN: 0018-9456

eISSN: 1557-9662

Publication year: 2022

Publication date: 18/07/2022

Volume: 71

Article number: 4006612

Publisher: Institute of Electrical and Electronics Engineers (IEEE)

Publication country: United States

Publication language: English

DOI: https://doi.org/10.1109/TIM.2022.3191710

Publication open access: Not open

Publication channel open access:

Publication is parallel published (JYX): https://jyx.jyu.fi/handle/123456789/85541


For real-world automatic sleep-stage classification tasks, various existing deep learning-based models are biased toward the majority with a high proportion. Because of the unique sleep structure, most of the current polysomnography (PSG) datasets suffer an inherent class imbalance problem (CIP), in which the number of each sleep stage is severely unequal. In this study, we first define the class imbalance factor (CIF) to describe the level of CIP quantitatively. Afterward, we propose two balancing methods to alleviate this problem from the dataset quantity and the relationship between the class distribution and the applied model, respectively. The first one is to employ the data augmentation (DA) with the generative adversarial network (GAN) model and different intensities of Gaussian white noise (GWN) to balance samples, thereinto, GWN addition is specifically tailored to deep learning-based models, which can work on raw electroencephalogram (EEG) data while preserving their properties. In addition, we try to balance the relationship between the imbalanced class and biased network model to achieve a balanced state with the help of class distribution and neuroscience principles. We further propose an effective deep convolutional neural network (CNN) model utilizing bidirectional long short-term memory (Bi-LSTM) with single-channel EEG as the baseline. It is used for evaluating the efficiency of two balancing approaches on three imbalanced PSG datasets (CCSHS, Sleep-EDF, and Sleep-EDF-V1). The qualitative and quantitative evaluation of experimental results demonstrates that the proposed methods could not only show the superiority of class balancing through the confusion matrix and classwise metrics, but also get better N1 stage and whole stages classification accuracies compared to other state-of-the-art approaches.

Keywords: sleep research; sleep; EEG; signal analysis; machine learning; deep learning; neural networks (information technology)

Free keywords: Class imbalance problem (CIP); data augmentation (DA); deep neural network; generative adversarial network (GAN); network connection; sleep-stage classification

Contributing organizations

Ministry reporting: Yes

Reporting Year: 2022

Preliminary JUFO rating: 2

Last updated on 2023-06-03 at 09:55