English

SuPreME: A Supervised Pre-training Framework for Multimodal ECG Representation Learning

Signal Processing 2025-09-22 v4 Artificial Intelligence Computation and Language Machine Learning

Abstract

Cardiovascular diseases are a leading cause of death and disability worldwide. Electrocardiogram (ECG) is critical for diagnosing and monitoring cardiac health, but obtaining large-scale annotated ECG datasets is labor-intensive and time-consuming. Recent ECG Self-Supervised Learning (eSSL) methods mitigate this by learning features without extensive labels but fail to capture fine-grained clinical semantics and require extensive task-specific fine-tuning. To address these challenges, we propose SuPreME\textbf{SuPreME}, a Su\textbf{Su}pervised Pre\textbf{Pre}-training framework for M\textbf{M}ultimodal E\textbf{E}CG representation learning. SuPreME is pre-trained using structured diagnostic labels derived from ECG report entities through a one-time offline extraction with Large Language Models (LLMs), which help denoise, standardize cardiac concepts, and improve clinical representation learning. By fusing ECG signals with textual cardiac queries instead of fixed labels, SuPreME enables zero-shot classification of unseen conditions without further fine-tuning. We evaluate SuPreME on six downstream datasets covering 106 cardiac conditions, achieving superior zero-shot AUC performance of 77.20%77.20\%, surpassing state-of-the-art eSSLs by 4.98%4.98\%. Results demonstrate SuPreME's effectiveness in leveraging structured, clinically relevant knowledge for high-quality ECG representations.

Keywords

Cite

@article{arxiv.2502.19668,
  title  = {SuPreME: A Supervised Pre-training Framework for Multimodal ECG Representation Learning},
  author = {Mingsheng Cai and Jiuming Jiang and Wenhao Huang and Che Liu and Rossella Arcucci},
  journal= {arXiv preprint arXiv:2502.19668},
  year   = {2025}
}

Comments

Findings of The 2025 Conference on Empirical Methods in Natural Language Processing (EMNLP 2025)

R2 v1 2026-06-28T21:59:30.735Z