A single-channel EEG based automatic sleep stage classification method leveraging deep one-dimensional convolutional neural network and hidden Markov model

Abstract

Sleep stage classification is an essential process for analyzing sleep and diagnosing sleep related disorders. Sleep staging by visual inspection of expert is a labor-intensive task and prone to subjective errors. In this paper, we proposed a single-channel EEG based automatic sleep stage classification model, called 1D-CNN-HMM. Our 1D-CNN-HMM combines deep one-dimensional convolutional neural network (1D-CNN) and hidden Markov model (HMM). We leveraged 1D-CNN for epoch-wise classification and HMM for subject-wise classification. The main idea of 1D-CNN-HMM model is to utilize the advantage of 1D-CNN that can automatically extract features from raw EEG, and HMM that can utilize sleep stage transition prior information of adjacent EEG epochs. To the best of author's knowledge, this is the first implementation of 1D-CNN connected with HMM in automatic sleep staging task. We used Sleep-EDFx dataset and DRM-SUB dataset, and performed subject-independent testing for model evaluation. Experimental results illustrated the overall accuracy and kappa coefficient of 1D-CNN-HMM could achieve 83.98% and 0.78 on Fpz-Oz channel EEG from Sleep-EDFx dataset, and achieve 81.68% and 0.74 on Cz-A1 channel EEG from DRM-SUB dataset. The overall accuracy and kappa coefficient of 1D-CNN-HMM outperformed other existing methods both on two datasets. In addition, the per-class performance of 1D-CNN-HMM is significantly higher than 1D-CNN on S1 and REM sleep stages with p<0.05. Our 1D-CNN-HMM outperformed other existing methods both on two datasets. Results also indicated that HMM improved the classification performance of 1D-CNN by improving the performance on S1 and REM stages.