Abstract
This paper addresses the crucial task of automatic sleep stage classification to assist sleep experts in diagnosing sleep disorders such as sleep apnea and insomnia. The proposed solution presents a novel attention-based deep learning model called, Efficient Attention-sleep Model (EASM), designed specifically for sleep apnea detection using EEG signals. EASM incorporates a streamlined architecture that includes a modified Muti-Resolution Convolutional Neural Network (MRCNN), Adaptive Feature Recalibration (AFR), and a simplified Temporal Context Encoder (TCE) module to reduce complexity. To mitigate overfitting, ridge regression is utilized, which incorporates a penalty term to enhance model generalization. Furthermore, the proposed EASM utilizes a class-balanced focal loss function to address data imbalance issues. The effectiveness of EASM is evaluated on two publicly available datasets, SLEEP EDF-20 and SLEEP EDF-78. Comparative analysis of EASM against state-of-the-art models demonstrates its superior performance in terms of accuracy, training time, and model complexity. Notably, the proposed model achieves a 50% reduction in training time and a 55.7% decrease in complexity compared to the Attnsleep model. The EASM achieves a classification accuracy of 85.8% with minimum loss when compared to the Attnsleep model.
Similar content being viewed by others
Data Availability
The Data used in this work is available in the public domain.
References
Luyster FS, Strollo PJ Jr, Zee PC, Walsh JK (2012) Sleep: a health imperative. Sleep 35(6):727–734
Zhao X, Wang X, Yang T, Ji S, Wang H, Wang J, Wang Y, Wu Q (2021) Classification of sleep apnea based on eeg sub-band signal characteristics. Sci Rep 11(1):5824
Zywietz C, Von Einem V, Widiger B, Joseph G (2004) Ecg analysis for sleep apnea detection. Methods Inf Med 43(01):56–59
Eldele E, Chen Z, Liu C, Wu M, Kwoh C-K, Li X, Guan C (2021) An attention-based deep learning approach for sleep stage classification with single-channel eeg. IEEE Trans Neural Syst Rehabil Eng 29:809–818. https://doi.org/10.1109/TNSRE.2021.3076234
Rauchs G, Desgranges B, Foret J, Eustache F (2005) The relationships between memory systems and sleep stages. J Sleep Res 14(2):123–140
Supratak A, Guo Y (2020) Tinysleepnet: an efficient deep learning model for sleep stage scoring based on raw single-channel eeg. In: 2020 42nd Annual international conference of the IEEE engineering in medicine & biology society (EMBC), pp 641–644. https://doi.org/10.1109/EMBC44109.2020.9176741
Seo H, Back S, Lee S, Park D, Kim T, Lee K (2020) Intra- and inter-epoch temporal context network (iitnet) using sub-epoch features for automatic sleep scoring on raw single-channel eeg. Biomed Signal Process Control 61:102037. https://doi.org/10.1016/j.bspc.2020.102037
Pant H, Dhanda HK, Taran S (2022) Sleep apnea detection using electrocardiogram signal input to fawt and optimize ensemble classifier. Measurement 189:110485. https://doi.org/10.1016/j.measurement.2021.110485
Zhu G, Li Y, Wen P (2014) Analysis and classification of sleep stages based on difference visibility graphs from a single-channel eeg signal. IEEE J Biomed Health Inform 18(6):1813–1821
Memar P, Faradji F (2017) A novel multi-class eeg-based sleep stage classification system. IEEE Trans Neural Syst Rehabil Eng 26(1):84–95
Dimitriadis SI, Salis C, Linden D (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
Dash S, Acharya BR, Mittal M, Abraham A, Kelemen A (2020) Deep learning techniques for biomedical and health informatics. Springer, ???
Mohanty C, Mahapatra S, Acharya B, Kokkoras F, Gerogiannis VC, Karamitsos I, Kanavos A (2023) Using deep learning architectures for detection and classification of diabetic retinopathy. Sensors 23(12):5726
Tsinalis O, Matthews PM, Guo Y, Zafeiriou S (2016) Automatic sleep stage scoring with single-channel eeg using convolutional neural networks. arXiv:1610.01683
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 Bioinf 17(6):1835–1845
Supratak A, Dong H, Wu C, Guo Y (2017) Deepsleepnet: a model for automatic sleep stage scoring based on raw single-channel eeg. IEEE Trans Neural Syst Rehabil Eng 25(11):1998–2008. https://doi.org/10.1109/TNSRE.2017.2721116
Phan H, Andreotti F, Cooray N, Chén OY, De Vos M (2018) Joint classification and prediction cnn framework for automatic sleep stage classification. IEEE Trans Biomed Eng 66(5):1285–1296
Li F, Yan R, Mahini R, Wei L, Wang Z, Mathiak K, Liu R, Cong F (2021) End-to-end sleep staging using convolutional neural network in raw single-channel eeg. Biomed Signal Process Control 63:102203
Tsinalis O, Matthews PM, Guo Y, Zafeiriou S (2016) Automatic sleep stage scoring with single-channel eeg using convolutional neural networks. arXiv:1610.01683
Chen Z, Wu M, Cui W, Liu C, Li X (2020) An attention based cnn-lstm approach for sleep-wake detection with heterogeneous sensors. IEEE J Biomed Health Inform 25(9):3270–3277
Zhu T, Luo W, Yu F (2020) Convolution-and attention-based neural network for automated sleep stage classification. Int J Environ Res Public Health 17(11):4152
Memar P, Faradji F (2018) A novel multi-class eeg-based sleep stage classification system. IEEE Trans Neural Syst Rehabil Eng 26(1):84–95. https://doi.org/10.1109/TNSRE.2017.2776149
Cheng G, Si Y, Hong H, Yao X, Guo L (2020) Cross-scale feature fusion for object detection in optical remote sensing images. IEEE Geosci Remote Sens Lett 18(3):431–435
Vaswani A, Shazeer N, Parmar N, Uszkoreit J, Jones L, Gomez AN, Kaiser Ł, Polosukhin I (2017) Attention is all you need. Adv Neural Inf Process Syst 30
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. Biomed Signal Process Control 42:107–114
Goldberger AL, Amaral LA, Glass L, Hausdorff JM, Ivanov PC, Mark RG, Mietus JE, Moody GB, Peng C-K, Stanley HE (2000) Physiobank, physiotoolkit, and physionet: components of a new research resource for complex physiologic signals. Circulation 101(23):215–220
Author information
Authors and Affiliations
Corresponding author
Ethics declarations
Conflicts of Interest
The authors declare that they have no known conflict of interest or personal relationships that could have appeared to influence the work reported in this paper.
Additional information
Publisher's Note
Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.
Rights and permissions
Springer Nature or its licensor (e.g. a society or other partner) holds exclusive rights to this article under a publishing agreement with the author(s) or other rightsholder(s); author self-archiving of the accepted manuscript version of this article is solely governed by the terms of such publishing agreement and applicable law.
About this article
Cite this article
Singh, M., Chauhan, S., Rajput, A.K. et al. EASM: An efficient AttnSleep model for sleep Apnea detection from EEG signals. Multimed Tools Appl (2024). https://doi.org/10.1007/s11042-024-19118-7
Received:
Revised:
Accepted:
Published:
DOI: https://doi.org/10.1007/s11042-024-19118-7