A single-channel EEG based automatic sleep stage classification method leveraging deep one-dimensional convolutional neural network and hidden Markov model
| Autor: | Hongxing Liu, Bufang Yang, Xilin Zhu, Yitian Liu |
|---|---|
| Rok vydání: | 2021 |
| Předmět: |
Sleep Stages
Channel (digital image) medicine.diagnostic_test Computer science business.industry 0206 medical engineering Biomedical Engineering Health Informatics Pattern recognition 02 engineering and technology Electroencephalography 020601 biomedical engineering Convolutional neural network Task (project management) 03 medical and health sciences 0302 clinical medicine Cohen's kappa Signal Processing medicine Artificial intelligence Sleep (system call) Hidden Markov model business 030217 neurology & neurosurgery |
| Zdroj: | Biomedical Signal Processing and Control. 68:102581 |
| ISSN: | 1746-8094 |
| DOI: | 10.1016/j.bspc.2021.102581 |
| Popis: | 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. |
| Databáze: | OpenAIRE |
| Externí odkaz: |
|
Full Text from ScienceDirect