用自适应微调让心电图大模型更准识别多种心脏病,适合临床筛查。
Domain-Adapted Fine-Tuning of ECG Foundation Models for Multi-Label Structural Heart Disease Screening

- 先在心电图上做领域自监督预训练,再选关键层微调。
- 最高宏平均AUC达0.8509,固定阈值下F1达0.3691。
- 适合心电图辅助心脏超声筛查,尤其资源有限场景。
经胸超声心动图是确诊结构性心脏病(SHD)的金标准,但一线筛查受限于成本、流程负担和专科医生不足。我们评估了开放预训练心电图(ECG)基础模型能否支持基于超声确诊的多标签SHD检测,使用公开的EchoNext Mini-Model基准。目标检测六种超声异常:左室射血分数降低、左室壁增厚、主动脉瓣狭窄、二尖瓣反流、三尖瓣反流及右室收缩功能障碍。在统一流程下,对比了人工特征+梯度提升、端到端波形学习和迁移自开放ECG基础模型的表现。随后,对一个ECG基础模型(ECG-FM)在EchoNext波形上进行领域内自监督适应,再进行选择性监督微调,并评估判别能力与适配成本的权衡。适应后的ECG-FM模型整体表现最佳:峰值宏平均AUROC为0.8509,宏平均AUPRC为0.4297;参数高效操作点保持AUROC 0.8501,且达到最高固定阈值宏F1 0.3691。融合协变量未提升阈值无关判别性能;评估的LoRA、替代骨干网络及混合基础模型策略均未超越最优单一骨干模型。结果表明,对于基于心电图的病例发现与超声分诊,结合目标域自监督适应与预训练模型选择性更新是最有效的迁移策略。
原文摘要 · Abstract (English)
Transthoracic echocardiography is the reference standard for confirming structural heart disease (SHD), but first-line screening is limited by cost, workflow burden, and specialist availability. We evaluated whether open pretrained electrocardiogram (ECG) foundation models can support echo-confirmed multi-label SHD detection using the public EchoNext Mini-Model benchmark. Six echocardiography-derived abnormalities were targeted: reduced left ventricular ejection fraction, increased left ventricular wall thickness, aortic stenosis, mitral regurgitation, tricuspid regurgitation, and right ventricular systolic dysfunction. Under a common pipeline, we compared engineered ECG features with gradient boosting, end-to-end waveform learning from scratch, and transfer from open ECG foundation models. We then applied in-domain self-supervised adaptation of an ECG foundation model (ECG-FM) on EchoNext waveforms followed by selective supervised fine-tuning, and evaluated trade-offs between discrimination and adaptation cost. Adapted ECG-FM models achieved the best overall performance: peak macro-AUROC 0.8509 and macro-AUPRC 0.4297, while a parameter-efficient operating point preserved AUROC (0.8501) and attained the highest fixed-threshold macro-F1 0.3691. Late fusion with covariates did not improve threshold-independent discrimination, and evaluated LoRA, alternative backbones, and mixture-of-foundations strategies did not surpass the best adapted single-backbone models. These results indicate that for ECG-based case finding and echocardiography triage, combining target-domain self-supervised adaptation with selective supervised updating of a pretrained ECG backbone is the most effective transfer strategy.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。