The Analysis and Classify of Sleep Stage Using Deep Learning Network from Single-Channel EEG Signal
Electroencephalogram (EEG)-based sleep stage analysis is helpful for diagnosis of sleep disorder. However, the accuracy of previous EEG-based method is still unsatisfactory. In order to improve the classification performance, we proposed an EEG-based automatic sleep stage classification method, which combined convolutional neural network (CNN) and time-frequency decomposition. The time-frequency image (TFI) of EEG signals is obtained by using the smoothed short-time Fourier transform. The features derived from the TFI have been used as an input feature of a CNN for sleep stage classification. The proposed method achieves the best accuracy of 88.83%. The experimental results demonstrate that deep learning method provides better classification performance compared to other methods.
KeywordsConvolutional neural networks (CNN) Time-frequency decomposition Sleep analysis
This work was supported in part by National Natural Science Foundation of China (61273250), the Fundamental Research Funds for the Central Universities (No. 3102017jc11002) and the graduate starting seed fund of Northwestern Polytechnical University (Z2017141).
- 2.Rechtschaffen, A.Q., Kales, A.A.: A manual of standardized terminology techniques and scoring system for sleep stages in human subjects. Psychiatry Clin. Neurosci. 26(6), 644 (1968)Google Scholar
- 3.Schulz, H.: Phasic or transient? Comment on the terminology of the AASM manual for the scoring of sleep and associated events. J. Clin. Sleep Med. JCSM Official Publ. Am. Acad. Sleep Med. 3(7), 752 (2007)Google Scholar
- 5.Herrera, L.J., Mora, A.M., Fernandes, C., Migotina, D., Guillen, A., Rosa, A.C.: Symbolic representation of the EEG for sleep stage classification. In: International Conference on Intelligent Systems Design and Applications 2011, vol. 60, pp. 253–258. IEEE, Cordoba, Spain (2011)Google Scholar
- 8.Hassan, A.R., Bashar, S.K., Bhuiyan, M.I.H.: Automatic classification of sleep stages from single-channel electroencephalogram. In: India Conference 2015, pp. 1–6. IEEE, New Delhi, India (2015)Google Scholar
- 15.Kang, L., Ye, P., Li, Y., Doermann, D.: Convolutional neural networks for no-reference image quality assessment. In: Conference on Computer Vision and Pattern Recognition 2014, pp. 1733–1740. IEEE, Columbus, OH, USA (2014)Google Scholar
- 16.Lajnef, T., Chaibi, S., Ruby, P., Aguera, P.E., Eichenlaub, J.B., Samet, M., Kachouri, A., Jerbi, K.: Learning machines and sleeping brains: automatic sleep stage classification using decision-tree multi-class support vector machines. J. Neurosci. Methods 250(30), 94–105 (2015)CrossRefGoogle Scholar