An End-to-End Sleep Staging Simulator Based on Mixed Deep Neural Networks

Sleep screening is not only a major tool in the assessment of pathophysiology, but also a bridge between the central neuronal systems and behaviour/cognition. Automatic sleep staging is an alternative for the time-consuming gold standard manual scoring procedure. Most of the existing works designed such procedure by using deep neural networks without considering the medical criterion of the sleep staging task. We argue that capturing the stage-specific features which meet the criterion is of significant importance for the automatic sleep staging alternative. In this work we propose an end-to-end sleep staging simulator based on mixed neural networks, i.e., CNN, LSTM, and Transformer. The framework consists of two subnetworks: stage dependent feature mapping network which is constructed by the idea of physiological sleep nature, and an attention-based parallel staging network. Moreover, we adopt a mixed precision training strategy to quantize the model for exploring feasible usage in the clinical settings. Through an experiment with a large EEG database (Sleep Heart Health Study), the proposed method has a competitive stage scoring performance, especially in stages Wake, N2, and N3, with higher precision of 0.92, 0.85, and 0.86, respectively. Our study proves that the quantized model has potential capability for further application in the clinical staging task.