An Automatic Sleep Staging Method Based on Spatio-Temporal JK-ViT Architecture

Yu An,Yikun Feng,Zhihong Li,Xi Zhang
DOI: https://doi.org/10.1109/case59546.2024.10711507
2024-01-01
Abstract:Sleep, a physiological process essential for human health, is characterized by distinct stages. The accurate identification of these stages is paramount for diagnosing sleep disorders and improving sleep quality. However, manual sleep staging is labor-intensive, which raises the need for automated sleep staging. Advances in machine learning, particularly deep learning, have shown promise in automating sleep stage classification, yet challenges persist in capturing the spatio-temporal dynamics of high-dimensional physiological data. Moreover, multi-source data from channels with different types are typically collected during the entire sleep process. It is essential to effectively utilize these data and capture the spatiotemporal correlations among different sources. Therefore, we propose a novel spatio-temporal sleep staging method based on a JK-ViT framework. In the proposed method, a Graph Encoder based on the Jumping Knowledge Network (JK-Net) structure is presented to capture the spatio-temporal dynamics. Leveraging the Vision-Transformer (ViT) framework, multisource data from channels of different types are fused and the attention layer captures the correlation between different sources. The effectiveness of method is validated using a private sleep PSG dataset, showcasing its potential to enhance the performance of sleep staging.
What problem does this paper attempt to address?