返回
Enhancing sleep stage classification through simultaneous time-frequency tokenization
DOI:10.1016/j.bspc.2025.107553.png)
摘要
En 中文
Accurate sleep stage classification is crucial for assessing sleep quality and diagnosing sleep disorders. The current sleep stage classification methods based on time-frequency images generally adopt Transformer architecture design, taking the time dimension or frequency dimension as the token, which will make it impossible to extract the time domain and frequency domain information at the same time. In order to address this issue, this paper proposes a novel Multi-branch Sleep Stage Classification Network (MBSleepNet), which consists of three branches: Time Branch, Frequency Branch, and Global Branch. The Time Branch and Frequency Branch are designed based on the Transformer, treating the time and frequency dimensions as tokens, respectively, to simultaneously extract time domain and frequency domain information. The Global Branch uses CNN to extract cross-modal global features from the time-frequency image. In addition, to fully utilize the information extracted by these three branches, we also designed a Multi-branch Information Fusion Module (MBIFM) to fully integrate time domain, frequency domain, and cross-modal global information to enhance sleep stage classification. Our method achieves a state-of-the-art result on the SleepEDF-20, SleepEDF78 and SHHS datasets. Specifically, MBSleepNet achieves 88.2% accuracy on the SleepEDF-20, 86.5% on the SleepEDF-78, and 89.2% on the SHHS.
Keyword:
Automatic sleep stage classification
Time-frequency image
Feature fusion
EEG
期刊
IF:
4.9
论文数:
1.0W
被引数:
2.4W
机构
引用论文
Leveraging discriminative features for automatic sleep stage classification based on raw single-channel EEG利用基于原始单通道EEG的判别特征进行自动睡眠阶段分类
Black Intellectuals' Critique of Early Mental Testing: A Little-Known Saga of the 1920s黑人知识分子对早期心理测试的批判: 一个鲜为人知的20世纪20年代传奇
Cross-scenario automatic sleep stage classification using transfer learning and single-channel EEG使用迁移学习和单通道EEG的跨场景自动睡眠阶段分类

