提出因果生理表征学习,让心电图模型更抗干扰且实时可用。
CPR: Causal Physiological Representation Learning for Robust ECG Analysis under Distribution Shifts
- 基于因果分解和生理结构先验,分离病理特征与干扰噪声。
- 在PTB-XL数据集上,对抗攻击下F1达0.632,优于平滑方法9.1%。
- 兼具随机平滑的认证鲁棒性与单次前向传播效率,适合临床部署。
心电图深度学习模型虽精度高,但对平滑对抗扰动(SAP)等生物形态类干扰敏感。现有防御方法存在两难:对抗训练计算开销大,而随机平滑虽具认证鲁棒性却引入显著推理延迟。本文认为根源在于模型依赖非鲁棒的虚假相关,而非不变的病理特征。为此提出因果生理表征学习(CPR),在因果解耦框架中引入生理结构先验,通过结构因果模型(SCM)建模心电图生成过程,强制分离不变的病理波形(P-QRS-T复合波)与非因果伪影。在PTB-XL数据集上的实验表明,面对SAP攻击时,CPR的F1得分为0.632,较中值平滑(0.541)提升9.1%。关键的是,其认证鲁棒性媲美随机平滑,同时保持单次前向传播的高效推理,实现了鲁棒性、效率与临床可解释性的最优平衡。
原文摘要 · Abstract (English)
Deep learning models for Electrocardiogram (ECG) diagnosis have achieved remarkable accuracy but exhibit fragility against adversarial perturbations, particularly Smooth Adversarial Perturbations (SAP) that mimic biological morphology. Existing defenses face a critical dilemma: Adversarial Training (AT) provides robustness but incurs a prohibitive computational burden, while certified methods like Randomized Smoothing (RS) introduce significant inference latency, rendering them impractical for real-time clinical monitoring. We posit that this vulnerability stems from the models' reliance on non-robust spurious correlations rather than invariant pathological features. To address this, we propose Causal Physiological Representation Learning (CPR). Unlike standard denoising approaches that operate without semantic constraints, CPR incorporates a Physiological Structural Prior within a causal disentanglement framework. By modeling ECG generation via a Structural Causal Model (SCM), CPR enforces a structural intervention that strictly separates invariant pathological morphology (P-QRS-T complex) from non-causal artifacts. Empirical results on PTB-XL demonstrate that CPR significantly outperforms standard clinical preprocessing methods. Specifically, under SAP attacks, CPR achieves an F1 score of 0.632, surpassing Median Smoothing (0.541 F1) by 9.1%. Crucially, CPR matches the certified robustness of Randomized Smoothing while maintaining single-pass inference efficiency, offering a superior trade-off between robustness, efficiency, and clinical interpretability.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。