TY - JOUR
T1 - Deep convolutional neural network for classification of sleep stages from single-channel EEG signals
AU - Mousavi, Z.
AU - Yousefi Rezaii, T.
AU - Sheykhivand, S.
AU - Farzamnia, A.
AU - Razavi, S. N.
N1 - Publisher Copyright:
© 2019
PY - 2019/8/1
Y1 - 2019/8/1
N2 - Using a smart method for automatic diagnosis in medical applications, such as sleep stage classification is considered as one of the important challenges of the last few years which can replace the time-consuming process of visual inspection done by specialists. One of the problems regarding the automatic diagnosis of sleep patterns is extraction and selection of discriminative features generally demanding high computational burden. This paper provides a new single-channel approach to automatic classification of sleep stages from EEG signal. The main idea is to directly apply the raw EEG signal to deep convolutional neural network, without involving feature extraction/selection, which is a challenging process in the previous literature. The proposed network architecture includes 9 convolutional layers followed by 2 fully connected layers. In order to make the samples of different classes balanced, we used a preprocessing method called data augmentation. The simulation results of the proposed method for classification of 2 to 6 classes of sleep stages show the accuracy of 98.10%, 96.86%, 93.11%, 92.95%, 93.55% and Cohen's Kappa coefficient of 0.98%, 0.94%, 0.90%, 0.86% and 0.89%, respectively. Furthermore, comparing the obtained results with the state-of-the-art methods reveals the performance improvement of the proposed sleep stage classification in terms of accuracy and Cohen's Kappa coefficient.
AB - Using a smart method for automatic diagnosis in medical applications, such as sleep stage classification is considered as one of the important challenges of the last few years which can replace the time-consuming process of visual inspection done by specialists. One of the problems regarding the automatic diagnosis of sleep patterns is extraction and selection of discriminative features generally demanding high computational burden. This paper provides a new single-channel approach to automatic classification of sleep stages from EEG signal. The main idea is to directly apply the raw EEG signal to deep convolutional neural network, without involving feature extraction/selection, which is a challenging process in the previous literature. The proposed network architecture includes 9 convolutional layers followed by 2 fully connected layers. In order to make the samples of different classes balanced, we used a preprocessing method called data augmentation. The simulation results of the proposed method for classification of 2 to 6 classes of sleep stages show the accuracy of 98.10%, 96.86%, 93.11%, 92.95%, 93.55% and Cohen's Kappa coefficient of 0.98%, 0.94%, 0.90%, 0.86% and 0.89%, respectively. Furthermore, comparing the obtained results with the state-of-the-art methods reveals the performance improvement of the proposed sleep stage classification in terms of accuracy and Cohen's Kappa coefficient.
KW - Classification
KW - Convolutional neural network
KW - Deep learning
KW - EEG
KW - Sleep stage analysis
UR - http://www.scopus.com/inward/record.url?scp=85067255197&partnerID=8YFLogxK
U2 - 10.1016/j.jneumeth.2019.108312
DO - 10.1016/j.jneumeth.2019.108312
M3 - Article
C2 - 31201824
AN - SCOPUS:85067255197
VL - 324
JO - Journal of Neuroscience Methods
JF - Journal of Neuroscience Methods
SN - 0165-0270
M1 - 108312
ER -