SCL-SSC: Supervised Contrastive Learning for Sleep Stage Classification
preprintposted on 05.01.2022, 22:50 authored by Chandra Bhushan KumarChandra Bhushan Kumar
In this study, we have proposed SCL-SSC(Supervised Contrastive Learning for Sleep Stage Classification), a deep learning-based framework for sleep stage classification which performs the task in two stages, 1) feature representation learning, and 2) classification. The feature learner is trained separately to represent the raw EEG signals in the feature space such that the distance between the embedding of EEG signals of the same sleep stage has less than the distance between the embedding of EEG signals of different sleep stages in the euclidean space. On top of feature learners, we have trained the classifier to perform the classification task. The distribution of sleep stages is not uniform in the PSG data, wake(W) and N2 sleep stages appear more frequently than the other sleep stages, which leads to an imbalance dataset problem. This paper addresses this issue by using weighted softmax cross-entropy loss function and also dataset oversampling technique utilized to produce synthetic data points for minority sleep stages for approximately balancing the number of sleep stages in the training dataset. The performance of our proposed model is evaluated on the publicly available Physionet datasets EDF-Sleep 2013 and 2018 versions. We have trained and evaluated our model on two EEG channels (Fpz-Cz and Pz-Oz) on these datasets separately. The evaluation result shows that the performance of SCL-SSC is the best annotation performance compared to the existing state-of art deep learning algorithms to our best of knowledge, with an overall accuracy of 94.1071% with a macro F1 score of 92.6416 and Cohen’s Kappa coefficient(κ) 0.9197. Our ablation studies on SCL-SSC shows that both triplet loss based pre-training of feature learner and oversampling of minority classes are contributing to better performance of the model(SCL-SSC).