用深度学习提升复杂环境下非接触心率测量的准确性和泛化能力
Robust and Generalizable Heart Rate Estimation via Deep Learning for Remote Photoplethysmography in Complex Scenarios
- 采用3D CNN与差分帧融合,捕捉面部视频中的血容量脉动变化
- 在MMPD数据集上实现7.58的平均绝对误差,优于现有最佳模型
- 适合需要高鲁棒性心率监测的医疗健康、远程诊断场景
非接触式远程光体积描记术(rPPG)技术可通过面部视频实现心率测量。然而,现有网络模型在复杂场景下的准确性、鲁棒性和泛化能力仍面临挑战。本文提出一种端到端的rPPG提取网络,采用3D卷积神经网络从原始面部视频中重建精确的rPPG信号。引入差分帧融合模块,将差分帧与原始帧结合,使帧级表征能够捕捉血容量脉动(BVP)变化。同时,结合时间移位模块(TSM)与自注意力机制,在计算开销极小的前提下有效增强rPPG特征。此外,提出一种新型动态混合损失函数,为网络提供更强监督,有效缓解过拟合。在PURE、UBFC-rPPG及复杂场景下的MMPD数据集上进行了全面实验,涵盖内部与跨数据集评估,结果表明该网络具备优异的鲁棒性与泛化能力。具体而言,于PURE数据集训练后,在MMPD测试集上取得7.58的平均绝对误差(MAE),超越当前最优模型。
原文摘要 · Abstract (English)
Non-contact remote photoplethysmography (rPPG) technology enables heart rate measurement from facial videos. However, existing network models still face challenges in accu racy, robustness, and generalization capability under complex scenarios. This paper proposes an end-to-end rPPG extraction network that employs 3D convolutional neural networks to reconstruct accurate rPPG signals from raw facial videos. We introduce a differential frame fusion module that integrates differential frames with original frames, enabling frame-level representations to capture blood volume pulse (BVP) variations. Additionally, we incorporate Temporal Shift Module (TSM) with self-attention mechanisms, which effectively enhance rPPG features with minimal computational overhead. Furthermore, we propose a novel dynamic hybrid loss function that provides stronger supervision for the network, effectively mitigating over fitting. Comprehensive experiments were conducted on not only the PURE and UBFC-rPPG datasets but also the challenging MMPD dataset under complex scenarios, involving both intra dataset and cross-dataset evaluations, which demonstrate the superior robustness and generalization capability of our network. Specifically, after training on PURE, our model achieved a mean absolute error (MAE) of 7.58 on the MMPD test set, outperforming the state-of-the-art models.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。