研究如何让PPG测血压模型在不同数据集间更通用,发现分布差异是关键挑战
Generalizable deep learning for photoplethysmography-based blood pressure estimation -- A Benchmarking Study
- 用样本级领域自适应提升模型跨数据集表现
- 无校准情况下外部数据集误差达15.0~25.1 mmHg(收缩压)
- 强调数据分布差异影响模型泛化,提出训练优化建议
基于光电容积脉搏波描记法(PPG)的血压(BP)估计算法为传统袖带测量提供了有前景的替代方案。近年来,大量深度学习模型被提出用于从原始PPG波形中推断血压。然而,这些模型主要在分布内测试集上评估,其泛化能力受到质疑。为此,我们在新发布的PulseDB数据集上训练了五种深度学习模型,在该数据集上提供分布内基准结果,并进一步评估其在多个外部数据集上的分布外性能。最佳模型(XResNet1d101)在有个体校准的情况下,于PulseDB上实现了9.4和6.0 mmHg的收缩压与舒张压平均绝对误差(MAE);无校准时分别为14.0和8.5 mmHg。在无校准条件下,外部测试集的等效MAE范围为15.0–25.1 mmHg(SBP)和7.0–10.4 mmHg(DBP)。结果表明,模型性能显著受数据集间血压分布差异的影响。我们探索了一种简单的样本级领域自适应方法以提升性能,并提出了具备良好泛化能力模型的训练建议。本工作旨在提高研究者对分布外泛化重要性与挑战的认识。
原文摘要 · Abstract (English)
Photoplethysmography (PPG)-based blood pressure (BP) estimation represents a promising alternative to cuff-based BP measurements. Recently, an increasing number of deep learning models have been proposed to infer BP from the raw PPG waveform. However, these models have been predominantly evaluated on in-distribution test sets, which immediately raises the question of the generalizability of these models to external datasets. To investigate this question, we trained five deep learning models on the recently released PulseDB dataset, provided in-distribution benchmarking results on this dataset, and then assessed out-of-distribution performance on several external datasets. The best model (XResNet1d101) achieved in-distribution MAEs of 9.4 and 6.0 mmHg for systolic and diastolic BP respectively on PulseDB (with subject-specific calibration), and 14.0 and 8.5 mmHg respectively without calibration. Equivalent MAEs on external test datasets without calibration ranged from 15.0 to 25.1 mmHg (SBP) and 7.0 to 10.4 mmHg (DBP). Our results indicate that the performance is strongly influenced by the differences in BP distributions between datasets. We investigated a simple way of improving performance through sample-based domain adaptation and put forward recommendations for training models with good generalization properties. With this work, we hope to educate more researchers for the importance and challenges of out-of-distribution generalization.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。