Sleep staging has always been a hot topic in the field of sleep medicine, and it is the cornerstone of research on sleep problems. At present, sleep staging heavily relies on manual interpretation, which is a time-consuming and laborious task with subjective interpretation factors. In this paper, we propose an automatic sleep stage classification model based on the Bidirectional Recurrent Neural Network (BiRNN) with data bundling augmentation and label redirection for accurate sleep staging. Through extensive analysis, we discovered that the incorrect classification labels are primarily concentrated in the transition and nonrapid eye movement stage I (N1). Therefore, our model utilizes a sliding window input to enhance data bundling and an attention mechanism to improve feature enhancement after label redirection. This approach focuses on mining latent features during the N1 and transition periods, which can further improve the network model’s classification performance. We evaluated on multiple public datasets and achieved an overall accuracy rate of 87.3%, with the highest accuracy rate reaching 93.5%. Additionally, the network model’s macro F1 score reached 82.5%. Finally, we used the optimal network model to study the impact of different EEG channels on the accuracy of each sleep stage.
scite is a Brooklyn-based organization that helps researchers better discover and understand research articles through Smart Citations–citations that display the context of the citation and describe whether the article provides supporting or contrasting evidence. scite is used by students and researchers from around the world and is funded in part by the National Science Foundation and the National Institute on Drug Abuse of the National Institutes of Health.