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 editorsZhou, Dongdong; Xu, Qi; Wang, Jian; Xu, Hongming; Kettunen, Lauri; Chang, Zheng; Cong, Fengyu

Journal or seriesIEEE Transactions on Instrumentation and Measurement

ISSN0018-9456

eISSN1557-9662

Publication year2022

Publication date18/07/2022

Volume71

Article number4006612

PublisherInstitute of Electrical and Electronics Engineers (IEEE)

Publication countryUnited States

Publication languageEnglish

DOIhttps://doi.org/10.1109/TIM.2022.3191710

Publication open accessNot open

Publication channel open access

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


Abstract

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.


Keywordssleep researchsleepEEGsignal analysismachine learningdeep learningneural networks (information technology)

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


Contributing organizations


Ministry reportingYes

Reporting Year2022

JUFO rating2


Last updated on 2024-03-04 at 18:56