International Journal of Fuzzy Logic and Intelligent Systems 2020; 20(3): 181-187
Published online September 25, 2020
https://doi.org/10.5391/IJFIS.2020.20.3.181
© The Korean Institute of Intelligent Systems
Urtnasan Erdenebayar1* , Yeewoong Kim1*
, Joung-Uk Park1
, SooYong Lee2, and Kyoung-Joung Lee1
1Department of Biomedical Engineering, College of Health Science, Yonsei University, Wonju, Korea
2Department of Liberal Education, Yonsei University, Wonju, Korea
Correspondence to :
Kyoung-Joung Lee (lkj5809@yonsei.ac.kr)
*These authors contributed equally to this work.
This is an Open Access article distributed under the terms of the Creative Commons Attribution Non-Commercial License (http://creativecommons.org/licenses/by-nc/3.0/) which permits unrestricted noncommercial use, distribution, and reproduction in any medium, provided the original work is properly cited.
A healthy sleep structure is clinically very important for overall health. The sleep structure can be represented by the percentage of different sleep stages during the total sleep time. In this study, we proposed a method for automatic classification of sleep stages from an electrocardiogram (ECG) signal using a gated-recurrent unit (GRU). The proposed method performed multiclass classification for three-class sleep stages such as awake, light, and deep sleep. A deep structured GRU was used in the proposed method, which is a common recurrent neural network. The proposed deep learning (SleepGRU) model consists of a 5-layer GRU and is optimized by batch-normalization, dropout, and Adam update rules. The ECG signal was recorded during nocturnal polysomnography from 112 subjects, and was normalized and segmented into units of 30-second duration. To train and evaluate the proposed method, the training set consisted of 80,316 segments from 89 subjects, and the test set used 20,079 segments from 23 subjects. We achieved good performances with an overall accuracy of 80.43% and F1-score of 80.07% for the test set. The proposed method can be an alternative and useful tool for sleep monitoring and sleep screening, which have previously been manually evaluated by a sleep technician or sleep expert.
Keywords: Sleep stage classification, Gated-recurrent unit, Deep learning, Electrocardiogram
On average, humans spend one-third of their lives sleeping. Sleep has a complex structure and cyclic rhythm that can define its quality and efficiency. The phases of sleep consist of wake, stages 1–4, and rapid eye movement (REM) sleep. Stages 3 and 4 constitute the slow-wave sleep (SWS) also known as delta sleep, and Stages 1–4 combine to form the non-rapid eye movement (NREM) stages [1]. During sleep, the brain organizes learned contents, eliminates toxins, and recharges energy [2]. Sleep disorders have a number of negative effects such as daytime sleepiness [3], headaches [4], cardiovascular diseases [5], decreased cognitive function [6], and decreased immunity [7]. The prevalence of sleep disorders, such as insomnia, sleep fragmentation, and sleep apnea, is increasing. Therefore, it is necessary to diagnose these problems properly through systematic sleep analysis [8].
Polysomnography (PSG) is the gold standard diagnostic test for sleep structure and sleep fragmentation. The patients attach equipment to their bodies to measure various bio-signals, such as electroencephalogram (EEG), electrooculogram (EOG), electrocardiogram (ECG), and electromyogram (EMG). They then sleep at the sleep center. Based on the bio-signals obtained from PSG, not only can sleep fragmentation and structure be identified, but sleep disorders can also be diagnosed objectively. Among the electrophysiological measurements, EEG and EOG can be used to assess sleep stages.
ECG is an alternative bio-signal for automatic identification of sleep stages in home healthcare. Recently, various studies have proposed different methods for automatic classification of sleep stages using ECG signals. Initially, these studies extracted intermediate vital signs such as heart rate (HR), beat-to-beat (RR) interval, and heart rate variability (HRV) from the raw ECG signal. Adnane et al. [9] proposed a method that can classify sleep and wake, and also calculate sleep efficiency using a detrended fluctuation analysis of the HRV. Xiao et al. [10] proposed an alternative method for sleep stage classification based on a random forest using the HRV signal. Singh et al. [11] investigated a method for differentiating REM from NREM sleep using the RR interval. Yucelbas et al. [12] identified wake, REM, and NREM sleep based on nonlinear and morphological feature sets extracted from ECG signals. However, in all these studies, it was necessary to obtain intermediate vital signs from the ECG signal, including HR, HRV, and RR interval, and extract a number of features by using high-dimensional domain transformation. Furthermore, it was also necessary to select high-discriminative features to reduce the number of features for classifier training.
In this study, we proposed a novel method for automatic classification of sleep stages from a single-lead ECG signal using a deep gated-recurrent unit. We called the proposed method a SleepGRU model because the constructed model was designed and optimized by a gated-recurrent unit (GRU) to analyze the complex structure and cyclic rhythm of human sleep. The single-lead ECG was used without extracting any of the abovementioned intermediate vital signs required in similar studies, or using any hand-crafted features. The SleepRGU model was constructed, trained, and evaluated using clinical PSG datasets obtained from normal subjects and sleep apnea patients.
For this study, 112 participants were enrolled to obtain the ECG dataset for automatic classification of sleep stages. The participants consisted of 52 control (Apnea-Hypopnea Index [AHI],: 2.3 ± 2.2) and 60 apnea (AHI, 17.5 ± 6.8) subjects. Subjects with severe sleep apnea were excluded from this study. The ECG dataset comprised the nocturnal PSG recordings of all participants. All the PSG recordings were obtained using a polygraphic amplifier (Model N7000; Embla, Reykjavik, Iceland) over an average period of 7.4 hour at the Samsung Medical Center, Seoul, Korea (Table 1). The ECG signals were recorded at a sampling rate of 200 Hz and segmented into episodes of 30 seconds. The sleep stages were labeled according to the criteria of the American Association of Sleep Medicine (AASM) [13]. All PSG procedures were approved by the Institutional Review Board of Samsung Medical Center (No. 2012-01-063).
The ECG signal of the subjects in each group was collected using a single-lead transducer and resampled at 100 Hz, resulting in 3,000 samples per episode. A total of 100,395 episodes were obtained after combining the signals obtained from all participants. The training and test datasets for the constructed SleepGRU model were randomly selected from each subject group (Table 1). The training set consisted of 80,316 episodes from 89 subjects (control 42, apnea 47), and the test set comprised 20,079 episodes from 23 subjects (control 10, apnea 13).
We constructed a SleepGRU model taking the characteristics of human sleep, such as complex structure and cyclic rhythm, into consideration. Thus, the SleepGRU model consists of 5-layer GRU. The GRU is a very robust gating mechanism used in recurrent neural networks [14]. For the optimization of the SleepGRU model, batch normalization [15], dropout [16], and rectified linear unit (ReLU) [17] were selected following trials of multiple optimization methods.
The GRU was introduced by Chung et al. [14] as a more efficient mechanism than the long short-term memory model. A GRU only has two gates: an update gate
where
Batch normalization was applied to the input ECG signal before training the constructed SleepGRU model, as shown in
where
Dropout is a technique that refers to randomly eliminating nodes to reduce overfitting in the network model by preventing complex adaptations on the training data [16].
A ReLU was used as the activation function of each layer of the SleepGRU model. It can be represented as follows:
where
Table 2 shows the detailed characteristics of the final architecture of the constructed SleepGRU model. The SleepGRU model is a structure consisting of a 5-layer GRU. Each recurrent layer contains (24, 20, 16, 8, 4) hidden nodes in the GRU. Finally, we used a fully connected multilayer perceptron with softmax activation for the final discrimination of the sleep stage classification from the ECG datasets.
For this study, the PSG data were processed using MATLAB (R2018b). The SleepGRU model was constructed using the Keras library with a TensorFlow backend [18]. A workstation with an Intel CPU (i9-9900X @3.5GHz) and NVIDIA GPU (GeForce RTX 2080 Ti) was used for deep learning.
The F1-score was used to evaluate the constructed SleepGRU model, which evaluates the classification accuracy of each class according to class equality. To obtain the F1-score, two evaluation measures, precision and recall, were combined. These are defined as follows:
where
The results of the SleepGRU model for the automatic classification of the sleep stage based on the single-lead ECG signal are shown in Table 3. The SleepGRU model was evaluated based on precision, recall, F1-score, and accuracy for multiclass sleep stage classification comprising wake, REM, and NREM. For this study, sleep stages N1, N2, and N3 were integrated into the NREM stage.
For the automatic classification of sleep stages (Table 3), the SleepGRU model demonstrated robust performance with an accuracy of 84.01% for the training set and 80.43% for the test set. In particular, the SleepGRU model was most effective in the NREM stage and least effective in the wake stage. Although the REM stage had the largest number of events, the SleepGRU model did not demonstrate remarkable accuracy for this stage.
In this study, a SleepGRU model was constructed for the automatic classification of sleep stages using a single-lead ECG signal. To ensure that the model was well suited to the complex characteristics of human sleep, the GRU was integrated into the SleepGRU model. We obtained a robust performance with a classification accuracy of 80.43% for the three-class sleep stages using the single-lead ECG signal. In addition, the Sleep-GRU model was evaluated using the ECG dataset obtained from the control and sleep apnea groups.
Table 4 compares and analyzes the existing studies that perform sleep stage scoring using either the ECG signal or the features obtained from the HRV and RR intervals. Most of these studies used a shallow learning classifier as the SVM, rather than deep learning models; many of the features were extracted from the ECG signal. In addition, binary (sleep or wake) and multiclass (wake, REM, and NREM) classification was performed using extracted hand-crafted feature sets. First, intermediate vital signs were extracted, including the HR [9], HRV [10], and RR interval [11]. Then, a number of features were extracted by analyzing them in the time, frequency, and nonlinear domain [12]. Finally, the top features were selected to reduce the number of features for training the random forest classifier. However, in these studies, the results cannot be generalized, and nonlinear features are required for complex calculations.
Some studies have been based on the deep learning framework for automatic sleep stage scoring using the ECG signal [19–21]. In these studies, deep learning frameworks (DNN [19], CNN [20], RNN [20], and LSTM [21]) were used as feature extractors or classifiers in previous studies based on the ECG signal. In addition, they demonstrated less accuracy in the three-class sleep stage classification than the current study.
We therefore constructed a SleepGRU model that can perform automatic classification of the sleep stage based on sequential feature extraction and discrimination using the ECG signal. The constructed SleepGRU model outperforms conventional methods because it considers the complex and cyclic characteristics of sleep. Another advantage of the SleepGRU model for sleep stage scoring is that it does not require the intermediate biosignals (HRV, RR interval, and ECG derived respiration) or any hand-crafted feature sets extracted through domain transformation analysis. Furthermore, the SleepGRU model has a simpler structure than the other deep learning models designed for sleep stage scoring. In addition, it was trained and tested on a clinical dataset consisting of normal and sleep apnea groups. Our results demonstrated that the constructed SleepGRU model has the potential to accurately perform multiclass classification using only a single-lead ECG signal.
However, this study has some limitations. First, we used a small dataset for the model evaluation. Larger and more diverse datasets are needed. Second, we only covered sleep apnea patients in this study, but some other datasets from common sleep disorders should be covered for a more robust model. Finally, the constructed SleepGRU model has a higher computational cost than conventional methods. These are the challenges that should be addressed in future studies.
In this study, a SleepGRU model was constructed for the automatic classification of sleep stages based on a single-lead ECG signal. In most ECG-based studies on sleep stage scoring, classification was performed using a binary class, but the SleepGRU model can perform multiclass classification, as it demonstrates efficiency for the three-class sleep stages. In addition, the Sleep-GRU model can automatically extract the feature maps and classify the sleep stages simultaneously based on the ECG signal. We achieved an overall high accuracy of 80.43% for the three-class sleep stages (Wake, REM, and NREM). Therefore, the SleepGRU model is suitable for sleep stage classification using a single-lead ECG signal without any feature extraction. In future research, the SleepGRU model based on the single-lead ECG signal should be validated using larger and more diverse datasets.
No potential conflicts of interest relevant to this article were reported.
Table 1. Information of the participants.
Training set | Test set | Total | |
---|---|---|---|
Sex | 89 | 23 | 112 |
Male | 55 | 16 | 71 |
Female | 34 | 7 | 41 |
Age (yr) | 53.1±10.5 | 55.1±10.4 | 53.5±10.5 |
BMI (kg/m2) | 24.2±3.0 | 24.2±3.0 | 24.2±3.0 |
AHI (/hr) | 10.6±9.5 | 10.0±8.5 | 10.4±9.3 |
TST (hr) | 6.2±1.0 | 6.2±0.7 | 6.2±0.9 |
SE (%) | 84.8±13.1 | 86.0±8.7 | 85.1±12.3 |
BMI, body mass index; AHI, Apnea-Hypopnea Index; TST, total sleep time; SE, sleep efficiency..
Table 2. Detailed structure of the constructed SleepGRU model.
No | Layers | Units | Dropout | Feature maps | Params |
---|---|---|---|---|---|
1 | bNorm | - | - | 3000×1 | 4 |
2 | GRU1 | 24 | 0.5 | 3000×24 | 1,920 |
3 | GRU2 | 20 | 0.5 | 3000×20 | 2,760 |
4 | GRU3 | 16 | 0.4 | 3000×16 | 1,824 |
5 | GRU4 | 8 | 0.3 | 3000×8 | 642 |
6 | GRU5 | 4 | 0.2 | 3000×4 | 168 |
7 | MLP0 | 3 | - | 15×3 | 45 |
Table 3. The performance of the SleepGRU model.
Datasets | Classes | Precision | Recall | F1 | Accuracy (%) |
---|---|---|---|---|---|
Training set | Wake | 0.86 | 0.68 | 0.76 | 84.01 |
REM | 0.88 | 0.67 | 0.77 | ||
NREM | 0.86 | 0.90 | 0.89 | ||
Test set | Wake | 0.58 | 0.42 | 0.51 | 80.43 |
REM | 0.77 | 0.81 | 0.73 | ||
NREM | 0.86 | 0.84 | 0.83 |
International Journal of Fuzzy Logic and Intelligent Systems 2020; 20(3): 181-187
Published online September 25, 2020 https://doi.org/10.5391/IJFIS.2020.20.3.181
Copyright © The Korean Institute of Intelligent Systems.
Urtnasan Erdenebayar1* , Yeewoong Kim1*
, Joung-Uk Park1
, SooYong Lee2, and Kyoung-Joung Lee1
1Department of Biomedical Engineering, College of Health Science, Yonsei University, Wonju, Korea
2Department of Liberal Education, Yonsei University, Wonju, Korea
Correspondence to:Kyoung-Joung Lee (lkj5809@yonsei.ac.kr)
*These authors contributed equally to this work.
This is an Open Access article distributed under the terms of the Creative Commons Attribution Non-Commercial License (http://creativecommons.org/licenses/by-nc/3.0/) which permits unrestricted noncommercial use, distribution, and reproduction in any medium, provided the original work is properly cited.
A healthy sleep structure is clinically very important for overall health. The sleep structure can be represented by the percentage of different sleep stages during the total sleep time. In this study, we proposed a method for automatic classification of sleep stages from an electrocardiogram (ECG) signal using a gated-recurrent unit (GRU). The proposed method performed multiclass classification for three-class sleep stages such as awake, light, and deep sleep. A deep structured GRU was used in the proposed method, which is a common recurrent neural network. The proposed deep learning (SleepGRU) model consists of a 5-layer GRU and is optimized by batch-normalization, dropout, and Adam update rules. The ECG signal was recorded during nocturnal polysomnography from 112 subjects, and was normalized and segmented into units of 30-second duration. To train and evaluate the proposed method, the training set consisted of 80,316 segments from 89 subjects, and the test set used 20,079 segments from 23 subjects. We achieved good performances with an overall accuracy of 80.43% and F1-score of 80.07% for the test set. The proposed method can be an alternative and useful tool for sleep monitoring and sleep screening, which have previously been manually evaluated by a sleep technician or sleep expert.
Keywords: Sleep stage classification, Gated-recurrent unit, Deep learning, Electrocardiogram
On average, humans spend one-third of their lives sleeping. Sleep has a complex structure and cyclic rhythm that can define its quality and efficiency. The phases of sleep consist of wake, stages 1–4, and rapid eye movement (REM) sleep. Stages 3 and 4 constitute the slow-wave sleep (SWS) also known as delta sleep, and Stages 1–4 combine to form the non-rapid eye movement (NREM) stages [1]. During sleep, the brain organizes learned contents, eliminates toxins, and recharges energy [2]. Sleep disorders have a number of negative effects such as daytime sleepiness [3], headaches [4], cardiovascular diseases [5], decreased cognitive function [6], and decreased immunity [7]. The prevalence of sleep disorders, such as insomnia, sleep fragmentation, and sleep apnea, is increasing. Therefore, it is necessary to diagnose these problems properly through systematic sleep analysis [8].
Polysomnography (PSG) is the gold standard diagnostic test for sleep structure and sleep fragmentation. The patients attach equipment to their bodies to measure various bio-signals, such as electroencephalogram (EEG), electrooculogram (EOG), electrocardiogram (ECG), and electromyogram (EMG). They then sleep at the sleep center. Based on the bio-signals obtained from PSG, not only can sleep fragmentation and structure be identified, but sleep disorders can also be diagnosed objectively. Among the electrophysiological measurements, EEG and EOG can be used to assess sleep stages.
ECG is an alternative bio-signal for automatic identification of sleep stages in home healthcare. Recently, various studies have proposed different methods for automatic classification of sleep stages using ECG signals. Initially, these studies extracted intermediate vital signs such as heart rate (HR), beat-to-beat (RR) interval, and heart rate variability (HRV) from the raw ECG signal. Adnane et al. [9] proposed a method that can classify sleep and wake, and also calculate sleep efficiency using a detrended fluctuation analysis of the HRV. Xiao et al. [10] proposed an alternative method for sleep stage classification based on a random forest using the HRV signal. Singh et al. [11] investigated a method for differentiating REM from NREM sleep using the RR interval. Yucelbas et al. [12] identified wake, REM, and NREM sleep based on nonlinear and morphological feature sets extracted from ECG signals. However, in all these studies, it was necessary to obtain intermediate vital signs from the ECG signal, including HR, HRV, and RR interval, and extract a number of features by using high-dimensional domain transformation. Furthermore, it was also necessary to select high-discriminative features to reduce the number of features for classifier training.
In this study, we proposed a novel method for automatic classification of sleep stages from a single-lead ECG signal using a deep gated-recurrent unit. We called the proposed method a SleepGRU model because the constructed model was designed and optimized by a gated-recurrent unit (GRU) to analyze the complex structure and cyclic rhythm of human sleep. The single-lead ECG was used without extracting any of the abovementioned intermediate vital signs required in similar studies, or using any hand-crafted features. The SleepRGU model was constructed, trained, and evaluated using clinical PSG datasets obtained from normal subjects and sleep apnea patients.
For this study, 112 participants were enrolled to obtain the ECG dataset for automatic classification of sleep stages. The participants consisted of 52 control (Apnea-Hypopnea Index [AHI],: 2.3 ± 2.2) and 60 apnea (AHI, 17.5 ± 6.8) subjects. Subjects with severe sleep apnea were excluded from this study. The ECG dataset comprised the nocturnal PSG recordings of all participants. All the PSG recordings were obtained using a polygraphic amplifier (Model N7000; Embla, Reykjavik, Iceland) over an average period of 7.4 hour at the Samsung Medical Center, Seoul, Korea (Table 1). The ECG signals were recorded at a sampling rate of 200 Hz and segmented into episodes of 30 seconds. The sleep stages were labeled according to the criteria of the American Association of Sleep Medicine (AASM) [13]. All PSG procedures were approved by the Institutional Review Board of Samsung Medical Center (No. 2012-01-063).
The ECG signal of the subjects in each group was collected using a single-lead transducer and resampled at 100 Hz, resulting in 3,000 samples per episode. A total of 100,395 episodes were obtained after combining the signals obtained from all participants. The training and test datasets for the constructed SleepGRU model were randomly selected from each subject group (Table 1). The training set consisted of 80,316 episodes from 89 subjects (control 42, apnea 47), and the test set comprised 20,079 episodes from 23 subjects (control 10, apnea 13).
We constructed a SleepGRU model taking the characteristics of human sleep, such as complex structure and cyclic rhythm, into consideration. Thus, the SleepGRU model consists of 5-layer GRU. The GRU is a very robust gating mechanism used in recurrent neural networks [14]. For the optimization of the SleepGRU model, batch normalization [15], dropout [16], and rectified linear unit (ReLU) [17] were selected following trials of multiple optimization methods.
The GRU was introduced by Chung et al. [14] as a more efficient mechanism than the long short-term memory model. A GRU only has two gates: an update gate
where
Batch normalization was applied to the input ECG signal before training the constructed SleepGRU model, as shown in
where
Dropout is a technique that refers to randomly eliminating nodes to reduce overfitting in the network model by preventing complex adaptations on the training data [16].
A ReLU was used as the activation function of each layer of the SleepGRU model. It can be represented as follows:
where
Table 2 shows the detailed characteristics of the final architecture of the constructed SleepGRU model. The SleepGRU model is a structure consisting of a 5-layer GRU. Each recurrent layer contains (24, 20, 16, 8, 4) hidden nodes in the GRU. Finally, we used a fully connected multilayer perceptron with softmax activation for the final discrimination of the sleep stage classification from the ECG datasets.
For this study, the PSG data were processed using MATLAB (R2018b). The SleepGRU model was constructed using the Keras library with a TensorFlow backend [18]. A workstation with an Intel CPU (i9-9900X @3.5GHz) and NVIDIA GPU (GeForce RTX 2080 Ti) was used for deep learning.
The F1-score was used to evaluate the constructed SleepGRU model, which evaluates the classification accuracy of each class according to class equality. To obtain the F1-score, two evaluation measures, precision and recall, were combined. These are defined as follows:
where
The results of the SleepGRU model for the automatic classification of the sleep stage based on the single-lead ECG signal are shown in Table 3. The SleepGRU model was evaluated based on precision, recall, F1-score, and accuracy for multiclass sleep stage classification comprising wake, REM, and NREM. For this study, sleep stages N1, N2, and N3 were integrated into the NREM stage.
For the automatic classification of sleep stages (Table 3), the SleepGRU model demonstrated robust performance with an accuracy of 84.01% for the training set and 80.43% for the test set. In particular, the SleepGRU model was most effective in the NREM stage and least effective in the wake stage. Although the REM stage had the largest number of events, the SleepGRU model did not demonstrate remarkable accuracy for this stage.
In this study, a SleepGRU model was constructed for the automatic classification of sleep stages using a single-lead ECG signal. To ensure that the model was well suited to the complex characteristics of human sleep, the GRU was integrated into the SleepGRU model. We obtained a robust performance with a classification accuracy of 80.43% for the three-class sleep stages using the single-lead ECG signal. In addition, the Sleep-GRU model was evaluated using the ECG dataset obtained from the control and sleep apnea groups.
Table 4 compares and analyzes the existing studies that perform sleep stage scoring using either the ECG signal or the features obtained from the HRV and RR intervals. Most of these studies used a shallow learning classifier as the SVM, rather than deep learning models; many of the features were extracted from the ECG signal. In addition, binary (sleep or wake) and multiclass (wake, REM, and NREM) classification was performed using extracted hand-crafted feature sets. First, intermediate vital signs were extracted, including the HR [9], HRV [10], and RR interval [11]. Then, a number of features were extracted by analyzing them in the time, frequency, and nonlinear domain [12]. Finally, the top features were selected to reduce the number of features for training the random forest classifier. However, in these studies, the results cannot be generalized, and nonlinear features are required for complex calculations.
Some studies have been based on the deep learning framework for automatic sleep stage scoring using the ECG signal [19–21]. In these studies, deep learning frameworks (DNN [19], CNN [20], RNN [20], and LSTM [21]) were used as feature extractors or classifiers in previous studies based on the ECG signal. In addition, they demonstrated less accuracy in the three-class sleep stage classification than the current study.
We therefore constructed a SleepGRU model that can perform automatic classification of the sleep stage based on sequential feature extraction and discrimination using the ECG signal. The constructed SleepGRU model outperforms conventional methods because it considers the complex and cyclic characteristics of sleep. Another advantage of the SleepGRU model for sleep stage scoring is that it does not require the intermediate biosignals (HRV, RR interval, and ECG derived respiration) or any hand-crafted feature sets extracted through domain transformation analysis. Furthermore, the SleepGRU model has a simpler structure than the other deep learning models designed for sleep stage scoring. In addition, it was trained and tested on a clinical dataset consisting of normal and sleep apnea groups. Our results demonstrated that the constructed SleepGRU model has the potential to accurately perform multiclass classification using only a single-lead ECG signal.
However, this study has some limitations. First, we used a small dataset for the model evaluation. Larger and more diverse datasets are needed. Second, we only covered sleep apnea patients in this study, but some other datasets from common sleep disorders should be covered for a more robust model. Finally, the constructed SleepGRU model has a higher computational cost than conventional methods. These are the challenges that should be addressed in future studies.
In this study, a SleepGRU model was constructed for the automatic classification of sleep stages based on a single-lead ECG signal. In most ECG-based studies on sleep stage scoring, classification was performed using a binary class, but the SleepGRU model can perform multiclass classification, as it demonstrates efficiency for the three-class sleep stages. In addition, the Sleep-GRU model can automatically extract the feature maps and classify the sleep stages simultaneously based on the ECG signal. We achieved an overall high accuracy of 80.43% for the three-class sleep stages (Wake, REM, and NREM). Therefore, the SleepGRU model is suitable for sleep stage classification using a single-lead ECG signal without any feature extraction. In future research, the SleepGRU model based on the single-lead ECG signal should be validated using larger and more diverse datasets.
Table 1 . Information of the participants.
Training set | Test set | Total | |
---|---|---|---|
Sex | 89 | 23 | 112 |
Male | 55 | 16 | 71 |
Female | 34 | 7 | 41 |
Age (yr) | 53.1±10.5 | 55.1±10.4 | 53.5±10.5 |
BMI (kg/m2) | 24.2±3.0 | 24.2±3.0 | 24.2±3.0 |
AHI (/hr) | 10.6±9.5 | 10.0±8.5 | 10.4±9.3 |
TST (hr) | 6.2±1.0 | 6.2±0.7 | 6.2±0.9 |
SE (%) | 84.8±13.1 | 86.0±8.7 | 85.1±12.3 |
BMI, body mass index; AHI, Apnea-Hypopnea Index; TST, total sleep time; SE, sleep efficiency..
Table 2 . Detailed structure of the constructed SleepGRU model.
No | Layers | Units | Dropout | Feature maps | Params |
---|---|---|---|---|---|
1 | bNorm | - | - | 3000×1 | 4 |
2 | GRU1 | 24 | 0.5 | 3000×24 | 1,920 |
3 | GRU2 | 20 | 0.5 | 3000×20 | 2,760 |
4 | GRU3 | 16 | 0.4 | 3000×16 | 1,824 |
5 | GRU4 | 8 | 0.3 | 3000×8 | 642 |
6 | GRU5 | 4 | 0.2 | 3000×4 | 168 |
7 | MLP0 | 3 | - | 15×3 | 45 |
Table 3 . The performance of the SleepGRU model.
Datasets | Classes | Precision | Recall | F1 | Accuracy (%) |
---|---|---|---|---|---|
Training set | Wake | 0.86 | 0.68 | 0.76 | 84.01 |
REM | 0.88 | 0.67 | 0.77 | ||
NREM | 0.86 | 0.90 | 0.89 | ||
Test set | Wake | 0.58 | 0.42 | 0.51 | 80.43 |
REM | 0.77 | 0.81 | 0.73 | ||
NREM | 0.86 | 0.84 | 0.83 |
Urtnasan Erdenebayar, Jong-Uk Park, SooYong Lee, Eun-Yeon Joo, and Kyoung-Joung Lee
International Journal of Fuzzy Logic and Intelligent Systems 2020; 20(2): 138-144 https://doi.org/10.5391/IJFIS.2020.20.2.138Chan Sik Han and Keon Myung Lee
International Journal of Fuzzy Logic and Intelligent Systems 2021; 21(4): 317-337 https://doi.org/10.5391/IJFIS.2021.21.4.317Nishant Chauhan and Byung-Jae Choi
International Journal of Fuzzy Logic and Intelligent Systems 2019; 19(4): 315-322 https://doi.org/10.5391/IJFIS.2019.19.4.315