Skip to main content

Advertisement

Springer Nature Link
Log in

A novel solution of enhanced loss function using deep learning in sleep stage classification: predict and diagnose patients with sleep disorders

  • Published:
Multimedia Tools and Applications Aims and scope Submit manuscript

Abstract

Sleep stage classification is important to accurately predict and diagnose patients with sleep disorders. Though various deep learning approaches have been implemented to classify sleep classes, these consist limitations that impact the accuracy and processing time of the classification model. The aim of this research is to enhance the accuracy and minimize the training time of the deep learning classification model. The proposed system consists of One Dimensional Convolutional Neural Network (CNN) with enhanced loss function to improve the accuracy of scoring of five different sleep classes. Preprocessing, Feature Extraction and Classification are the main components of the proposed system. Initially, EEG signals are fed to an adaptive filter for preprocessing, in order to remove any noise in signal. Thereafter, feature is extracted through multiple convolutional and pooling layers, and finally the classification is done by fully connected layer using softmax activation with enhanced loss function. The proposed solution is tested on data samples from multiple datasets with five classes of Sleep classification. Based on the obtained results, the proposed solution has found to achieve an accuracy of 96.26% which is almost 4.2% higher than the state-of-the-art solution which is 92.76%. Furthermore, the processing time has been reduced by 11 milliseconds against the state-of-the-art solution. The proposed system focused on classifying sleep stages in five classes using EEG signals with deep learning approach. It enhances the loss function in order to minimize errors in the prediction of sleep classes and improves the accuracy of the model. Furthermore, the training speed of the model has also been reduced by applying batch normalization techniques inside the model. In the future, larger datasets of different sleep disorder patients with varying features can be used for training and implementing the proposed solution. The datasets can also be pre-processed using additional techniques to refine the data before feeding to the neural network model.

This is a preview of subscription content, log in via an institution to check access.

Access this article

Log in via an institution

Subscribe and save

Springer+ Basic
$34.99 /Month
  • Get 10 units per month
  • Download Article/Chapter or eBook
  • 1 Unit = 1 Article or 1 Chapter
  • Cancel anytime
Subscribe now

Buy Now

Price excludes VAT (USA)
Tax calculation will be finalised during checkout.

Instant access to the full article PDF.

Institutional subscriptions

Fig. 1
Fig. 2
Fig. 3
Fig. 4
Fig. 5
Fig. 6
Fig. 7
Fig. 8
Fig. 9
Fig. 10
Fig. 11

Similar content being viewed by others

