用生理波形预测心脏骤停,提升长期预警能力
Wav2Arrest 2.0: Long-Horizon Cardiac Arrest Prediction with Time-to-Event Modeling, Identity-Invariance, and Pseudo-Lab Alignment
- 采用生存分析建模事件发生时间,增强时序预测能力
- 通过身份无关特征学习,避免模型过拟合记忆个体特征
- 用预训练模型生成伪标签,弥补稀疏真实检验数据
高频生理波形能提供患者状态的深层实时信息。近年来,基于光电容积脉搏波(PPG)的生理基础模型(如PPG-GPT)已显示出对心脏骤停(CA)等危急事件的预测能力。然而,其强大表征仍需更有效利用,尤其在下游数据与标签稀缺的情况下。本文提出三项独立改进:一是采用事件发生时间建模,通过回归或细粒度离散生存分析;二是通过构建大规模去标识生物特征识别模型(p-vector),并以对抗方式消除患者身份等干扰因素,实现身份不变特征学习;三是利用预训练辅助估计网络生成伪标签(如乳酸、钠、肌钙蛋白、钾),通过零样本预测补充稀疏的真实实验室数据,形成连续目标值。这些方法使24小时平均AUC从0.74提升至0.78–0.80。尤其在长时程预警中表现优异,同时近事件阶段性能稳定,推动早期预警系统发展。最后,采用多任务学习并发现损失间梯度冲突率高,通过PCGrad优化缓解。
原文摘要 · Abstract (English)
High-frequency physiological waveform modality offers deep, real-time insights into patient status. Recently, physiological foundation models based on Photoplethysmography (PPG), such as PPG-GPT, have been shown to predict critical events, including Cardiac Arrest (CA). However, their powerful representation still needs to be leveraged suitably, especially when the downstream data/label is scarce. We offer three orthogonal improvements to improve PPG-only CA systems by using minimal auxiliary information. First, we propose to use time-to-event modeling, either through simple regression to the event onset time or by pursuing fine-grained discrete survival modeling. Second, we encourage the model to learn CA-focused features by making them patient-identity invariant. This is achieved by first training the largest-scale de-identified biometric identification model, referred to as the p-vector, and subsequently using it adversarially to deconfound cues, such as person identity, that may cause overfitting through memorization. Third, we propose regression on the pseudo-lab values generated by pre-trained auxiliary estimator networks. This is crucial since true blood lab measurements, such as lactate, sodium, troponin, and potassium, are collected sparingly. Via zero-shot prediction, the auxiliary networks can enrich cardiac arrest waveform labels and generate pseudo-continuous estimates as targets. Our proposals can independently improve the 24-hour time-averaged AUC from the 0.74 to the 0.78-0.80 range. We primarily improve over longer time horizons with minimal degradation near the event, thus pushing the Early Warning System research. Finally, we pursue multi-task formulation and diagnose it with a high gradient conflict rate among competing losses, which we alleviate via the PCGrad optimization technique.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。