Automatic Sleep Stage Classification Based on LSTM

• Published in: Computer Supported Cooperative Work and Social Computing Vol. 917; pp. 478 - 486
• Main Authors: Shi, Peiying, Zheng, Xiangwei, Du, Ping, Yuan, Feng
• Format: Book Chapter
• Language: English
• Published: Singapore Springer 2019; Springer Singapore
• Series: Communications in Computer and Information Science
• Subjects: Deep learning; Fuzzy entropy; Long Short-Term memory; Recurrent neural networks; Sleep stage classification
• ISBN: 9789811330438; 9811330433
• ISSN: 1865-0929; 1865-0937
• DOI: 10.1007/978-981-13-3044-5_35

Sleep stage classification is closely related to human health and plays an effective role in sleep disorders diagnose. At present, the accuracy rate of conventional automatic sleep staging system is still not high, and there is still a large space for improvement. To solve the above problems, this paper proposed an automatic sleep stage classification method based on the Long Short-Term Memory (LSTM) and fuzzy entropy. We firstly adopt fuzzy entropy to extract the feature from electroencephalography (EEG) signals, then these features are put into the designed LSTM to carry out the automatic sleep stage classification and attain the final sleep stages. To verify the validity of our experiment, we tested it on public datasets called ISRUC_Sleep. The results demonstrate that this method improves the classification accuracy of sleep stages.