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

基于超过1000万条记录构建的心电图基础模型及跨多领域外部评估

An Electrocardiogram Foundation Model Built on over 10 Million Recordings with External Evaluation across Multiple Domains

National Institute of Health Data Science, Peking University(北京大学国家健康数据科学研究院);Department of Cardiology, Massachusetts General Hospital(麻省总医院心内科);Department of Biomedical Informatics, School of Medicine, Emory University(埃默里大学医学院生物医学信息学系);Department of Biomedical Engineering, Georgia Institute of Technology(佐治亚理工学院生物医学工程系);Harvard Medical School(哈佛医学院);Department of Neurology, Beth Israel Deaconess Medical Center(贝斯以色列医疗中心神经科);Department of Computing, Data Science Institute, Imperial College London(伦敦帝国学院计算系、数据科学研究所);Zhongshan School of Medicine, Sun Yat-sen University(中山医学院,孙逸仙大学);Institute of Medical Technology, Peking University Health Science Center(北京大学医学技术研究所,北京大学医学部);Institute for Artificial Intelligence, Peking University(北京大学人工智能研究院)

Jun Li, Aaron Aguirre, Junior Moura, Che Liu, Lanhai Zhong, Chenxi Sun, Gari Clifford, Brandon Westover, Shenda Hong

AI总结:

提出ECGFounder,基于超千万心电图训练的基础模型,实现专家级诊断性能,支持单导联及多领域泛化,可微调用于多种下游任务。

摘要:

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

英文摘要:

Artificial intelligence (AI) has demonstrated significant potential in ECG analysis and cardiovascular disease assessment. Recently, foundation models have played a remarkable role in advancing medical AI. The development of an ECG foundation model holds the promise of elevating AI-ECG research to new heights. However, building such a model faces several challenges, including insufficient database sample sizes and inadequate generalization across multiple domains. Additionally, there is a notable performance gap between single-lead and multi-lead ECG analyses. We introduced an ECG Foundation Model (ECGFounder), a general-purpose model that leverages real-world ECG annotations from cardiology experts to broaden the diagnostic capabilities of ECG analysis. ECGFounder was trained on over 10 million ECGs with 150 label categories from the Harvard-Emory ECG Database, enabling comprehensive cardiovascular disease diagnosis through ECG analysis. The model is designed to be both an effective out-of-the-box solution, and a to be fine-tunable for downstream tasks, maximizing usability. Importantly, we extended its application to lower rank ECGs, and arbitrary single-lead ECGs in particular. ECGFounder is applicable to supporting various downstream tasks in mobile monitoring scenarios. Experimental results demonstrate that ECGFounder achieves expert-level performance on internal validation sets, with AUROC exceeding 0.95 for eighty diagnoses. It also shows strong classification performance and generalization across various diagnoses on external validation sets. When fine-tuned, ECGFounder outperforms baseline models in demographic analysis, clinical event detection, and cross-modality cardiac rhythm diagnosis. The trained model and data will be publicly released upon publication through the bdsp.io. Our code is available at https://github.com/PKUDigitalHealth/ECGFounder

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

↑