MSSC-BiMamba: Multimodal Sleep Stage Classification and Early Diagnosis of Sleep Disorders with Bidirectional Mamba

Chao Zhang,Weirong Cui,Jingjing Guo
2024-05-31
Abstract:Monitoring sleep states is essential for evaluating sleep quality and diagnosing sleep disorders. Traditional manual staging is time-consuming and prone to subjective bias, often resulting in inconsistent outcomes. Here, we developed an automated model for sleep staging and disorder classification to enhance diagnostic accuracy and efficiency. Considering the characteristics of polysomnography (PSG) multi-lead sleep monitoring, we designed a multimodal sleep state classification model, MSSC-BiMamba, that combines an Efficient Channel Attention (ECA) mechanism with a Bidirectional State Space Model (BSSM). The ECA module allows for weighting data from different sensor channels, thereby amplifying the influence of diverse sensor inputs. Additionally, the implementation of bidirectional Mamba (BiMamba) enables the model to effectively capture the multidimensional features and long-range dependencies of PSG data. The developed model demonstrated impressive performance on sleep stage classification tasks on both the ISRUC-S3 and ISRUC-S1 datasets, respectively containing data with healthy and unhealthy sleep patterns. Also, the model exhibited a high accuracy for sleep health prediction when evaluated on a combined dataset consisting of ISRUC and Sleep-EDF. Our model, which can effectively handle diverse sleep conditions, is the first to apply BiMamba to sleep staging with multimodal PSG data, showing substantial gains in computational and memory efficiency over traditional Transformer-style models. This method enhances sleep health management by making monitoring more accessible and extending advanced healthcare through innovative technology.
Artificial Intelligence
What problem does this paper attempt to address?
The paper attempts to address the issues of sleep stage classification and early diagnosis of sleep disorders. Traditionally, the classification of sleep stages requires manual intervention by professionals, a process that is not only time-consuming but also prone to subjective bias, leading to inconsistent results. Therefore, researchers have developed an automated model to improve the accuracy and efficiency of sleep staging and sleep disorder classification. Specifically, the paper proposes a multimodal sleep stage classification model named MSSC-BiMamba. This model combines the Efficient Channel Attention (ECA) mechanism with the Bidirectional State Space Model (BSSM) to effectively handle Polysomnography (PSG) data. The ECA module can weight the data from different sensor channels, thereby amplifying the impact of diverse sensor inputs. The introduction of BiMamba enables the model to capture the multidimensional features and long-range dependencies of PSG data. Experimental results show that this model performs excellently in the sleep stage classification tasks on the ISRUC-S3 and ISRUC-S1 datasets, and also achieves a high accuracy (0.952) in sleep health prediction on the combined ISRUC and Sleep-EDF datasets. This indicates that the model can effectively handle various sleep conditions and is computationally and memory efficient compared to traditional Transformer-style models. Through this approach, researchers hope to enhance the level of sleep health management, make monitoring more convenient, and expand advanced healthcare services through innovative technology.