Deep Identity Confusion for Automatic Sleep Staging Based on Single-Channel EEG

Sleep Staging (SS) is a vital step in sleep neurobiology. Though many previous approaches have been proposed to solve it, most of them suffer from poor generalization for unknown identity. In this paper, we proposed a deep identity confusion method to extract powerful task-specific and identity-invariant feature and then score sleep stages with non-linear machine learning model. With an unified CNN-LSTM structure employed for feature extraction, we implement identity confusion with an extra identity prediction branch and apply inversed gradients to frontal layers during back-propagation. Then the deep feature is used to train a XGBoost classifier. Experiments on Sleep-EDF benchmarks achieve classification accuracy and macro F1 score of 84.1% and 78.9%, and it suggests proposed method boost performance of origin deep learning base model and show competitive result comparing to state-of-the-art methods.

[1]  Panagiotis D. Bamidis,et al.  Automatic Sleep Stage Classification Applying Machine Learning Algorithms on EEG Recordings , 2017, 2017 IEEE 30th International Symposium on Computer-Based Medical Systems (CBMS).

[2]  Sergey Ioffe,et al.  Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shift , 2015, ICML.

[3]  Amy Loutfi,et al.  Sleep Stage Classification Using Unsupervised Feature Learning , 2012, Adv. Artif. Neural Syst..

[4]  Michael X Cohen,et al.  Analyzing Neural Time Series Data: Theory and Practice , 2014 .

[5]  Farhad Faradji,et al.  A Novel Multi-Class EEG-Based Sleep Stage Classification System , 2018, IEEE Transactions on Neural Systems and Rehabilitation Engineering.

[6]  Aeilko H. Zwinderman,et al.  Analysis of a sleep-dependent neuronal feedback loop: the slow-wave microcontinuity of the EEG , 2000, IEEE Transactions on Biomedical Engineering.

[7]  Tomas Mikolov,et al.  RNNLM - Recurrent Neural Network Language Modeling Toolkit , 2011 .

[8]  Stanislas Chambon,et al.  A Deep Learning Architecture for Temporal Sleep Stage Classification Using Multivariate and Multimodal Time Series , 2017, IEEE Transactions on Neural Systems and Rehabilitation Engineering.

[9]  Victor S. Lempitsky,et al.  Unsupervised Domain Adaptation by Backpropagation , 2014, ICML.

[10]  Rajeev Sharma,et al.  Automatic sleep stages classification based on iterative filtering of electroencephalogram signals , 2017, Neural Computing and Applications.

[11]  Yike Guo,et al.  Automatic Sleep Stage Scoring with Single-Channel EEG Using Convolutional Neural Networks , 2016, ArXiv.

[12]  Yike Guo,et al.  Automatic Sleep Stage Scoring Using Time-Frequency Analysis and Stacked Sparse Autoencoders , 2015, Annals of Biomedical Engineering.

[13]  Geoffrey E. Hinton,et al.  ImageNet classification with deep convolutional neural networks , 2012, Commun. ACM.

[14]  Abdulhamit Subasi,et al.  A decision support system for automated identification of sleep stages from single-channel EEG signals , 2017, Knowl. Based Syst..

[15]  Jeffrey M. Hausdorff,et al.  Physionet: Components of a New Research Resource for Complex Physiologic Signals". Circu-lation Vol , 2000 .

[16]  Chuang-Chien Chiu,et al.  Sleep Stages Recognition Based on Combined Artificial Neural Network and Fuzzy System Using Wavelet Transform Features , 2013 .

[17]  Francesco Rundo,et al.  A new quantitative automatic method for the measurement of non‐rapid eye movement sleep electroencephalographic amplitude variability , 2012, Journal of sleep research.

[18]  Chao Wu,et al.  DeepSleepNet: A Model for Automatic Sleep Stage Scoring Based on Raw Single-Channel EEG , 2017, IEEE Transactions on Neural Systems and Rehabilitation Engineering.

[19]  Paul Geladi,et al.  Principal Component Analysis , 1987, Comprehensive Chemometrics.

[20]  Tianqi Chen,et al.  XGBoost: A Scalable Tree Boosting System , 2016, KDD.

[21]  Geoffrey E. Hinton,et al.  Speech recognition with deep recurrent neural networks , 2013, 2013 IEEE International Conference on Acoustics, Speech and Signal Processing.

[22]  Xiaogang Wang,et al.  Deep Learning Face Representation from Predicting 10,000 Classes , 2014, 2014 IEEE Conference on Computer Vision and Pattern Recognition.

[23]  B. Weiss,et al.  Spatio-temporal analysis of monofractal and multifractal properties of the human sleep EEG , 2009, Journal of Neuroscience Methods.

[24]  Yu-Liang Hsu,et al.  Automatic sleep stage recurrent neural classifier using energy features of EEG signals , 2013, Neurocomputing.

[25]  Yuanqing Li,et al.  An EEG-based brain-computer interface for automatic sleep stage classification , 2018, 2018 13th IEEE Conference on Industrial Electronics and Applications (ICIEA).