Original Article

Automatic Sleep Stage Classification Using 1D Convolutional Neural Network

Abstract

Purpose: Sleep is one of the necessities of the body, such as eating, drinking, etc., that affects different aspects of human life. Sleep monitoring and sleep stage classification play an important role in the diagnosis of sleep-related diseases and neurological disorders. Empirically, classification of sleep stages is a time-consuming, tedious, and complex task, which heavily depends on the experience of the experts. As a result, there is a crucial need for an automatic efficient sleep staging system.
Materials and Methods: This study develops a 13-layer 1D Convolutional Neural Network (CNN) using single-channel Electroencephalogram (EEG) signal for extracting features automatically and classifying the sleep stages. To overcome the negative effect of an imbalance dataset, we have used the Synthetic Minority Oversampling Technique (SMOTE). In our study, the single-channel EEG signal is given to a 1D CNN, without any feature extraction/selection processes. This deep network can self-learn the discriminative features from the EEG signal.
Results: Applying the proposed method to sleep-EDF dataset resulted in overall accuracy, sensitivity, specificity, and Precision of 94.09%, 74.73%, 96.43%, and 71.02%, respectively, for classifying five sleep stages. Using single-channel EEG and providing a network with fewer trainable parameters than most of the available deep learning-based methods are the main advantages of the proposed method.
Conclusion: In this study, a 13-layer 1D CNN model was proposed for sleep stage classification. This model has an end-to-end complete architecture and does not require any separate feature extraction/selection and classification stages. Having a low number of network parameters and layers while still having high classification accuracy, is the main advantage of the proposed method over most of the previous deep learning-based approaches.

