轻量级模型实现单导联脑电睡眠分期高精度分类
NanoSleep: A Parameter-Efficient Hybrid Temporal Convolutional Network for Single-Channel Sleep Stage Classification

- 融合多尺度时频特征的双分支卷积结构
- 在两个数据集上准确率优于六种基线方法
- 适合可穿戴设备与资源受限场景使用
从单导联脑电图(EEG)进行睡眠分期分类对可穿戴及居家睡眠监测至关重要。然而,许多深度学习模型虽精度高,但参数量大,难以部署在资源受限设备上。本文提出NanoSleep,一种紧凑型混合时间卷积网络,用于自动睡眠分期分类。该模型结合可学习的Sinc卷积前端、双分支特征提取器(融合多尺度时序与频域表示)、带通道重校准的门控扩张时间卷积主干网络,以及用于序列级解码的条件随机场。此外,采用加权校准焦点损失以缓解类别不平衡问题。在Sleep-EDF和Sleep-EDF-Expanded数据集上,通过受试者间交叉验证评估,所提模型始终优于六种代表性基线方法;消融实验验证了各主要模块的有效性。结果表明,NanoSleep在精度与效率间取得良好平衡,适用于可穿戴设备、居家睡眠监测及资源受限的临床应用。
原文摘要 · Abstract (English)
Sleep stage classification from single-channel electroencephalography (EEG) is essential for wearable and home-based sleep monitoring. However, many deep learning models achieve high accuracy at the cost of large model sizes, which limits their deployment on resource-constrained devices. In this work, we present NanoSleep, a compact hybrid temporal convolutional network for automatic sleep stage classification. NanoSleep combines a learnable Sinc-convolutional front end, a dual-branch feature extractor that fuses multi-scale temporal and spectral representations, a gated dilated temporal convolutional backbone with channel recalibration, and a conditional random field for sequence-level decoding. We further employ a weighted calibrated focal loss to address class imbalance. We evaluate NanoSleep on the Sleep-EDF and Sleep-EDF-Expanded datasets using subject-wise cross-validation. The proposed model consistently outperforms six representative baseline methods, and an ablation study confirms the contribution of each major component. These results demonstrate that NanoSleep provides an effective balance between accuracy and efficiency, making it well suited for wearable devices, home-based sleep monitoring, and resource-constrained clinical applications.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。