arXiv:2512.02180cs.LGcs.AI2025-12被引 7

用临床风险评分指导心电图自监督学习,提升单导联心电图分析准确率。

CLEF: Clinically-Guided Contrastive Learning for Electrocardiogram Foundation Models

  • 基于临床风险评分动态加权负样本,让模型更关注有临床意义的差异。
  • 在16万患者数据上预训练,单导联心电图分类任务平均AUC提升2.6%以上。
  • 无需标注即可利用常规临床数据,适合远程健康监测场景。

心电图(ECG)是心血管健康的关键诊断工具,单导联记录已集成于临床级与消费类可穿戴设备中。尽管无标签心电图的自监督预训练能提升诊断性能,但现有方法未融合临床元数据中的领域知识。本文提出一种新型对比学习方法——临床引导对比学习(CLEF),利用已有的临床风险评分自适应地加权负样本对,使心电图嵌入的相似性与受试者间具有临床意义的差异对齐,并具备处理缺失元数据的显式机制。在包含16.1万患者的MIMIC-IV数据集上,使用常规临床元数据,仅通过12导联心电图数据对三种规模的单导联心电图基础模型进行预训练,统称为CLEF。在7个独立数据集上的18项临床分类与回归任务中评估,相比5个基础模型基线和3种自监督算法,当在12导联数据上预训练、在I导联数据上测试时,中等规模的CLEF在分类任务中平均AUC提升至少2.6%,在回归任务中平均MAE降低至少3.2%;相较于现有自监督算法,平均AUC提升至少1.8%。当仅用I导联数据预训练用于分类任务时,其性能可媲美经监督训练的SOTA模型ECGFounder。总体而言,CLEF实现了更准确、可扩展的单导联心电图分析,推动远程健康监测发展。代码与预训练模型已开源:github.com/Nokia-Bell-Labs/ecg-foundation-model。

原文摘要 · Abstract (English)

The electrocardiogram (ECG) is a key diagnostic tool in cardiovascular health. Single-lead ECG recording is integrated into both clinical-grade and consumer wearables. While self-supervised pretraining of foundation models on unlabeled ECGs improves diagnostic performance, existing approaches do not incorporate domain knowledge from clinical metadata. We introduce a novel contrastive learning approach that utilizes an established clinical risk score to adaptively weight negative pairs: clinically-guided contrastive learning. It aligns the similarities of ECG embeddings with clinically meaningful differences between subjects, with an explicit mechanism to handle missing metadata. On 12-lead ECGs from 161K patients in the MIMIC-IV dataset, we pretrain single-lead ECG foundation models at three scales, collectively called CLEF, using only routinely collected metadata without requiring per-sample ECG annotations. We evaluate CLEF on 18 clinical classification and regression tasks across 7 held-out datasets, and benchmark against 5 foundation model baselines and 3 self-supervised algorithms. When pretrained on 12-lead ECG data and tested on lead-I data, CLEF outperforms self-supervised foundation model baselines: the medium-sized CLEF achieves average AUROC improvements of at least 2.6% in classification and average reductions in MAEs of at least 3.2% in regression. Comparing with existing self-supervised learning algorithms, CLEF improves the average AUROC by at least 1.8%. Moreover, when pretrained only on lead-I data for classification tasks, CLEF performs comparably to the state-of-the-art ECGFounder, which was trained in a supervised manner. Overall, CLEF enables more accurate and scalable single-lead ECG analysis, advancing remote health monitoring. Code and pretrained CLEF models are available at: github.com/Nokia-Bell-Labs/ecg-foundation-model.

心电图对比学习自监督远程监测

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。