Daily Arxiv

This is a page that curates AI-related papers published worldwide.
All content here is summarized using Google Gemini and operated on a non-profit basis.
Copyright for each paper belongs to the authors and their institutions; please make sure to credit the source when sharing.

CREMA: A Contrastive Regularized Masked Autoencoder for Robust ECG Diagnostics across Clinical Domains

Created by
  • Haebom

Author

Junho Song, Jong-Hwan Jang, Dong-Gyun Hong, Joon-myoung Kwon, Yong-Yeon Jo

Outline

Electrocardiogram (ECG) diagnosis remains challenging due to limited labeled data and the challenge of capturing subtle yet clinically relevant rhythm and morphological changes. In this paper, we present the Contrastive Regularized Masked Autoencoder (CREMA), a foundational model for 12-lead ECGs designed to learn generalizable representations through self-supervised pretraining. CREMA combines generative learning with contrastive regularization via the Contrastive Regularized Masked Autoencoder (MAE) loss and utilizes the Signal Transformer (SiT) architecture to capture both local waveform details and global temporal dependencies. We evaluate CREMA on benchmark datasets and in real-world clinical settings, including deployment scenarios with significant distributional shift. CREMA outperforms both supervised learning baseline models and existing self-supervised learning models in linear probing and fine-tuning evaluations. Its robustness in real-world settings is demonstrated by its superior performance across diverse clinical domains, particularly in emergency care. These results demonstrate that CREMA serves as a scalable and robust foundational model for ECG diagnosis, supporting downstream applications in heterogeneous and high-risk clinical settings.

Takeaways, Limitations

Takeaways:
CREMA, an electrocardiogram diagnosis model based on self-supervised learning, is presented.
Superior performance compared to existing supervised learning-based models and self-supervised learning models.
Outstanding performance and robustness in real-world clinical environments and across a wide range of clinical domains.
Presenting the potential as a scalable and stable ECG diagnostic baseline model.
Support for downstream applications.
Limitations:
This paper does not explicitly mention Limitations. Further research may be needed to improve the model's generalization performance, improve its performance on specific diseases, and enhance its interpretability.
👍