将生理知识融入自监督学习,提升心电图分析的临床有效性。
Domain Knowledge is Power: Leveraging Physiological Priors for Self Supervised Representation Learning in Electrocardiography
- 用生理相似性引导对比学习,让模型学出有意义的心电特征。
- 在多个数据集上相对最强基线,平均AUROC提升12%。
- 适合做心电图诊断的少样本或无监督学习研究者参考。
心电图(ECG)在心脏疾病诊断中至关重要,但人工智能分析常受限于标注数据不足。自监督学习(SSL)可利用大量未标注数据缓解此问题。本文提出PhysioCLR(生理感知对比学习框架),通过引入心电生理先验知识,增强心电图心律失常分类的泛化性和临床相关性。预训练阶段,PhysioCLR学习将具有相似临床特征的样本嵌入拉近,而差异样本则推远。与现有方法不同,该方法将心电生理相似性线索融入对比学习,促进临床有意义表征的学习。此外,设计了保持类别一致性的心电特异性数据增强策略,并提出混合损失函数以进一步优化表征质量。在两个公开心电数据集Chapman和Georgia及一个私有ICU数据集(二分类标注)上评估,结果表明,相较于最强基线,PhysioCLR在三组数据上平均AUROC相对提升12%,展现出优异的跨数据集泛化能力。结论:将生理知识嵌入对比学习,使模型能够学习到临床可解释且可迁移的心电特征。意义在于,生理信息驱动的自监督学习为更高效、低标注依赖的心电诊断提供了可行路径。
原文摘要 · Abstract (English)
Objective: Electrocardiograms (ECGs) play a crucial role in diagnosing heart conditions; however, the effectiveness of artificial intelligence (AI)-based ECG analysis is often hindered by the limited availability of labeled data. Self-supervised learning (SSL) can address this by leveraging large-scale unlabeled data. We introduce PhysioCLR (Physiology-aware Contrastive Learning Representation for ECG), a physiology-aware contrastive learning framework that incorporates domain-specific priors to enhance the generalizability and clinical relevance of ECG-based arrhythmia classification. Methods: During pretraining, PhysioCLR learns to bring together embeddings of samples that share similar clinically relevant features while pushing apart those that are dissimilar. Unlike existing methods, our method integrates ECG physiological similarity cues into contrastive learning, promoting the learning of clinically meaningful representations. Additionally, we introduce ECG- specific augmentations that preserve the ECG category post augmentation and propose a hybrid loss function to further refine the quality of learned representations. Results: We evaluate PhysioCLR on two public ECG datasets, Chapman and Georgia, for multilabel ECG diagnoses, as well as a private ICU dataset labeled for binary classification. Across the Chapman, Georgia, and private cohorts, PhysioCLR boosts the mean AUROC by 12% relative to the strongest baseline, underscoring its robust cross-dataset generalization. Conclusion: By embedding physiological knowledge into contrastive learning, PhysioCLR enables the model to learn clinically meaningful and transferable ECG eatures. Significance: PhysioCLR demonstrates the potential of physiology-informed SSL to offer a promising path toward more effective and label-efficient ECG diagnostics.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。