Tian Yuanyuan, Li Zhiyuan, Jin Yanrui, Wang Mengxiao, Wei Xiaoyang, Zhao Liqun, Liu Yunqing, Liu Jinlei, Liu Chengliang
State Key Laboratory of Mechanical System and Vibration, School of Mechanical Engineering, Shanghai Jiao Tong University, Shanghai 200240, China; MoE Key Lab of Artificial Intelligence, AI Institute, Shanghai Jiao Tong University, Shanghai 200240, China.
State Key Laboratory of Mechanical System and Vibration, School of Mechanical Engineering, Shanghai Jiao Tong University, Shanghai 200240, China; MoE Key Lab of Artificial Intelligence, AI Institute, Shanghai Jiao Tong University, Shanghai 200240, China.
Cell Rep Med. 2024 Dec 17;5(12):101875. doi: 10.1016/j.xcrm.2024.101875.
We propose a knowledge-enhanced electrocardiogram (ECG) diagnosis foundation model (KED) that utilizes large language models to incorporate domain-specific knowledge of ECG signals. This model is trained on 800,000 ECGs from nearly 160,000 unique patients. Despite being trained on single-center data, KED demonstrates exceptional zero-shot diagnosis performance across various regions, including different locales in China, the United States, and other regions. This performance spans across all age groups for various conditions such as morphological abnormalities, rhythm abnormalities, conduction blocks, hypertrophy, myocardial ischemia, and infarction. Moreover, KED exhibits robust performance on diseases it has not encountered during its training. When compared to three experienced cardiologists on real clinical datasets, the model achieves comparable performance in zero-shot diagnosis of seven common clinical ECG types. We concentrate on the zero-shot diagnostic capability and the generalization performance of the proposed ECG foundation model, particularly in the context of external multi-center data and previously unseen disease.
我们提出了一种知识增强型心电图(ECG)诊断基础模型(KED),该模型利用大语言模型纳入心电图信号的领域特定知识。该模型在来自近160,000名独特患者的800,000份心电图上进行训练。尽管是在单中心数据上进行训练,但KED在各个地区都表现出出色的零样本诊断性能,包括中国、美国和其他地区的不同地点。这种性能涵盖了所有年龄组,适用于各种情况,如形态异常、节律异常、传导阻滞、肥大、心肌缺血和梗死。此外,KED在其训练过程中未遇到的疾病上也表现出强大的性能。在真实临床数据集上与三位经验丰富的心脏病专家进行比较时,该模型在七种常见临床心电图类型的零样本诊断中达到了可比的性能。我们专注于所提出的心电图基础模型的零样本诊断能力和泛化性能,特别是在外部多中心数据和以前未见过的疾病的背景下。