References

  1. Aboalayon K, Faezipour M, Almuhammadi W, Moslehpour S (2016) Sleep stage classification using EEG signal analysis: a comprehensive survey and new investigation. Entropy 18(9). https://doi.org/10.3390/e18090272

  2. Alickovic E, Subasi A (2018) Ensemble SVM method for automatic sleep stage classification. IEEE Trans Instrum Meas 67(6):1258–1265. https://doi.org/10.1109/tim.2018.2799059

    Article  Google Scholar 

  3. Biswal S, Sun H, Goparaju B, Westover MB, Sun J, Bianchi MT (2018) Expert-level sleep scoring with deep neural networks. J Am Med Inform Assoc 25(12):1643–1650. https://doi.org/10.1093/jamia/ocy131

    Article  Google Scholar 

  4. Boostani R, Karimzadeh F, Nami M (Mar 2017) A comparative review on sleep stage classification methods in patients and healthy individuals. Comput Methods Prog Biomed 140:77–91. https://doi.org/10.1016/j.cmpb.2016.12.004

    Article  Google Scholar 

  5. Bresch E, Grossekathofer U, Garcia-Molina G (2018) Recurrent deep neural networks for real-time sleep stage classification from Single Channel EEG. Front Comput Neurosci 12:85. https://doi.org/10.3389/fncom.2018.00085

    Article  Google Scholar 

  6. Chambon S, Galtier MN, Arnal PJ, Wainrib G, Gramfort A (Apr 2018) A deep learning architecture for temporal sleep stage classification using multivariate and multimodal time series. IEEE Trans Neural Syst Rehabil Eng 26(4):758–769. https://doi.org/10.1109/TNSRE.2018.2813138

    Article  Google Scholar 

  7. Dimitriadis SI, Salis C, Linden D (Apr 2018) A novel, fast and efficient single-sensor automatic sleep-stage classification based on complementary cross-frequency coupling estimates. Clin Neurophysiol 129(4):815–828. https://doi.org/10.1016/j.clinph.2017.12.039

    Article  Google Scholar 

  8. Dong H, Supratak A, Pan W, Wu C, Matthews PM, Guo Y (Feb 2018) Mixed neural network approach for temporal sleep stage classification. IEEE Trans Neural Syst Rehabil Eng 26(2):324–333. https://doi.org/10.1109/TNSRE.2017.2733220

    Article  Google Scholar 

  9. Ghasemzadeh P, Kalbkhani H, Sartipi S, Shayesteh MG (2019) Classification of sleep stages based on LSTAR model. Appl Soft Comput 75:523–536. https://doi.org/10.1016/j.asoc.2018.11.007

    Article  Google Scholar 

  10. Hassan AR, Subasi A (2017) A decision support system for automated identification of sleep stages from single-channel EEG signals. Knowl-Based Syst 128:115–124. https://doi.org/10.1016/j.knosys.2017.05.005

    Article  Google Scholar 

  11. Jeon Y, Kim S, Choi HS, Chung YG, Choi SA, Kim H, Yoon S, Hwang H, Kim KJ (2019) Pediatric sleep stage classification using multi-domain hybrid neural networks. IEEE Access 7:96495–96505. https://doi.org/10.1109/access.2019.2928129

    Article  Google Scholar 

  12. Kotchoubey B, Pavlov YG (Dec 2018) Machine learning versus human expertise: the case of sleep stage classification in disorders of consciousness. Response to Wislowska et al. Clin Neurophysiol 129(12):2682–2683. https://doi.org/10.1016/j.clinph.2018.09.020

    Article  Google Scholar 

  13. Mousavi S, Afghah F, Acharya UR (2019) SleepEEGNet: Automated sleep stage scoring with sequence to sequence deep learning approach. PLoS One 14(5):e0216456. https://doi.org/10.1371/journal.pone.0216456

    Article  Google Scholar 

  14. Mousavi Z, Yousefi Rezaii T, Sheykhivand S, Farzamnia A, Razavi SN (2019) Deep convolutional neural network for classification of sleep stages from single-channel EEG signals. J Neurosci Methods 324:108312. https://doi.org/10.1016/j.jneumeth.2019.108312

    Article  Google Scholar 

  15. Phan H, Andreotti F, Cooray N, Chen OY, De Vos M (May 2019) Joint classification and prediction CNN framework for automatic sleep stage classification. IEEE Trans Biomed Eng 66(5):1285–1296. https://doi.org/10.1109/TBME.2018.2872652

    Article  Google Scholar 

  16. Phan H, Andreotti F, Cooray N, Chen OY, De Vos M (Mar 2019) SeqSleepNet: end-to-end hierarchical recurrent neural network for sequence-to-sequence automatic sleep staging. IEEE Trans Neural Syst Rehabil Eng 27(3):400–410. https://doi.org/10.1109/TNSRE.2019.2896659

    Article  Google Scholar 

  17. Procházka A, Kuchyňka J, Vyšata O, Cejnar P, Vališ M, Mařík V (2018) Multi-class sleep stage analysis and adaptive& pattern recognition. Applied Sciences 8(5). https://doi.org/10.3390/app8050697

  18. Rahman MM, Bhuiyan MIH, Hassan AR (Nov 1 2018) Sleep stage classification using single-channel EOG. Comput Biol Med 102:211–220. https://doi.org/10.1016/j.compbiomed.2018.08.022

    Article  Google Scholar 

  19. Seifpour S, Niknazar H, Mikaeili M, Nasrabadi AM (2018) A new automatic sleep staging system based on statistical behavior of local Extrema using single channel EEG signal. Expert Syst Appl 104:277–293. https://doi.org/10.1016/j.eswa.2018.03.020

    Article  Google Scholar 

  20. Shen H, Xu M, Guez A, Li A, Ran F (2019) An accurate sleep stages classification method based on state space model. IEEE Access 7:125268–125279. https://doi.org/10.1109/access.2019.2939038

    Article  Google Scholar 

  21. Sokolovsky M, Guerrero F, Paisarnsrisomsuk S, Ruiz C, Alvarez SA (2019) Deep learning for automated feature discovery and classification of sleep stages. IEEE/ACM Trans Comput Biol Bioinform, https://doi.org/10.1109/TCBB.2019.2912955

  22. Sors A, Bonnet S, Mirek S, Vercueil L, Payen J-F (2018) A convolutional neural network for sleep stage scoring from raw single-channel EEG. Biomedical Signal Processing and Control 42:107–114. https://doi.org/10.1016/j.bspc.2017.12.001

    Article  Google Scholar 

  23. Sun C, Fan J, Chen C, Li W, Chen W (2019) A two-stage neural network for sleep stage classification based on feature learning, sequence learning, and data augmentation. IEEE Access 7:109386–109397. https://doi.org/10.1109/access.2019.2933814

    Article  Google Scholar 

  24. Tsinalis O, Matthews PM, Guo Y (May 2016) Automatic sleep stage scoring using time-frequency analysis and stacked sparse autoencoders. Ann Biomed Eng 44(5):1587–1597. https://doi.org/10.1007/s10439-015-1444-y

    Article  Google Scholar 

  25. Yildirim O, Baloglu UB, Acharya UR (2019) A deep learning model for automated sleep stages classification using PSG signals. Int J Environ Res Public Health 16(4). https://doi.org/10.3390/ijerph16040599

  26. Yuan Y, Jia K (2019) FusionAtt: deep fusional attention networks for multi-channel biomedical signals. Sensors (Basel) 19(11). https://doi.org/10.3390/s19112429

  27. Zhang J, Wu Y (Oct 2018) Complex-valued unsupervised convolutional neural networks for sleep stage classification. Comput Methods Prog Biomed 164:181–191. https://doi.org/10.1016/j.cmpb.2018.07.015

    Article  Google Scholar 

  28. Zhang X, Kou W, Chang EI, Gao H, Fan Y, Xu Y (Dec 1 2018) Sleep stage classification based on multi-level feature learning and recurrent neural networks via wearable device. Comput Biol Med 103:71–81. https://doi.org/10.1016/j.compbiomed.2018.10.010

    Article  Google Scholar 

  29. Zhang L, Fabbri D, Upender R, Kent D (2019) Automated sleep stage scoring of the sleep heart health study using deep neural networks. Sleep. https://doi.org/10.1093/sleep/zsz159

    Book  Google Scholar 