1- C. Stepnowsky, D. Levendowski, D. Popovic, I. A.-S. Medicine, and U. 2013‏, “Scoring accuracy of automated sleep staging from a bipolar electroocular recording compared to manual scoring by multiple raters‏,” Elsevier‏, Accessed: Aug. 06, 2020. [Online]. Available: https://www.sciencedirect.com/science/article/pii/S1389945713002347.
2- Y. Wang, K. Loparo, … M. K.-N. and science of, and undefined 2015‏, “Evaluation of an automated single-channel sleep staging algorithm‏,” ncbi.nlm.nih.gov‏, Accessed: Aug. 06, 2020. [Online]. Available: https://www.ncbi.nlm.nih.gov/pmc/articles/PMC4583116/.
3- D. Cogan, J. Birjandtalab, M. Nourani, J. Harvey, and V. Nagaraddi, “Multi-Biosignal Analysis for Epileptic Seizure Monitoring,” in International Journal of Neural Systems, Feb. 2017, vol. 27, no. 1, doi: 10.1142/S0129065716500313.
4- E. W.-A. of G. Psychiatry and undefined 1969‏, “A Manual of Standardized Terminology, Techniques and Scoring System for Sleep Stages of Human Subjects.‏,” jamanetwork.com‏, Accessed: Aug. 06, 2020. [Online]. Available: https://jamanetwork.com/journals/jamapsychiatry/article-abstract/489892.
5- H. Schulz, “Rethinking sleep analysis: comment on the AASM manual for the scoring of sleep and associated events‏,” 2008‏, Accessed: Aug. 06, 2020. [Online]. Available: https://jcsm.aasm.org/doi/abs/10.5664/jcsm.27124.
6- P. Tian et al., “A hierarchical classification method for automatic sleep scoring using multiscale entropy features and proportion information of sleep architecture‏,” Elsevier‏, Accessed: Aug. 06, 2020. [Online]. Available: https://www.sciencedirect.com/science/article/pii/S0208521616302455.
7- M. Carskadon and W. Dement, “Normal human sleep: An overview principles and practice of sleep medicine (pp. 13–24) 2005‏,” ‏.
8- “Stages of Sleep: The Sleep Cycle | American Sleep Association.” https://www.sleepassociation.org/about-sleep/stages-of-sleep/ (accessed Aug. 06, 2020).
9- S.-F. Liang, C.-E. Kuo, Y.-H. Hu, and Y.-S. Cheng, “A rule-based automatic sleep staging method Computational Neuroscience A rule-based automatic sleep staging method,” Artic. J. Neurosci. Methods, vol. 205, pp. 169–176, 2012, doi: 10.1016/j.jneumeth.2011.12.022.
10- S. Charbonnier, L. Zoubek, … S. L.-C. in B. and, and undefined 2011‏, “Self-evaluated automatic classifier as a decision-support tool for sleep/wake staging‏,” Elsevier‏, Accessed: Aug. 06, 2020. [Online]. Available: https://www.sciencedirect.com/science/article/pii/S0010482511000606.
11- O. Yildirim, U. Baran Baloglu, and U. Rajendra Acharya, “A Deep Learning Model for Automated Sleep Stages Classification Using PSG Signals,” Int. J. Environ. Res. Public Heal. Artic., 2019, doi: 10.3390/ijerph16040599.
12- G. Zhu, Y. Li, and P. Wen, “Analysis and Classification of Sleep Stages Based on Difference Visibility Graphs From a Single-Channel EEG Signal Ensemble of adaboost cascades of 3L-LBPs classifiers for license plates detection with low quality images View project Estimating the Effects of Carbon Dioxide, Temperature and Nitrogen on Grain Protein and Grain Yield Using Meta-Analysis View project Analysis and Classification of Sleep Stages Based on Difference Visibility Graphs From a Single-Channel EEG Signal,” Artic. IEEE J. Biomed. Heal. Informatics, vol. 00, no. 00, p. 1, 2014, doi: 10.1109/JBHI.2014.2303991.
13- M. Ronzhina, T. Potocnak, O. Janousek, J. Kolarova, M. Novakova, and I. Provaznik, “Spectral and Higher-Order Statistical Analysis of the ECG: Application to the Study of Ischemia in Rabbit Isolated Hearts.” Accessed: Aug. 06, 2020. [Online]. Available: https://ieeexplore.ieee.org/abstract/document/6420476/.
14- M. Ronzhina, O. Janoušek, J. Kolářová, M. N.-… medicine reviews, and undefined 2012‏, “Sleep scoring using artificial neural networks‏,” Elsevier‏, Accessed: Aug. 06, 2020. [Online]. Available: https://www.sciencedirect.com/science/article/pii/S1087079211000700.
15- B. Şen, M. Peker, A. Çavuşoğlu, and F. V Çelebi, “A Comparative Study on Classification of Sleep Stage Based on EEG Signals Using Feature Selection and Classification Algorithms,” Springer‏, vol. 38, no. 3, 2014, doi: 10.1007/s10916-014-0018-0.
16- A. R. Sereshkeh, R. Trott, A. Bricout, and T. Chau, “Online EEG Classification of Covert Speech for Brain-Computer Interfacing,” Int. J. Neural Syst., vol. 27, no. 8, Dec. 2017, doi: 10.1142/S0129065717500332.
17- I. Gath, C. Feuerstein, A. G.-P. recognition letters, and undefined 1994‏, “Unsupervised classification and adaptive definition of sleep patterns‏,” Elsevier‏, Accessed: Aug. 06, 2020. [Online]. Available: https://www.sciencedirect.com/science/article/pii/0167865594900299.
18- S. Güneş, K. Polat, Ş. Y.-E. S. with Applications, and undefined 2010‏, “Efficient sleep stage recognition system based on EEG signal using k-means clustering based feature weighting‏,” Elsevier‏, Accessed: Aug. 06, 2020. [Online]. Available: https://www.sciencedirect.com/science/article/pii/S095741741000343X.
19- A. Flexer, G. Gruber, G. D.-A. intelligence in Medicine, and undefined 2005‏, “A reliable probabilistic sleep stager based on a single EEG signal‏,” Elsevier‏, Accessed: Aug. 06, 2020. [Online]. Available: https://www.sciencedirect.com/science/article/pii/S093336570400079X.
20- L. Doroshenkov, V. Konyshev, S. S.-B. Engineering, and undefined 2007‏, “Classification of human sleep stages based on EEG processing using hidden Markov models‏,” Springer‏, Accessed: Aug. 06, 2020. [Online]. Available: https://link.springer.com/content/pdf/10.1007/s10527-007-0006-5.pdf.
21- M. Yoshimura, N. M.- Neurocomputing, and undefined 2006‏, ““Predicting sleep stages based on time-dependent hidden Markov model‏.”
22- S. Liang, C. Chen, … J. Z.-… C. on M., and undefined 2014‏, “Application of Genetic Algorithm and Fuzzy Vector Quantization on EEG-based automatic sleep staging by using Hidden Markov Model‏,” ieeexplore.ieee.org‏, Accessed: Aug. 06, 2020. [Online]. Available: https://ieeexplore.ieee.org/abstract/document/7009670/.
23- A. Hassan, M. B.-J. of neuroscience methods, and undefined 2016‏, “A decision support system for automatic sleep staging from EEG signals using tunable Q-factor wavelet transform and spectral features‏,” Elsevier‏, Accessed: Aug. 06, 2020. [Online]. Available: https://www.sciencedirect.com/science/article/pii/S0165027016301650.
24- A. Kozakevicius, C. Rodrigues, T. L. T. Da Silveira, A. J. Kozakevicius, and C. R. Rodrigues, “Single-channel EEG sleep stage classification based on a streamlined set of statistical features in wavelet domain,” Springer 2016‏, , doi: 10.1007/s11517-016-1519-4.
25- B. Koley, D. D.-C. in biology and medicine, and undefined 2012‏, “An ensemble system for automatic sleep stage classification using single channel EEG signal‏,” Elsevier‏, Accessed: Aug. 06, 2020. [Online]. Available: https://www.sciencedirect.com/science/article/pii/S0010482512001588.
26- V. Bajaj, R. P.-C. methods and programs in biomedicine, and undefined 2013‏, “Automatic classification of sleep stages based on the time-frequency image of EEG signals‏,” Elsevier‏, Accessed: Aug. 06, 2020. [Online]. Available: https://www.sciencedirect.com/science/article/pii/S0169260713002265.
27- T. Lajnef, S. Chaibi, P. Ruby, … P. A.-J. of neuroscience, and undefined 2015‏, “Learning machines and sleeping brains: automatic sleep stage classification using decision-tree multi-class support vector machines‏,” Elsevier‏, Accessed: Aug. 06, 2020. [Online]. Available: https://www.sciencedirect.com/science/article/pii/S0165027015000230.
28- Y. Peng, C. Lin, … M. S.-… on B. C., and undefined 2007‏, “Multimodality sensor system for long-term sleep quality monitoring‏,” ieeexplore.ieee.org‏, Accessed: Aug. 06, 2020. [Online]. Available: https://ieeexplore.ieee.org/abstract/document/4432397/.
29- K. Mikkelsen and P. Kidmose, “Automatic sleep stage classification using ear-EEG Threshold Games and Cooperation on Multiplayer Graphs View project mobile sleep monitoring with ear-EEG View project,” ieeexplore.ieee.org2016‏, , doi: 10.1109/EMBC.2016.7591789.
30- M. E. Tagluk, N. Sezgin, and M. Akin, “Estimation of sleep stages by an artificial neural network employing EEG, EMG and EOG,” J. Med. Syst., vol. 34, no. 4, pp. 717–725, Aug. 2010, doi: 10.1007/s10916-009-9286-5.
31- P. Vijaylaxmi, Jain, V. D. Mytri, V. V Shete, and B. K. Shiragapur, Sleep Stages Classification Using Wavelet Transform & Neural Network Sleep Stages Classification Using WaveletTransform & Neural Network. 2012.
32- W. Karlen, C. Mattiussi, and D. Floreano, “Sleep and Wake Classification With ECG and Respiratory Effort Signals.” Accessed: Aug. 06, 2020. [Online]. Available: http://ieeexplore.ieee.org.
33- R. Boostani, F. Karimzadeh, and M. Torabi-Nami, “A Comparative Review on Sleep Stage Classification Methods in Patients and healthy Individuals A Comparative Review on Sleep Stage Classification Methods in Patients and healthy Individuals A Comparative Review on Sleep Stage Classification Methods in Patients and healthy Individuals.” Accessed: Aug. 07, 2020. [Online]. Available: https://hal.archives-ouvertes.fr/hal-01390384.
34- L. Fraiwan, K. Lweesy, N. Khasawneh, … H. W.-C. methods and, and undefined 2012‏, “Automated sleep stage identification system based on time–frequency analysis of a single EEG channel and random forest classifier‏,” Elsevier‏, Accessed: Aug. 07, 2020. [Online]. Available: https://www.sciencedirect.com/science/article/pii/S0169260711003105.
35- O. Tsinalis, P. M. Matthews, and Y. Guo, “Automatic Sleep Stage Scoring Using Time-Frequency Analysis and Stacked Sparse Autoencoders,” Ann. Biomed. Eng., vol. 44, no. 5, pp. 1587–1597, May 2016, doi: 10.1007/s10439-015-1444-y.
36- S. Chambon, M. N. Galtier, P. J. Arnal, G. Wainrib, and A. Gramfort, “A deep learning architecture for temporal sleep stage classification using multivariate and multimodal time series.” Accessed: Aug. 07, 2020. [Online]. Available: https://ieeexplore.ieee.org/abstract/document/8307462/.
37- A. Supratak, H. Dong, C. Wu, and Y. Guo, “DeepSleepNet: a Model for Automatic Sleep Stage Scoring based on Raw Single-Channel EEG,” ieeexplore.ieee.org‏, doi: 10.1109/TNSRE.2017.2721116.
38- A. L. Goldberger et al., “PhysioBank, PhysioToolkit, and PhysioNet: components of a new research resource for complex physiologic signals.,” Circulation, vol. 101, no. 23, 2000, doi: 10.1161/01.cir.101.23.e215.
39- “Sleep-EDF Database v1.0.0.” https://physionet.org/content/sleep-edf/1.0.0/ (accessed Aug. 07, 2020).
40- A. Krizhevsky, I. Sutskever, and G. E. Hinton, “ImageNet Classification with Deep Convolutional Neural Networks.” Accessed: Aug. 07, 2020. [Online]. Available: http://code.google.com/p/cuda-convnet/.
41- O. Faust, Y. Hagiwara, T. J. Hong, S. Lih, and R. Acharya, “Deep learning for healthcare applications based on physiological signals: A review.” Accessed: Aug. 07, 2020. [Online]. Available: http://shura.shu.ac.uk/21073/.
42- Z. Mousavi, T. Rezaii, … S. S.-J. of neuroscience, and undefined 2019‏, “Deep convolutional neural network for classification of sleep stages from single-channel EEG signals‏,” Elsevier‏, Accessed: Aug. 07, 2020. [Online]. Available: https://www.sciencedirect.com/science/article/pii/S0165027019301700.
43- P. Memar, F. F.-I. T. on N. S. and, and undefined 2017‏, “A novel multi-class EEG-based sleep stage classification system‏,” ieeexplore.ieee.org‏, Accessed: Aug. 07, 2020. [Online]. Available: https://ieeexplore.ieee.org/abstract/document/8116601/.
44- N. V. Chawla, K. W. Bowyer, L. O. Hall, and W. P. Kegelmeyer, “SMOTE: Synthetic minority over-sampling technique,” J. Artif. Intell. Res., vol. 16, pp. 321–357, Jan. 2002, doi: 10.1613/jair.953.
45- O. Tsinalis, P. M. Matthews, Y. Guo, and S. Zafeiriou, “Automatic Sleep Stage Scoring with Single-Channel EEG Using Convolutional Neural Networks,” Oct. 2016, Accessed: Aug. 08, 2020. [Online]. Available: http://arxiv.org/abs/1610.01683.
46- A. Supratak, H. Dong, C. Wu, and Y. Guo, “DeepSleepNet: a Model for Automatic Sleep Stage Scoring based on Raw Single-Channel EEG,” ieeexplore.ieee.org‏, doi: 10.1109/TNSRE.2017.2721116.
47- A. Sors, S. Bonnet, S. Mirek, … L. V.-… S. P. and, and undefined 2018‏, “A convolutional neural network for sleep stage scoring from raw single-channel EEG‏,” Elsevier‏, Accessed: Aug. 08, 2020. [Online]. Available: https://www.sciencedirect.coAutomatic sleep stage classification using 1D convolutional neural network m/science/article/pii/S1746809417302847.
48- E. Fernandez-Blanco, D. Rivero, and A. Pazos, “Convolutional neural networks for sleep stage scoring on a two-channel EEG signal,” Soft Comput., vol. 24, no. 6, pp. 4067–4079, Mar. 2020, doi: 10.1007/s00500-019-04174-1.
Files
IssueVol 7 No 3 (2020) QRcode
SectionOriginal Article(s)
DOI https://doi.org/10.18502/fbt.v7i3.4616
Keywords
Sleep Staging 1D Convolutional Neural Network Classification Electroencephalogram

Rights and permissions
Creative Commons License This work is licensed under a Creative Commons Attribution-NonCommercial 4.0 International License.
How to Cite
1.
Salamatian A, Khadem A. Automatic Sleep Stage Classification Using 1D Convolutional Neural Network. Frontiers Biomed Technol. 2020;7(3):142-150.