Abstract
With the development of telemedicine and edge computing, edge artificial intelligence (AI) will become a new development trend for smart medicine. On the other hand, nearly one-third of children suffer from sleep disorders. However, all existing sleep staging methods are for adults. Therefore, we adapted edge AI to develop a lightweight automatic sleep staging method for children using single-channel EEG. The trained sleep staging model will be deployed to edge smart devices so that the sleep staging can be implemented on edge devices which will greatly save network resources and improving the performance and privacy of sleep staging application. Then the results and hypnogram will be uploaded to the cloud server for further analysis by the physicians to get sleep disease diagnosis reports and treatment opinions. We utilized 1D convolutional neural networks (1D-CNN) and long short term memory (LSTM) to build our sleep staging model, named CSleepNet. We tested the model on our childrens sleep (CS) dataset and sleep-EDFX dataset. For the CS dataset, we experimented with F4-M1 channel EEG using four different loss functions, and the logcosh performed best with overall accuracy of 83.06% and F1-score of 76.50%. We used Fpz-Cz and Pz-Oz channel EEG to train our model in Sleep-EDFX dataset, and achieved an accuracy of 86.41% without manual feature extraction. The experimental results show that our method has great potential. It not only plays an important role in sleep-related research, but also can be widely used in the classification of other time sequences physiological signals.
Similar content being viewed by others
References
Agarwal, R., Gotman, J.: Computer-assisted sleep staging. IEEE transactions on bio-medical engineering 48(12), 1412–1423 (2001). https://doi.org/10.1109/10.966600
Agarwal, A., Sharma, S., Kumar, V., Kaur, M.: Effect of E-Learning on Public Health and Environment During COVID-19 Lockdown. Big Data Mining and Analytics 4(2), 104–115 (2021)
Alickovic, E., Subasi, A.: Ensemble SVM Method for Automatic Sleep Stage Classification. IEEE Transactions on Instrumentation and Measurement 67(6), 1258–1265 (2018). https://doi.org/10.1109/TIM.2018.2799059
Berry, R.B., Quan, S.F., Abreu, A.R., et al.: The AASM manual for the scoring of sleep and associated events: rules, terminology and technical specifications(Version 2.6), american academy of sleep medicine (2020)
Bevans, K.B., Meltzer, L.J., Anna, D.L.M., Kratchman, A., Vil, D., Forrest, C.B.: Qualitative development and content validation of the PROMIS pediatric sleep health items. Behav. Sleep Med. 657–671 (2018). https://doi.org/10.1080/15402002.2018.1461102
Boostani, R., Karimzadeh, F., Nami, M.: A comparative review on sleep stage classification methods in patients and healthy individuals. Computer Methods and Programs in Biomedicine 140, 77–91 (2017). https://doi.org/10.1016/j.cmpb.2016.12.004
Estrada, E., Nazeran, H., Nava, P., Behbehani, K., Burk, J., Lucas, E.: EEG feature extraction for classification of sleep stages. In: The 26th Annual International Conference of the IEEE Engineering in Medicine and Biology Society, pp. 196–199 (2004).https://doi.org/10.1109/IEMBS.2004.1403125
Fraiwan, L., Alkhodari, M.: Investigating the use of uni-directional and bi-directional long short-term memory models for automatic sleep stage scoring. Inform. Med. Unlocked 20, (2020). https://doi.org/10.1016/j.imu.2020.100370
Goldberger, A., Amaral, L., Glass, I., et al.: PhysioBank, physiotoolkit, and physionet: components of a new research resource for complex physiologic signals. Circulation Online 101(23), 215–220 (2000)
Grigg-Damberger, M., Gozal, D., Quan, S.F., Rosen, C.L.: The visual scoring of sleep and arousal in infants and children. Journal of Clinical Sleep Medicine Jcsm Official Publication of the American Academy of Sleep Medicine 3(2), 201–240 (2007)
Hanan, E.S.: Sleep physiology and sleep disorders in childhood. Nature and Science of Sleep 3, 101–114 (2011). https://doi.org/10.2147/NSS.S22839
Hassan, A.R., Bhuiyan, M.: A decision support system for automatic sleep staging from EEG signals using tunable Q-factor wavelet transform and spectral features. Journal of Neuroscience Methods 271, 107–118 (2016). https://doi.org/10.1016/j.jneumeth.2016.07.012
Hassan, A.R., Bhuiyan, M.: An automated method for sleep staging from EEG signals using normal inverse Gaussian parameters and adaptive boosting. Neurocomputing 219, 76–87 (2016). https://doi.org/10.1016/j.neucom.2016.09.011
Hassan, A.R., Bhuiyan, M.: A decision support system for automated identification of sleep stages from single-channel EEG signals. Knowledge-Based Systems 128, 115–124 (2017). https://doi.org/10.1016/j.knosys.2017.05.005
He, Q., Cui, G., Zhang, X., Chen, F., Deng, S., Jin, H., Yang, Y.: A Game-Theoretical Approach for User Allocation in Edge Computing Environment. IEEE Transactions on Parallel and Distributed Systems 31(3), 515–529 (2020). https://doi.org/10.1109/TPDS.2019.2938944
Hsu, Y., Yang, Y., Wang, J., Hsu, C.: Automatic sleep stage recurrent neural classifier using energy features of EEG signals. Neurocomputing 104(1), 105–114 (2013). https://doi.org/10.1016/j.neucom.2012.11.003
Huang, Z., Lin, K.J., Tsai, B.L., Yan, S., Shih, C.: Building edge intelligence for online activity recognition in service-oriented IoT systems. Future Generation Computer Systems 87, 557–567 (2018). https://doi.org/10.1016/j.future.2018.03.003
Huang, H., Yang, L., Wang, Y., Xu, X., Lu, Y.: Digital Twin-driven online anomaly detection for an automation system based on edge intelligence. Journal of Manufacturing Systems 59, 138–150 (2021). https://doi.org/10.1016/j.jmsy.2021.02.010
Kemp, B., Zwinderman, A.H., Tuk, B., Kamphuisen, H.A.C., Obery, J.J.L.: Analysis of a sleep-dependent neuronal feedback loop: the slow-wave microcontinuity of the EEG. IEEE-BME 47(9), 1185–1194 (2000). https://doi.org/10.1109/10.867928
Khandewale, A., Gohokar, V., Nawandar, P.: Edge intelligence-based object detection system using neural compute stick for visually impaired people. In: Senjyu, T., Mahalle, P.N., Perumal, T., Joshi, A. (eds) Information and Communication Technology for Intelligent Systems. ICTIS 2020. Smart Innovation, Systems and Technologies, 195, 433–439. Springer, Singapore (2021). https://doi.org/10.1007/978-981-15-7078-0_41
Liu, L., Lu, Z., Wang, L., Chen, X., Wen, X.: Large-volume data dissemination for cellular-assisted automated driving with edge intelligence. J. Netw. Comput. Appl. 155(2) (2020). https://doi.org/10.1016/j.jnca.2020.102535
Reschtchaffen, A., Kales, A.: A manual of standardized terminology, techniques and scoring system for sleep stages of human subjects. Electroencephalography and Clinical Neurophysiology 26(2), 644 (1969). https://doi.org/10.1016/0013-4694(69)90021-2
Rincon, J.A., Julian, V., Carrascosa, C.: Towards the edge intelligence: robot assistant for the detection and classification of human Emotions. In: De La Prieta, F., et al. (eds) Highlights in Practical Applications of Agents, Multi-Agent Systems, and Trust-worthiness. The PAAMS Collection. PAAMS 2020. Communications in Computer and Information Science, 1233, 31-41, Springer, Cham (2020). https://doi.org/10.1007/978-3-030-51999-5_3
See A.R., Liang C.K.: A study on sleep EEG using sample entropy and power spectrum analysis. In: 2011 Defense Science Research Conference and Expo (DSR), pp. 1–4 (2011). https://doi.org/10.1109/DSR.2011.6026802
Shi, W., Cao, J., Zhang, Q., Li, Y., Xu, L.: Edge Computing: Vision and Challenges. IEEE Internet of Things Journal 3(5), 637–646 (2016). https://doi.org/10.1109/JIOT.2016.2579198
Song, Z., Cao, Z., Li, Z., Wang, J., Liu, Y.: Inertial Motion Tracking on Mobile and Wearable Devices: Recent Advancements and Challenges. Tsinghua Science and Technology 26(5), 692–705 (2021)
Sors, A., Bonnet, S., Mirek, S., Vercueil, L., Payen, J.: A convolutional neural network for sleep stage scoring from raw single-channel EEG. Biomedical Signal Processing and Control 42, 107–114 (2018). https://doi.org/10.1016/j.bspc.2017.12.001
Stepnowsky, C., Levendowski, D., Popovic, D., Ayappa, I., Rapoport, D.M.: Scoring accuracy of automated sleep staging from a bipolar electroocular recording compared to manual scoring by multiple raters. Sleep Medicine 14(11), 1199–1207 (2013). https://doi.org/10.1016/j.sleep.2013.04.022
Supratak, A., Hao, D., Chao, W., Guo, Y.: 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). https://doi.org/10.1109/TNSRE.2017.2721116
Uddin, M.Z., Nilsson, E.G.: Emotion recognition using speech and neural structured learning to facilitate edge intelligence - ScienceDirect. Eng. Appl. Artif. Intell. 94 (2020). https://doi.org/10.1016/j.engappai.2020.103775
Zhang, J., Wu, Y.: Competition convolutional neural network for sleep stage classification. Biomed. Signal Process. Control 64 (2021). https://doi.org/10.1016/j.bspc.2020.102318
Zhang, J., Yan, W.: A New Method for Automatic Sleep Stage Classification. IEEE Transactions on Biomedical Circuits and Systems 11(5), 1097–1110 (2017). https://doi.org/10.1109/TBCAS.2017.2719631
Zhang, Z., Cong, X., Feng, W., Zhang, H., Fu, G., Chen, J.: WAEAS: An Optimization Scheme of EAS Scheduler for Wearable Applications. Tsinghua Science and Technology 26(1), 72–84 (2021)
Zhang, C., Liu, S., Han, F., Nie, Z., Lo, B., Zhang, Y.: Hybrid manifold-deep convolutional neural network for sleep staging. Methods (2021). https://doi.org/10.1016/j.ymeth.2021.02.014
Zhen, P., Han, Y., Dong, A., Yu, J.: CareEdge: A Lightweight Edge Intelligence Framework for ECG-Based Heartbeat Detection. Procedia Computer Science 187, 329–334 (2021). https://doi.org/10.1016/j.procs.2021.04.070
Acknowledgements
This work was supported in part by the National Natural Science Foundation of China under Grant 62172340, in part by the Natural Science Foundation of Chongqing under Grant cstc2021jcyj-msxmX0041, in part by the Young and Middle-aged Senior Medical Talents Studio of Chongqing under grant ZQNYXGDRCGZS2021002, and in part by the Introduced Talent Program of Southwest University under Grant SWU020008. The corresponding author is Yuan Zhang.
Author information
Authors and Affiliations
Corresponding author
Ethics declarations
Conflict of interest
The authors declare that they have no conflict of interest.
Additional information
Publisher’s note
Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.
Liqiang Zhu and Changming Wang contributed equally to this work.
This article belongs to the Topical Collection: Special Issue on Resource Management at the Edge for Future Web, Mobile and IoT Applications
Guest Editors: Qiang He, Fang Dong, Chenshu Wu, and Yun Yang
Rights and permissions
About this article
Cite this article
Zhu, L., Wang, C., He, Z. et al. A lightweight automatic sleep staging method for children using single-channel EEG based on edge artificial intelligence. World Wide Web 25, 1883–1903 (2022). https://doi.org/10.1007/s11280-021-00983-3
Received:
Revised:
Accepted:
Published:
Version of record:
Issue date:
DOI: https://doi.org/10.1007/s11280-021-00983-3