Download references

Author information

Authors and Affiliations

  1. School of Computing and Mathematics, Charles Sturt University, Sydney, Australia

    Ereena Rajbhandari, Abeer Alsadoon, P. W. C. Prasad & Indra Seher

  2. The University of Da Nang – University of Science and Education, Faculty of Information Technology, 459 Tôn Đức Thắng, Hoà Khánh Nam, Liên Chiểu, Đà Nẵng, 550000, Vietnam

    Tran Quoc Vinh Nguyen & Duong Thu Hang Pham

Authors
  1. Ereena Rajbhandari
    View author publications

    You can also search for this author inPubMed Google Scholar

  2. Abeer Alsadoon
    View author publications

    You can also search for this author inPubMed Google Scholar

  3. P. W. C. Prasad
    View author publications

    You can also search for this author inPubMed Google Scholar

  4. Indra Seher
    View author publications

    You can also search for this author inPubMed Google Scholar

  5. Tran Quoc Vinh Nguyen
    View author publications

    You can also search for this author inPubMed Google Scholar

  6. Duong Thu Hang Pham
    View author publications

    You can also search for this author inPubMed Google Scholar

Corresponding author

Correspondence to Abeer Alsadoon.

Additional information

Publisher’s note

Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.

Appendix

Appendix

Table 9 Abbreviation table
Full size table

Rights and permissions

Reprints and permissions

About this article

Check for updates. Verify currency and authenticity via CrossMark

Cite this article

Rajbhandari, E., Alsadoon, A., Prasad, P.W.C. et al. A novel solution of enhanced loss function using deep learning in sleep stage classification: predict and diagnose patients with sleep disorders. Multimed Tools Appl 80, 11607–11630 (2021). https://doi.org/10.1007/s11042-020-10199-8

Download citation

  • Received:

  • Revised:

  • Accepted:

  • Published:

  • Issue Date:

  • DOI: https://doi.org/10.1007/s11042-020-10199-8

Keywords