Universidade Federal de Minas Gerais (UFMG) / Uppsala University / EPFL (Ribeiro et al.)
Multi-label 12-lead ECG diagnosis model submitted to the PhysioNet/Computing in Cardiology Challenge 2020, built on the same residual 1D CNN family as the authors' earlier Nature Communications model but retrained and validated across the challenge's large, multi-institutional pooled training set (CPSC2018, China 12-Lead ECG Database, St. Petersburg INCART, PTB and PTB-XL, and the Georgia 12-Lead ECG Database). The model uses an unsupervised pretraining stage -- predicting unseen samples of a partially masked ECG signal -- before supervised fine-tuning to jointly detect nine diagnostic classes (atrial fibrillation, first-degree AV block, left and right bundle branch block, normal rhythm, premature atrial/ventricular contraction, and ST-segment depression/elevation). The 2020 Challenge was notable for requiring every team to publicly release both their trained model weights and full training code, making this one of relatively few 12-lead ECG classifiers with an end-to-end reproducible public pipeline.
Architecture
CNN (1D)
Residual 1D convolutional neural network (matching the architecture family used in the authors' earlier automatic-ecg-diagnosis model) with an unsupervised masked-signal-prediction pretraining stage followed by supervised multi-label fine-tuning
Framework
PyTorch
Added to catalog
2026-08-12
MIT
License for model weights only. Associated code may be licensed seperately, check code source for specific terms.
12-lead ECG multi-label arrhythmia-classification dataset released for the 2018 China Physiological Signal Challenge.
Georgia 12-lead ECG Challenge (G12EC) database, part of the PhysioNet/CinC 2020-2021 Challenge training data.
268 subjects with 12-lead ECGs plus patient information; commonly used as a small external-validation cohort for ECG-based diagnosis models.
52% male / 48% female; age range 0-95 (median ~62). 21,837 10-second 12-lead ECG records.
75 annotated 30-minute 12-lead ambulatory ECG recordings from 32 subjects, collected at the St. Petersburg Institute of Cardiological Technics; one of three corpora (with CODE and Chapman-Shaoxing-Ningbo, ~8M ECGs total) used to pretrain xECG.
Multi-label detection of 9 diagnostic classes (atrial fibrillation, first-degree AV block, left/right bundle branch block, normal, premature atrial/ventricular contraction, ST depression/elevation) from a 12-lead ECG