用视觉变压器同时识别睡眠阶段和呼吸暂停,提升睡眠分析效率。
Multimodal Sleep Stage and Sleep Apnea Classification Using Vision Transformer: A Multitask Explainable Learning Approach
- 用1D-ViT模型融合多模态信号,同步预测睡眠阶段与呼吸暂停
- 睡眠分期准确率78%(κ=0.66),呼吸暂停检测准确率74%(κ=0.58)
- 通过注意力机制揭示呼吸波形关键特征,增强模型可解释性
睡眠是人体生理的重要组成部分,对整体健康和生活质量具有显著影响。准确的睡眠分期与睡眠障碍检测对于评估睡眠质量至关重要。现有研究多基于多导睡眠图(PSG)或单模态信号的机器学习方法,但普遍缺乏多模态、多标签联合框架,且将睡眠分期与疾病分类分开处理。本文提出一种1D-Vision Transformer模型,实现睡眠阶段与睡眠障碍的联合分类。该模型利用呼吸暂停与特定睡眠阶段模式的相关性,同步识别睡眠阶段与睡眠障碍。在包含光电容积脉搏波描记法(photoplethysmogram)、呼吸气流和呼吸努力信号的多模态多标签数据集上进行训练与测试。实验结果表明,五阶段睡眠分期的总体准确率为78%(Cohen's Kappa = 0.66),睡眠呼吸暂停分类准确率为74%(Cohen's Kappa = 0.58)。此外,通过分析编码器注意力权重,揭示了呼吸波形的谷值与峰值等模式对分类决策贡献更大,提升了模型可解释性。
原文摘要 · Abstract (English)
Sleep is an essential component of human physiology, contributing significantly to overall health and quality of life. Accurate sleep staging and disorder detection are crucial for assessing sleep quality. Studies in the literature have proposed PSG-based approaches and machine-learning methods utilizing single-modality signals. However, existing methods often lack multimodal, multilabel frameworks and address sleep stages and disorders classification separately. In this paper, we propose a 1D-Vision Transformer for simultaneous classification of sleep stages and sleep disorders. Our method exploits the sleep disorders' correlation with specific sleep stage patterns and performs a simultaneous identification of a sleep stage and sleep disorder. The model is trained and tested using multimodal-multilabel sensory data (including photoplethysmogram, respiratory flow, and respiratory effort signals). The proposed method shows an overall accuracy (cohen's Kappa) of 78% (0.66) for five-stage sleep classification and 74% (0.58) for sleep apnea classification. Moreover, we analyzed the encoder attention weights to clarify our models' predictions and investigate the influence different features have on the models' outputs. The result shows that identified patterns, such as respiratory troughs and peaks, make a higher contribution to the final classification process.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。