Broad Institute (ML4H)
Self-supervised ECG representation learned purely from patient identity: the model is trained so that ECGs from the same patient, recorded at different times, map to nearby points in latent space, with no other labels required. Linear models trained on these representations showed a 51% average performance gain over training from scratch across sex classification, age regression, LVH detection, and AF detection. Developed by the Broad Institute's ML4H group and trained on 3.2 million private ECGs from Massachusetts General Hospital; 12-lead, lead-I-only, and lead-II-only checkpoints are all released.
Architecture
CNN (1D)
1D CNN ECG encoder trained with a patient-identity contrastive (SimCLR-style) objective; outputs 320-dim representations
Framework
TensorFlow
Added to catalog
2026-07-10
GPL 3.0
License for model weights only. Associated code may be licensed seperately, check code source for specific terms.
Massachusetts General Hospital ECG Cohort (PCLR)
3,229,408 ECGs from 404,929 patients (patients with only 1 ECG excluded)
Embedding