arXiv:2412.01929cs.AIcs.LG2024-12被引 5

用心电图准确识别五种睡眠阶段,无需复杂设备。

ECG-SleepNet: Deep Learning-Based Comprehensive Sleep Stage Classification Using ECG Signals

  • 三阶段深度学习框架,融合特征模仿与时间频率分析
  • 整体准确率80.79%,N1阶段达60.36%(经SMOTE增强)
  • 适合睡眠研究、可穿戴健康设备开发者参考

准确的睡眠分期对理解睡眠障碍和改善整体健康至关重要。本研究提出一种基于心电信号的三阶段深度学习方法,作为传统依赖脑电图等复杂模态的替代方案。第一阶段通过特征模仿网络(FINs)提取关键特征,提升精度并加快收敛;第二阶段利用心电信号的时间-频率表示识别N1期;第三阶段整合前两阶段模型,并采用柯尔莫哥洛夫-阿诺德网络(KAN)完成五类睡眠阶段分类。此外,针对数据稀疏的N1期,采用SMOTE数据增强技术提升分类性能。实验结果表明,整体准确率达80.79%,总体卡帕系数为0.73。各阶段准确率为:清醒86.70%、N1期60.36%、N2期83.89%、N3期84.85%、REM期87.16%。研究强调权重初始化与数据增强在优化心电睡眠分期中的关键作用。

原文摘要 · Abstract (English)

Accurate sleep stage classification is essential for understanding sleep disorders and improving overall health. This study proposes a novel three-stage approach for sleep stage classification using ECG signals, offering a more accessible alternative to traditional methods that often rely on complex modalities like EEG. In Stages 1 and 2, we initialize the weights of two networks, which are then integrated in Stage 3 for comprehensive classification. In the first phase, we estimate key features using Feature Imitating Networks (FINs) to achieve higher accuracy and faster convergence. The second phase focuses on identifying the N1 sleep stage through the time-frequency representation of ECG signals. Finally, the third phase integrates models from the previous stages and employs a Kolmogorov-Arnold Network (KAN) to classify five distinct sleep stages. Additionally, data augmentation techniques, particularly SMOTE, are used in enhancing classification capabilities for underrepresented stages like N1. Our results demonstrate significant improvements in the classification performance, with an overall accuracy of 80.79% an overall kappa of 0.73. The model achieves specific accuracies of 86.70% for Wake, 60.36% for N1, 83.89% for N2, 84.85% for N3, and 87.16% for REM. This study emphasizes the importance of weight initialization and data augmentation in optimizing sleep stage classification with ECG signals.

睡眠分期心电图深度学习KAN

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。