用二维谱图+时间建模提升跨数据集睡眠分期准确率
STDA-Net: Spectrogram-Based Domain Adaptation for cross-dataset Sleep Stage Classification
- 将脑电图转为二维谱图,结合双向LSTM与对抗域适应
- 跨数据集平均准确率89.03%,宏F1达87.64%,稳定性更高
- 适合做睡眠分期且标签稀疏场景的科研与临床应用
由于脑电通道配置、采样率、记录环境和受试者群体的差异,跨数据集的睡眠分期仍具挑战。尽管深度学习在自动化睡眠分期中展现出潜力,但现有跨数据集方法多依赖一维脑电信号表示,而基于二维谱图的无监督域适应框架尚未充分探索。本文提出STDA-Net(基于谱图的时间域适应网络),结合卷积神经网络进行谱图特征提取、双向长短期记忆模块建模睡眠动态,并利用域对抗神经网络实现源域到目标域特征对齐,无需目标域标签即可训练。在三个公开数据集Sleep-EDF、SHHS-1和SHHS-2上,六种跨数据集转移设置下,平均准确率达到89.03%,平均宏F1为87.64%,显著优于现有1D基线方法,且五次独立实验方差更小,表明性能更稳定可复现。结果表明,二维谱图结合时序建模与对抗域适应,是跨数据集睡眠分期的一种鲁棒高效替代方案。
原文摘要 · Abstract (English)
Accurate sleep stage classification across datasets remains challenging due to variability in EEG channel montages, sampling rates, recording environments, and subject populations. Although deep learning has shown considerable promise for automated sleep staging, most existing cross-dataset methods rely on one-dimensional EEG signal representations, whereas the use of two-dimensional spectrogram-based inputs within an unsupervised domain adaptation framework has remained largely unexplored. Here, we propose STDA-Net (Spectrogram-based Temporal Domain Adaptation Network), a framework that combines a convolutional neural network (CNN) for spectrogram-based feature extraction, a bidirectional long short-term memory (BiLSTM) module for temporal modeling of sleep dynamics, and a domain-adversarial neural network (DANN) for source-to-target feature alignment without requiring any labeled target-domain data during training. Experiments are conducted on three publicly available datasets Sleep-EDF, SHHS-1, and SHHS-2 under six cross-dataset transfer settings. Results show that the proposed framework achieves an average accuracy of 89.03% and an average macro F1-score of 87.64%, consistently outperforming existing 1D baseline methods in terms of balanced classification performance, with substantially lower variance across five independent runs, indicating improved stability and reproducibility. Overall, these findings demonstrate that 2D spectrogram-based representations, combined with temporal modeling and adversarial domain adaptation, provide a robust and competitive alternative to conventional 1D EEG inputs for cross-dataset sleep staging.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。