提升单导联脑电睡眠分期准确率,尤其改善易被误判的N1期识别
A Context-Aware Temporal Modeling through Unified Multi-Scale Temporal Encoding and Hierarchical Sequence Learning for Single-Channel EEG Sleep Staging
- 融合多尺度时序编码与分层序列学习,捕捉局部与长程依赖关系
- 在SleepEDF数据集上达89.72%准确率,N1期F1-score提升至61.7%
- 通过分段平均预测增强上下文感知,模型结果可解释且适合临床应用
自动睡眠分期对全球普遍存在的睡眠障碍诊疗至关重要。本研究聚焦于单导联脑电图(EEG),这是一种实用且广泛应用的信号源。现有方法存在类别不平衡、感受野建模有限及可解释性不足等挑战。本文提出一种上下文感知且可解释的单导联EEG睡眠分期框架,重点提升对N1期的检测能力。多数先前模型为黑箱结构,堆叠多层却缺乏明确的特征提取分工。所提模型结合紧凑的多尺度特征提取与时序建模,同时利用类别加权损失函数与数据增强缓解数据不平衡问题。将EEG信号分割为子时段片段,最终通过平均各片段softmax概率获得预测结果,增强上下文表征与鲁棒性。该框架在SleepEDF数据集上取得89.72%的整体准确率和85.46%的宏平均F1分数,其中对具有挑战性的N1期实现61.7%的F1分数,显著优于此前方法。结果表明,该方法有效提升了睡眠分期性能,同时保持良好可解释性,适用于真实临床场景。
原文摘要 · Abstract (English)
Automatic sleep staging is a critical task in healthcare due to the global prevalence of sleep disorders. This study focuses on single-channel electroencephalography (EEG), a practical and widely available signal for automatic sleep staging. Existing approaches face challenges such as class imbalance, limited receptive-field modeling, and insufficient interpretability. This work proposes a context-aware and interpretable framework for single-channel EEG sleep staging, with particular emphasis on improving detection of the N1 stage. Many prior models operate as black boxes with stacked layers, lacking clearly defined and interpretable feature extraction roles.The proposed model combines compact multi-scale feature extraction with temporal modeling to capture both local and long-range dependencies. To address data imbalance, especially in the N1 stage, classweighted loss functions and data augmentation are applied. EEG signals are segmented into sub-epoch chunks, and final predictions are obtained by averaging softmax probabilities across chunks, enhancing contextual representation and robustness.The proposed framework achieves an overall accuracy of 89.72% and a macro-average F1-score of 85.46%. Notably, it attains an F1- score of 61.7% for the challenging N1 stage, demonstrating a substantial improvement over previous methods on the SleepEDF datasets. These results indicate that the proposed approach effectively improves sleep staging performance while maintaining interpretability and suitability for real-world clinical applications.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。