用分层双流结构提升睡眠分期准确率,尤其改善了模糊阶段识别。
SWINSleepNet: A Hierarchical Context-Aware Framework for Sleep Staging (v2)

- 分两路处理原始脑电波和时频图,分别提取细微波形与频域特征
- 在三个数据集上对难分的N1期和过渡期识别准确率显著提升
- 适合需要高精度睡眠分析的临床诊断与长期健康监测场景
自动睡眠分期在睡眠障碍诊断、睡眠质量评估和长期健康监测中至关重要,但现有方法在模糊及过渡性睡眠阶段表现不佳,主要源于对细粒度帧内结构和跨区域谱依赖关系建模不足。传统帧级编码器难以捕捉微小时间结构和帧内跨区域交互,导致对N1等困难阶段识别效果差。为此,本文提出SwinSleepNet,一种分层上下文感知双流框架,分别优化帧内表征学习与帧间上下文建模。具体地,从原始时域脑电信号和时频变换两个互补视角刻画每个睡眠帧:时域分支采用卷积编码器捕捉精细波形时间细节;时频分支使用Swin Transformer提取局部时频特征、多尺度层次信息和长程空间依赖。多分支特征融合为集成嵌入,并通过双向上下文模块捕获跨帧时间依赖,用于最终睡眠分期分类。在Sleep-EDF-20、Sleep-EDF-78和SHHS数据集上的综合实验表明,该方法整体性能具有竞争力,且在困难的N1阶段和过渡帧上表现出更强鲁棒性和稳定性。结果证明,基于分层架构的优化帧内表征学习极大提升了自动睡眠分期任务的表现。
原文摘要 · Abstract (English)
Automatic sleep staging is a critical role in sleep disorder diagnosis, sleep quality assessment, and long-term health monitoring; however, existing approaches suffer poor performance on ambiguous and transition-related sleep stages, caused by inadequate modeling of fine-grained intra-epoch structures and complex cross-region spectral dependencies. Traditional epoch-level encoders commonly fail to extract subtle temporal microstructures and intra-epoch cross-region interactions, resulting in unsatisfactory recognition accuracy for hard categories such as the N1 stage. To tackle these drawbacks, we propose SwinSleepNet, a hierarchical context-aware dual-stream framework that separately optimizes intra-epoch representation learning and inter-epoch contextual modeling. Concretely, we characterize each sleep epoch from two complementary perspectives: raw time-domain EEG signal and its time-frequency transformation. The time-domain branch adopts convolutional encoders to capture fine waveform temporal details, and the time-frequency branch uses Swin Transformer to extract local spectro-temporal features, hierarchical multi-scale information and long-range spatial dependencies. The multi-branch extracted features are fused into integrated embeddings, which are optimized by a bidirectional context module to capture cross-epoch temporal dependencies for final sleep stage classification. Comprehensive experiments on Sleep-EDF-20, Sleep-EDF-78 and SHHS datasets verify that our method achieves competitive overall performance, and exhibits stronger robustness and stability on difficult N1 stages and transitional epochs. The results prove that optimized intra-epoch representation learning based on hierarchical architecture greatly benefits automatic sleep staging tasks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。