用临床数据提升心电图模型准确率与可解释性
Transferring Clinical Knowledge into ECGs Representation
- 通过多模态临床数据预训练心电图编码器,增强表示能力
- 在MIMIC-IV-ECG上达到接近全模态模型的多标签诊断性能
- 让模型从心电图直接预测实验室异常,提供生理可解释性
深度学习模型在心电图(ECG)分类中表现优异,但其黑箱特性阻碍了临床应用。为此,我们提出一种三阶段训练范式,将来自多模态临床数据(实验室检查、生命体征、生物特征)的知识迁移至单一心电图编码器。采用自监督联合嵌入预训练,使心电图表示融入临床上下文信息,推理时仅需心电图信号。此外,通过让模型从心电图嵌入直接预测相关实验室异常,间接实现输出解释。在MIMIC-IV-ECG数据集上,该模型在多标签诊断分类任务中优于仅使用信号的基线,并显著缩小了与需全部数据推理的全模态模型之间的性能差距。本工作展示了构建更准确、可信的心电图分类模型的实用有效方法,将抽象预测转化为基于生理学的解释,为人工智能安全融入临床流程提供了可行路径。
原文摘要 · Abstract (English)
Deep learning models have shown high accuracy in classifying electrocardiograms (ECGs), but their black box nature hinders clinical adoption due to a lack of trust and interpretability. To address this, we propose a novel three-stage training paradigm that transfers knowledge from multimodal clinical data (laboratory exams, vitals, biometrics) into a powerful, yet unimodal, ECG encoder. We employ a self-supervised, joint-embedding pre-training stage to create an ECG representation that is enriched with contextual clinical information, while only requiring the ECG signal at inference time. Furthermore, as an indirect way to explain the model's output we train it to also predict associated laboratory abnormalities directly from the ECG embedding. Evaluated on the MIMIC-IV-ECG dataset, our model outperforms a standard signal-only baseline in multi-label diagnosis classification and successfully bridges a substantial portion of the performance gap to a fully multimodal model that requires all data at inference. Our work demonstrates a practical and effective method for creating more accurate and trustworthy ECG classification models. By converting abstract predictions into physiologically grounded \emph{explanations}, our approach offers a promising path toward the safer integration of AI into clinical workflows.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。