LLM2D
基于超过1000万记录构建的心电图基础模型,并在多个领域进行外部评估
An Electrocardiogram Foundation Model Built on over 10 Million Recordings with External Evaluation across Multiple Domains
作者: Jun Li, Aaron Aguirre, Junior Moura, Che Liu, Lanhai Zhong, Chenxi Sun, Gari Clifford, Brandon Westover, Shenda Hong
发布日期: 4/4/2025
arXiv ID: oai:arXiv.org:2410.04133v3

摘要

arXiv:2410.04133v3 宣布类型:替换交叉 摘要:人工智能(AI)在心电图(ECG)分析和心血管疾病评估方面展示了巨大的潜力。近年来,基础模型在推动医疗AI方面发挥了重要作用。开发ECG基础模型有望将AI-ECG研究提升到新的高度。然而,建立这样的模型面临着多个挑战,包括数据库样本量不足和在多个领域泛化的不足。此外,单导联和多导联心电图分析之间存在明显的性能差距。我们引入了ECG基础模型(ECGFounder),这是一种通用模型,通过利用心脏病学专家的真实世界心电图注解,增强了心电图分析的诊断能力。ECGFounder在哈佛-埃默里心电图数据库上进行了超过100万份心电图的训练,这包括150个标签类别,从而通过心电图分析实现全面的心血管疾病诊断。该模型旨在不仅作为有效的开箱即用解决方案,而且可以针对下游任务进行微调,以最大化其实用性。更重要的是,我们将其应用扩展到了较低等级的心电图,尤其是任意单导联心电图。ECGFounder适用于移动监测场景下的多种下游任务。实验结果显示,ECGFounder在内部验证集上达到了专家级性能,针对80种诊断的AUROC超过0.95。此外,在外部验证集上,ECGFounder在各种诊断分类和泛化方面也表现出强大的性能。经过微调后,ECGFounder在人口统计分析、临床事件检测和跨模态心律诊断方面优于基线模型。经过训练的模型和数据将在发表后通过bdsp.io公开发布。我们的代码可在https://github.com/PKUDigitalHealth/ECGFounder获取。