轻量自监督模型提升睡眠分期效率,适合临床部署
Sleep-stage efficient classification using a lightweight self-supervised model

- 用1D卷积替换主干网络,降低模型复杂度
- 特征拼接+线性SVM实现92.3%准确率,优于原模型
- 数据量减少反而提升性价比,适合资源受限场景
精准的睡眠分期对睡眠障碍诊断至关重要,自动化可显著提升临床评估效率。本研究探索将自监督模型(具体为改进版mulEEG)与线性SVM分类器结合用于睡眠分期。方法上,将mulEEG中的ResNet-50替换为1D卷积与时序编码器,采用ResNet-18主干;对比了不同模型配置与数据量的影响,并测试时序特征、频谱图特征及两者拼接作为线性SVM输入的效果。结果表明,减少数据量比简化模型带来更好的成本效益比;使用拼接特征的ResNet-18方案优于原始mulEEG的线性评估,达到92.3%的分类准确率。结论:通过简化模型提取特征并搭配强分类器,可实现更高效准确的睡眠分期,该方法有望推广至其他生物信号分类任务。
原文摘要 · Abstract (English)
Accurate classification of sleep stages is crucial for diagnosing sleep disorders and automating this process can significantly enhance clinical assessments. This study aims to explore the use of a self-supervised model (more specifically, an adapted version of mulEEG) combined with a Linear SVM classifier to improve sleep stage classification. \textbf{Methods:} The mulEEG model, which learns electroencephalogram signal representations in a self-supervised manner, was simplified here by replacing ResNet-50 with 1D-convolutions used as time series encoder by a ResNet-18 backbone. Two other adaptations were conducted: the first one evaluated different configurations of the model and data volume for training, while the second tested the effectiveness of time series features, spectrogram features, and their concatenation as inputs to a Linear SVM classifier. \textbf{Results:} The results showed that reducing the volume of data offered a better cost-benefit ratio compared to simplifying the model. Using the concatenated features with ResNet-18 also outperformed the linear evaluations of the original mulEEG model, achieving higher classification performance. \textbf{Conclusions:} Simplifying the mulEEG model to extract features and pairing it with a robust classifier leads to more efficient and accurate sleep stage classification. This approach holds promise for improving clinical sleep assessments and can be extended to other biological signal classification tasks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。