Papers
Topics
Authors
Recent
Gemini 2.5 Flash
Gemini 2.5 Flash
153 tokens/sec
GPT-4o
7 tokens/sec
Gemini 2.5 Pro Pro
45 tokens/sec
o3 Pro
4 tokens/sec
GPT-4.1 Pro
38 tokens/sec
DeepSeek R1 via Azure Pro
28 tokens/sec
2000 character limit reached

Transparency in Sleep Staging: Deep Learning Method for EEG Sleep Stage Classification with Model Interpretability (2309.07156v4)

Published 10 Sep 2023 in eess.SP and cs.LG

Abstract: Automated Sleep stage classification using raw single channel EEG is a critical tool for sleep quality assessment and disorder diagnosis. However, modelling the complexity and variability inherent in this signal is a challenging task, limiting their practicality and effectiveness in clinical settings. To mitigate these challenges, this study presents an end-to-end deep learning (DL) model which integrates squeeze and excitation blocks within the residual network to extract features and stacked Bi-LSTM to understand complex temporal dependencies. A distinctive aspect of this study is the adaptation of GradCam for sleep staging, marking the first instance of an explainable DL model in this domain with alignment of its decision-making with sleep expert's insights. We evaluated our model on the publically available datasets (SleepEDF-20, SleepEDF-78, and SHHS), achieving Macro-F1 scores of 82.5, 78.9, and 81.9, respectively. Additionally, a novel training efficiency enhancement strategy was implemented by increasing stride size, leading to 8x faster training times with minimal impact on performance. Comparative analyses underscore our model outperforms all existing baselines, indicating its potential for clinical usage.

