D-BETA: Boosting Masked ECG-Text Auto-Encoders as Discriminative Learners

ICML 2025

Singapore Management University Eindhoven University of Technology
Illustration of our contrastive masked ECG-language modeling technique

Illustration of our contrastive masked ECG-language modeling technique.

Abstract

Accurate interpretation of Electrocardiogram (ECG) signals is pivotal for diagnosing cardiovascular diseases. Integrating ECG signals with their accompanying textual reports holds immense potential to enhance clinical diagnostics through the combination of physiological data and qualitative insights. However, this integration faces significant challenges due to inherent modality disparities and the scarcity of labeled data for robust cross-modal learning. To address these obstacles, we propose D-BETA, a novel framework that pre-trains ECG and text data using a contrastive masked auto-encoder architecture. D-BETA uniquely combines the strengths of generative with enhanced discriminative capabilities to achieve robust cross-modal representations. This is accomplished through masked modality modeling, specialized loss functions, and an improved negative sampling strategy tailored for cross-modal alignment. Extensive experiments on five public datasets across diverse downstream tasks demonstrate that D-BETA significantly outperforms existing methods, achieving 15% and 2% increases in linear probing and zero-shot performance over state-of-the-art models, respectively. These results highlight the effectiveness of D-BETA, underscoring its potential to advance automated clinical diagnostics through multi-modal representations.

-->

BibTeX


        @misc{pham2025dbeta,
        title={Boosting Masked ECG-Text Auto-Encoders as Discriminative Learners}, 
        author={Hung Manh Pham and Aaqib Saeed and Dong Ma},
        year={2025},
        eprint={2410.02131},
        archivePrefix={arXiv},
        primaryClass={cs.LG},
        url={https://arxiv.org/abs/2410.02131}}