摘要
人工智能(AI)在心电图(ECG)分析和心血管疾病检测方面展现出巨大潜力。然而,由于个体差异和心电图诊断的多样性,开发通用的 AI-ECG 模型一直面临挑战,这限制了现有模型在特定诊断任务和数据集中的应用。此外,目前的 AI-ECG 模型在单导联和 12 导联心电图之间难以实现相当的性能,限制了 AI-ECG 在便携式和可穿戴心电图设备中的应用。为了解决这些限制,我们引入了心电图基础模型(ECGFounder),这是一个通用模型,它利用来自心脏病学专家的真实世界心电图注释来扩展心电图分析的诊断能力。ECGFounder 在超过 1000 万份来自哈佛-埃默里心电图数据库的心电图上进行训练,这些心电图包含 150 个标签类别,从而能够通过心电图分析进行全面的心血管疾病诊断。该模型旨在既能有效地开箱即用,又能针对下游任务进行微调,从而最大限度地提高可用性。更重要的是,我们将它的应用扩展到单导联心电图,能够进行复杂情况诊断,并在移动和远程监测场景中支持各种下游任务。实验结果表明,ECGFounder 在内部验证集中对 12 导联和单导联心电图都取得了专家级性能,同时在外部验证集中也展现出强大的分类性能和跨诊断的泛化能力。当进行微调时,ECGFounder 在人口统计学检测、临床事件检测和跨模态心律诊断方面优于基线模型。训练后的模型和数据将在发表后通过 bdsp.io 公开发布。我们的代码可在 https://github.com/bdsp-core/ECGFounder 获取。