Definition Search Book Streamline Icon: https://streamlinehq.com
References (49)
  1. Sleep and its importance in adolescence and in common adolescent somatic and psychiatric conditions. International Journal of General Medicine, 4:425–442, 2011.
  2. Rules for scoring respiratory events in sleep: update of the 2007 aasm manual for the scoring of sleep and associated events: deliberations of the sleep apnea definitions task force of the american academy of sleep medicine. Journal of clinical sleep medicine, 8:597–619, 2012.
  3. Allan Rechtschaffen. A manual of standardized terminology, techniques and scoring system for sleep stages of human subjects. pages 1–55, 1968.
  4. Automated sleep scoring: A review of the latest approaches. Sleep medicine reviews, 48:101204, 2019.
  5. Recent advances in convolutional neural networks. Pattern recognition, 77:354–377, 2018.
  6. Long short-term memory. Neural computation, 9(8):1735–1780, 1997.
  7. Bidirectional recurrent neural networks. IEEE transactions on Signal Processing, 45(11):2673–2681, 1997.
  8. Empirical evaluation of gated recurrent neural networks on sequence modeling. arXiv preprint arXiv:1412.3555, 2014.
  9. Squeeze-and-excitation networks. In Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition (CVPR), June 2018.
  10. A comparative study on classification of sleep stage based on eeg signals using feature selection and classification algorithms. Journal of medical systems, 38:1–21, 2014.
  11. Combination of heterogeneous eeg feature extraction methods and stacked sequential learning for sleep stage classification. International journal of neural systems, 23(03):1350012, 2013.
  12. Estimation of sleep stages by an artificial neural network employing eeg, emg and eog. Journal of medical systems, 34:717–725, 2010.
  13. Learning machines and sleeping brains: automatic sleep stage classification using decision-tree multi-class support vector machines. Journal of neuroscience methods, 250:94–105, 2015.
  14. A new automatic sleep staging system based on statistical behavior of local extrema using single channel eeg signal. Expert Systems with Applications, 104:277–293, 2018.
  15. Automatic sleep stage classification using single-channel eeg: Learning sequential features with attention-based recurrent neural networks. In 2018 40th Annual International Conference of the IEEE Engineering in Medicine and Biology Society (EMBC), pages 1452–1455, 2018.
  16. Ahnaf Rashik Hassan and Mohammed Imamul Hassan Bhuiyan. Computer-aided sleep staging using complete ensemble empirical mode decomposition with adaptive noise and bootstrap aggregating. Biomedical Signal Processing and Control, 24:1–10, 2016.
  17. Automated sleep stage identification system based on time–frequency analysis of a single eeg channel and random forest classifier. Computer methods and programs in biomedicine, 108(1):10–19, 2012.
  18. Sleep stage classification using single-channel eog. Computers in biology and medicine, 102:211–220, 2018.
  19. Dnn filter bank improves 1-max pooling cnn for single-channel eeg automatic sleep stage classification. In 2018 40th Annual International Conference of the IEEE Engineering in Medicine and Biology Society (EMBC), pages 453–456, 2018.
  20. Visualising convolutional neural network decisions in automatic sleep scoring. In CEUR Workshop Proceedings, pages 70–81. CEUR Workshop Proceedings, 2018.
  21. Joint classification and prediction cnn framework for automatic sleep stage classification. IEEE Transactions on Biomedical Engineering, 66(5):1285–1296, 2019.
  22. A deep learning architecture for temporal sleep stage classification using multivariate and multimodal time series. IEEE Transactions on Neural Systems and Rehabilitation Engineering, 26(4):758–769, 2018.
  23. muleeg: A multi-view representation learning on eeg signals. In Medical Image Computing and Computer Assisted Intervention – MICCAI 2022, pages 398–407. Springer Nature Switzerland, 2022.
  24. A residual based attention model for eeg based sleep staging. IEEE journal of biomedical and health informatics, 24(10):2833–2843, 2020.
  25. Deepsleepnet: A model for automatic sleep stage scoring based on raw single-channel eeg. IEEE Transactions on Neural Systems and Rehabilitation Engineering, 25(11):1998–2008, 2017.
  26. Mixed neural network approach for temporal sleep stage classification. IEEE Transactions on Neural Systems and Rehabilitation Engineering, 26(2):324–333, 2018.
  27. Cascaded lstm recurrent neural network for automated sleep stage classification using single-channel eeg signals. Computers in biology and medicine, 106:71–81, 2019.
  28. Automatic sleep stage classification method based on transformer-in-transformer. In 2023 11th International Winter Conference on Brain-Computer Interface (BCI), pages 1–5, 2023.
  29. Sleeptransformer: Automatic sleep staging with interpretability and uncertainty quantification. IEEE Transactions on Biomedical Engineering, 69(8):2456–2467, 2022.
  30. Seqsleepnet: End-to-end hierarchical recurrent neural network for sequence-to-sequence automatic sleep staging. IEEE Transactions on Neural Systems and Rehabilitation Engineering, 27(3):400–410, 2019.
  31. Enhancing healthcare with eog: A novel approach to sleep stage classification. arXiv preprint arXiv:2310.03757, 2023.
  32. Automatic sleep staging based on a hybrid stacked lstm neural network: Verification using large-scale dataset. IEEE Access, 8:111837–111849, 2020.
  33. Fusion of end-to-end deep learning models for sequence-to-sequence sleep staging. In 2019 41st Annual International Conference of the IEEE Engineering in Medicine and Biology Society (EMBC), pages 1829–1833, 2019.
  34. A two-stage neural network for sleep stage classification based on feature learning, sequence learning, and data augmentation. IEEE Access, 7:109386–109397, 2019.
  35. Sleepeegnet: Automated sleep stage scoring with sequence to sequence deep learning approach. PloS one, 14(5):e0216456, 2019.
  36. Xsleepnet: Multi-view sequential model for automatic sleep staging. IEEE Transactions on Pattern Analysis and Machine Intelligence, 44(9):5903–5915, 2021.
  37. Deep residual learning for image recognition. In 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), pages 770–778, 2016.
  38. Imagenet: A large-scale hierarchical image database. In 2009 IEEE Conference on Computer Vision and Pattern Recognition, pages 248–255, 2009.
  39. Grad-cam: Visual explanations from deep networks via gradient-based localization. In 2017 IEEE International Conference on Computer Vision (ICCV), pages 618–626, 2017.
  40. Eeg-based sleep stage classification via neural architecture search. IEEE Transactions on Neural Systems and Rehabilitation Engineering, 31:1075–1085, 2023.
  41. Intra-and inter-epoch temporal context network (iitnet) using sub-epoch features for automatic sleep scoring on raw single-channel eeg. Biomedical signal processing and control, 61:102037, 2020.
  42. Improved deep learning classification of human sleep stages. In 2020 IEEE 33rd international symposium on computer-based medical systems (CBMS), pages 338–343. IEEE, 2020.
  43. Sleepcontextnet: A temporal context network for automatic sleep staging based single-channel eeg. Computer Methods and Programs in Biomedicine, 220:106806, 2022.
  44. An attention-based deep learning approach for sleep stage classification with single-channel eeg. IEEE Transactions on Neural Systems and Rehabilitation Engineering, 29:809–818, 2021.
  45. Automatic sleep stage classification using temporal convolutional neural network and new data augmentation technique from raw single-channel eeg. Computer Methods and Programs in Biomedicine, 204:106063, 2021.
  46. The national sleep research resource: towards a sleep data commons. Journal of the American Medical Informatics Association, 25(10):1351–1358, 2018.
  47. The sleep heart health study: design, rationale, and methods. Sleep, 20(12):1077–1085, 1997.
  48. Physiobank, physiotoolkit, and physionet: components of a new research resource for complex physiologic signals. circulation, 101(23):e215–e220, 2000.
  49. Analysis of a sleep-dependent neuronal feedback loop: the slow-wave microcontinuity of the eeg. IEEE Transactions on Biomedical Engineering, 47(9):1185–1194, 2000.
Citations (1)
View on Semantic Scholar

Summary

We haven't generated a summary for this paper yet.

Ai Sparkles Streamline Icon: https://streamlinehq.com Summarize Now