arXiv:2502.18733cs.LGcs.AI2025-02被引 10

用Transformer模型在多模态生理信号上实现高精度压力检测

Cross-Modality Investigation on WESAD Stress Classification

  • 基于Transformer构建单模态分析模型,处理心电、皮电等六类信号
  • 准确率、精确率、召回率均达99.73%至99.95%,性能业界领先
  • 首次可视化解释嵌入空间,揭示跨模态性能差异,适合可穿戴健康研究者

深度学习在医疗领域的广泛应用推动了人工智能与传感器技术在诊断、治疗和监测中的融合。在移动健康领域,基于AI的工具实现了压力等疾病的早期诊断与持续监测。可穿戴设备与多模态生理数据使压力检测更具可行性,但模型性能依赖于数据质量、数量及模态。本研究利用WESAD数据集,构建Transformer模型,对心电(ECG)、皮电活动(EDA)、肌电(EMG)、呼吸频率(RESP)、温度(TEMP)及三轴加速度计(ACC)信号进行压力检测。结果表明,单模态Transformer在分析生理信号方面具有优异效果,在压力检测中准确率、精确率与召回率均达到99.73%至99.95%。此外,研究还探索了跨模态表现,并通过二维嵌入空间可视化与数据方差量化分析加以解释。尽管已有大量压力检测研究,但模型在不同模态间的鲁棒性与泛化能力仍缺乏探讨。本工作是首批对压力检测嵌入空间进行解释的研究之一,为跨模态性能提供了重要洞察。

原文摘要 · Abstract (English)

Deep learning's growing prevalence has driven its widespread use in healthcare, where AI and sensor advancements enhance diagnosis, treatment, and monitoring. In mobile health, AI-powered tools enable early diagnosis and continuous monitoring of conditions like stress. Wearable technologies and multimodal physiological data have made stress detection increasingly viable, but model efficacy depends on data quality, quantity, and modality. This study develops transformer models for stress detection using the WESAD dataset, training on electrocardiograms (ECG), electrodermal activity (EDA), electromyography (EMG), respiration rate (RESP), temperature (TEMP), and 3-axis accelerometer (ACC) signals. The results demonstrate the effectiveness of single-modality transformers in analyzing physiological signals, achieving state-of-the-art performance with accuracy, precision and recall values in the range of $99.73\%$ to $99.95\%$ for stress detection. Furthermore, this study explores cross-modal performance and also explains the same using 2D visualization of the learned embedding space and quantitative analysis based on data variance. Despite the large body of work on stress detection and monitoring, the robustness and generalization of these models across different modalities has not been explored. This research represents one of the initial efforts to interpret embedding spaces for stress detection, providing valuable information on cross-modal performance.

压力检测多模态Transformer可穿戴

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