Automatic Sleep Staging based on Curriculum Learning Approach

Automatic sleep staging is helpful to improve diagnosis efficiency of sleep-related diseases. This work introduces the many-to-many formulation for automatic sleep staging, which means using a many-to-many mapping to convert the contextual input to the corresponding contextual output. We use convolutional neural networks (CNNs) to perform the many-to-many mapping, and use multilayer perceptron (MLP) to merge the contextual output into the final prediction for a particular epoch. In order to avoid the influence of unobvious characteristic waves and wrong labels on the training process, this work leverages the technology of curriculum learning. By clustering algorithm based on local density, the training set is divided into several subsets according to the signal quality. We design a learning strategy by successively leveraging these subsets. To the best of our current knowledge, this is the first work using curriculum learning for automatic sleep staging. It is showed by experiments that our scheme yields an accuracy comparable to the state-of-the-art on the public dataset Sleep-EDF.

[1]  Mohammed Imamul Hassan Bhuiyan,et al.  Computer-aided sleep staging using Complete Ensemble Empirical Mode Decomposition with Adaptive Noise and bootstrap aggregating , 2016, Biomed. Signal Process. Control..

[2]  Mohammed Imamul Hassan Bhuiyan,et al.  Automatic sleep scoring using statistical features in the EMD domain and ensemble methods , 2016 .

[3]  Sebastian Thrun,et al.  Dermatologist-level classification of skin cancer with deep neural networks , 2017, Nature.

[4]  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.

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

[6]  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.

[7]  Jason Weston,et al.  Curriculum learning , 2009, ICML '09.

[8]  J. Siegel.  Clues to the functions of mammalian sleep , 2005, Nature.

[9]  Sheng-Fu Liang,et al.  Automatic Stage Scoring of Single-Channel Sleep EEG by Using Multiscale Entropy and Autoregressive Models , 2012, IEEE Transactions on Instrumentation and Measurement.

[10]  Mohammed Imamul Hassan Bhuiyan,et al.  Automated identification of sleep states from EEG signals by means of ensemble empirical mode decomposition and random under sampling boosting , 2017, Comput. Methods Programs Biomed..

[11]  Oliver Y. Chén,et al.  Joint Classification and Prediction CNN Framework for Automatic Sleep Stage Classification , 2018, IEEE Transactions on Biomedical Engineering.

[12]  R. M. Siegel,et al.  Pattern recognition of the electroencephalogram by artificial neural networks. , 1993, Electroencephalography and clinical neurophysiology.

[13]  Hao Dong,et al.  Mixed Neural Network Approach for Temporal Sleep Stage Classification , 2016, IEEE Transactions on Neural Systems and Rehabilitation Engineering.

[14]  Laurent Vercueil,et al.  A convolutional neural network for sleep stage scoring from raw single-channel EEG , 2018, Biomed. Signal Process. Control..

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

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

[17]  Jimeng Sun,et al.  Multi-layer Representation Learning for Medical Concepts , 2016, KDD.

[18]  Natheer Khasawneh,et al.  Automated sleep stage identification system based on time-frequency analysis of a single EEG channel and random forest classifier , 2012, Comput. Methods Programs Biomed..

[19]  E. Wolpert.  A Manual of Standardized Terminology, Techniques and Scoring System for Sleep Stages of Human Subjects. , 1969 .

[20]  Jimeng Sun,et al.  RAIM: Recurrent Attentive and Intensive Model of Multimodal Patient Monitoring Data , 2018, KDD.

[21]  J. Allan Hobson,et al.  A manual of standardized terminology, techniques and scoring system for sleep stages of human subjects: A. Rechtschaffen and A. Kales (Editors). (Public Health Service, U.S. Government Printing Office, Washington, D.C., 1968, 58 p., $4.00) , 1969 .

[22]  A. Chesson,et al.  The AASM Manual for the Scoring of Sleep and Associated Events: Rules, Terminology, and Techinical Specifications , 2007 .

[23]  A. Chesson,et al.  The American Academy of Sleep Medicine (AASM) Manual for the Scoring of Sleep and Associated Events: Rules, Terminology and Technical Specifications , 2007 .

[24]  Kaare B. Mikkelsen,et al.  Personalizing deep learning models for automatic sleep staging , 2018, 1801.02645.

[25]  Yan Li,et al.  Analysis and Classification of Sleep Stages Based on Difference Visibility Graphs From a Single-Channel EEG Signal , 2014, IEEE Journal of Biomedical and Health Informatics.

[26]  Gunnar Rätsch,et al.  Real-valued (Medical) Time Series Generation with Recurrent Conditional GANs , 2017, ArXiv.

[27]  A. Hassan,et al.  A decision support system for automatic sleep staging from EEG signals using tunable Q-factor wavelet transform and spectral features , 2016, Journal of Neuroscience Methods.