Ebrahim Khalili1, Babak Mohammadzadeh Asl2. 1. Department of Biomedical Engineering, Tarbiat Modares University, Tehran, Iran. 2. Department of Biomedical Engineering, Tarbiat Modares University, Tehran, Iran. Electronic address: babakmasl@modares.ac.ir.
Abstract
BACKGROUND AND OBJECTIVE: This paper presents a new framework for automatic classification of sleep stages using a deep learning algorithm from single-channel EEG signals. Each segmented EEG signal appended with its label of stages is fed into a deep learning model to create an automatic sleep stage classification. This is one of the most important problems that is critical to the realization of monitoring patients with sleep disorder. METHODS: In the present study, a neural network architecture is introduced utilizing Convolutional Neural Networks (CNNs) to extract features, followed by Temporal Convolutional Neural Network to extract the temporal features from the extracted features vector of CNN. Finally, the performance of our model is improved by a Conditional Random Field layer. We also employed a new data augmentation technique to enhance the CNNs training which has auxiliary effects. RESULTS: We evaluated our model by two different single-channel EEG signals (i.e., Fpz-Cz and Pz-Oz EEG channels) from two public sleep datasets, named Sleep-EDF-2013 and Sleep-EDF-2018. The evaluation results on both datasets showed that our model obtains the best total accuracy and kappa score (EDF-2013: 85.39%- 0.80, EDF-2018: 82.46%- 0.76) compared to the state-of-the-art methods. CONCLUSIONS: This study will possibly allow us to have a wearable sleep monitoring system with a single-channel EEG. Also, unlike hand-crafted features methods, our model finds its own patterns through training epochs, and therefore, it may minimize engineering bias.
BACKGROUND AND OBJECTIVE: This paper presents a new framework for automatic classification of sleep stages using a deep learning algorithm from single-channel EEG signals. Each segmented EEG signal appended with its label of stages is fed into a deep learning model to create an automatic sleep stage classification. This is one of the most important problems that is critical to the realization of monitoring patients with sleep disorder. METHODS: In the present study, a neural network architecture is introduced utilizing Convolutional Neural Networks (CNNs) to extract features, followed by Temporal Convolutional Neural Network to extract the temporal features from the extracted features vector of CNN. Finally, the performance of our model is improved by a Conditional Random Field layer. We also employed a new data augmentation technique to enhance the CNNs training which has auxiliary effects. RESULTS: We evaluated our model by two different single-channel EEG signals (i.e., Fpz-Cz and Pz-Oz EEG channels) from two public sleep datasets, named Sleep-EDF-2013 and Sleep-EDF-2018. The evaluation results on both datasets showed that our model obtains the best total accuracy and kappa score (EDF-2013: 85.39%- 0.80, EDF-2018: 82.46%- 0.76) compared to the state-of-the-art methods. CONCLUSIONS: This study will possibly allow us to have a wearable sleep monitoring system with a single-channel EEG. Also, unlike hand-crafted features methods, our model finds its own patterns through training epochs, and therefore, it may minimize engineering bias.
Authors: Chengfan Li; Yueyu Qi; Xuehai Ding; Junjuan Zhao; Tian Sang; Matthew Lee Journal: Int J Environ Res Public Health Date: 2022-05-23 Impact factor: 4.614