A Developed LSTM-Ladder-Network-Based Model for Sleep Stage Classification

Abstract

Sleep staging is crucial for diagnosing sleep-related disorders. The heavy and time-consuming task of manual staging can be released by automatic techniques. However, the automatic staging model would have a relatively poor performance when working on unseen new data due to individual differences. In this research, a developed LSTM-Ladder-Network (LLN) model is proposed for automatic sleep stage classification. Several features are extracted for each epoch and combined with the following epochs to form a cross-epoch vector. The long short-term memory (LSTM) network is added into the basic ladder network (LN) to learn the sequential information of adjacent epochs. The developed model is implemented based on a transductive learning scheme to avoid the issue of accuracy loss caused by individual differences. In this process, the labeled data pre-trains the encoder, and the unlabeled data re- fine the model parameters by minimizing the reconstruction loss. The proposed model is evaluated on the data from public database and hospital. Comparison experiments were conducted where the developed LLN model achieved rather satisfied performance while dealing with the unseen new data. The obtained results demonstrate the effectiveness of the proposed approach in addressing individual differences. This can improve the quality of automatic sleep staging when assessed on different individuals and has strong application potential as a computer aided approach for sleep staging.