arrow
返回

Enhancing sleep stage classification through simultaneous time-frequency tokenization

delete2025-02-28
delete0
PRE
AI
Q
Qiaoli Zhou
S
Shurui Li
X
Xiyuan Ye
X
Xiaoqiang Shi
Q
Qiang Du
K
Ke Li *
DOI:10.1016/j.bspc.2025.107553delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

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

期刊

Biomedical Signal Processing and Control 封面图
Biomedical Signal Processing and Control
IF:
4.9
论文数:
1.0W
被引数:
2.4W

机构

S
Shenyang Aerospace University
学者数:
3.1K
论文数: 1.9K
被引数: 2.0K
S
Shenyang University of Technology
学者数:
5.0K
论文数: 3.3K
被引数: 3.4K
C
chinese academy of sciences
学者数:
56.7W
论文数: 45.0W
被引数: 704
学者 查看更多机构
引用论文

引用论文

Signs, Symptoms, and Ill-Defined Conditions in Persian Gulf War Veterans
err1998-01-01
err0
PREAI
errMichael J. Roy; Patricia A. Koslowe; Kurt Kroenke; Charles Magruder
err分享
err收藏
err
IF0
err
err0
PREAI
err
err分享
err收藏
err
IF0
err
err0
PREAI
err
err分享
err收藏
Relational local electroencephalography representations for sleep scoring
err2022-10-01
err4
errOAAI
errBrandmayr, Georg; Hartmann, Manfred; Furbass, Franz; Matz, Gerald; Samwald, Matthias; Kluge, Tilmann; Dorffner, Georg
err分享
err收藏
err
IF0
err
err0
PREAI
err
err分享
err收藏
学者 查看更多内容