Contrasting by Augmented Patient Electrocardiograms to Learn Representations for a Foundation Model
摘要
Electrocardiograms (ECGs) record the heart’s electrical activity, offering valuable diagnostic and prognostic insights. We introduce Contrasting by Augmented Patient Electrocardiograms (CAPE), a foundation model trained using patient contrastive learning enhanced with temporal augmentations, on over six million unlabeled ECGs from diverse cohorts across three continents. We demonstrate CAPE’s effectiveness to generate a meaningful generic representation and validate its performance on an open-source external cohort, where it surpasses existing pretraining methods and fully supervised deep networks.