arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2512.02180cs.LGcs.AI

基于临床指导的对比学习用于心电图基础模型

CLEF: Clinically-Guided Contrastive Learning for Electrocardiogram Foundation Models

University College London(伦敦大学学院);Nokia Bell Labs(诺基亚贝尔实验室);University of Glasgow(格拉斯哥大学);Heart Hospital, Tampere University(塔尔皮奥大学心医院)

Yuxuan Shu, Peter H. Charlton, Fahim Kawsar, Jussi Hernesniemi, Mohammad Malekzadeh

AI总结:

CLEF通过临床指导的对比学习方法,在未标注的ECG数据上预训练单导联基础模型,提升分类和回归任务的性能,实现更准确的远程健康监测。

摘要:

心电图(ECG)是心血管健康诊断中的关键工具。单导联ECG记录已被集成到临床级和消费级可穿戴设备中。虽然在未标记的ECG上对基础模型进行自监督预训练可以提高诊断性能,但现有方法并未利用临床元数据中的领域知识。我们提出了一种新的对比学习方法,利用已建立的临床风险评分来适应性地加权负样本对:临床指导的对比学习。该方法使ECG嵌入的相似性与患者之间具有临床意义的差异对齐,并具有处理缺失元数据的显式机制。在MIMIC-IV数据集的12导联ECG数据上,我们预训练了单导联ECG基础模型,共三个尺度,统称为CLEF,仅使用常规收集的元数据,而无需每个样本的ECG注释。我们评估了CLEF在18个临床分类和回归任务上的7个留出数据集,并与5个基础模型基线和3个自监督算法进行基准测试。当在12导联ECG数据上预训练并在导联I数据上测试时,CLEF在分类任务中优于自监督基础模型基线:中等规模的CLEF在分类任务中平均AUROC提升至少2.6%,在回归任务中平均MAE减少至少3.2%。与现有自监督学习算法相比,CLEF的平均AUROC提高了至少1.8%。此外,当仅在导联I数据上预训练用于分类任务时,CLEF的表现与最先进的ECGFounder相当,后者是通过监督方式训练的。总体而言,CLEF使单导联ECG分析更加准确和可扩展,推动了远程健康监测的发展。代码和预训练的CLEF模型可在:github.com/Nokia-Bell-Labs/ecg-foundation-model获取。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