评估心电图大模型的泛化能力,不止看准确率。
Looking Beyond Accuracy: A Holistic Benchmark of ECG Foundation Models
- 用嵌入表示分析+SHAP/UMAP,全面评估模型内在特性。
- 在跨洲数据集和数据稀缺场景下验证多个心电图大模型。
- 揭示模型表征结构,帮助理解其医疗应用可靠性。
心电图(ECG)是一种成本低、易获取且广泛应用的诊断工具。随着基础模型(FMs)的发展,人工智能辅助心电图解读开始演进,因其可依赖嵌入实现任务间复用。然而,在医疗等高风险领域,负责任地使用基础模型需严格评估其嵌入的泛化能力。尽管已有研究关注心电图专家基础模型的基准测试,但多聚焦下游性能。为填补此空白,本研究提出一种深入、全面的基准框架,重点针对心电图专家模型。我们引入结合性能评估与表示层面分析的基准方法,利用SHAP和UMAP技术。基于该方法,对多种通过先进预训练技术在不同跨大陆数据集及数据可用性设置下训练的心电图专家模型进行广泛评估,涵盖数据稀缺这一真实医疗场景中常见情况。实验结果表明,该基准协议能揭示心电图专家模型的嵌入模式,深化对模型表征结构与泛化能力的理解。
原文摘要 · Abstract (English)
The electrocardiogram (ECG) is a cost-effective, highly accessible and widely employed diagnostic tool. With the advent of Foundation Models (FMs), the field of AI-assisted ECG interpretation has begun to evolve, as they enable model reuse across different tasks by relying on embeddings. However, to responsibly employ FMs, it is crucial to rigorously assess to which extent the embeddings they produce are generalizable, particularly in error-sensitive domains such as healthcare. Although prior works have already addressed the problem of benchmarking ECG-expert FMs, they focus predominantly on the evaluation of downstream performance. To fill this gap, this study aims to find an in-depth, comprehensive benchmarking framework for FMs, with a specific focus on ECG-expert ones. To this aim, we introduce a benchmark methodology that complements performance-based evaluation with representation-level analysis, leveraging SHAP and UMAP techniques. Furthermore, we rely on the methodology for carrying out an extensive evaluation of several ECG-expert FMs pretrained via state-of-the-art techniques over different cross-continental datasets and data availability settings; this includes ones featuring data scarcity, a fairly common situation in real-world medical scenarios. Experimental results show that our benchmarking protocol provides a rich insight of ECG-expert FMs' embedded patterns, enabling a deeper understanding of their representational structure and generalizability.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。