CVAI Catalog

·

View Catalog

tune

8 models found

·

7 public code

·

6 public weights

ECG Scalogram-Phasogram Fusion CNN

Sapienza University of Rome · 2024

code

Training code public

CNN-based arrhythmia classifier that fuses the magnitude (scalogram) and phase (phasogram) of the continuous wavelet transform (CWT) of ECG heartbeats, rather than relying on magnitude information alone as most prior 2D-representation approaches do. Several fusion strategies (input-level, intermediate-layer, and output-level fusion) were compared on the public PhysioNet MIT-BIH Arrhythmia database. Despite a simple CNN architecture, the best fusion strategy achieved about 98.5% overall accuracy, 98.5% sensitivity and 95.6% specificity, competitive with more complex state-of-the-art approaches.

Single-lead ECG

Filter by Modality:
ECG

Multi-label classification

Filter by Task Type:
Classification

CNN (2D)

Filter by Architecture:
Convolutional (CNN)

Keras

Filter by Framework:
TensorFlow / Keras


Model ID: 0143

·

Subject Count: 47

QTcNet

University of Muenster (Institute of Medical Informatics) · 2025

graph_1

Code & model weights public

Regression-based convolutional neural network that directly estimates the heart-rate-corrected QT interval (QTc) from a 12-lead ECG, aiming to improve on automated QTc measurements from commercial ECG systems, which often diverge from expert readings. QTcNet was trained on 120,300 algorithm-labeled ECGs from an internal hospital cohort and the public MIMIC-IV database (after correcting for a vendor-specific measurement bias), and evaluated against expert QTc measurements in three independent external cohorts (PTB Diagnostic ECG Database, QTcMS, and ECGRDVQ). It roughly halved the mean absolute error compared with standard ECG analysis software (from 23.4ms to 13.4ms across external validation cohorts), with explainability analysis confirming the model focuses on physiologically plausible QRS-onset and T-offset regions.

12-lead ECG

Filter by Modality:
ECG

Long QT syndrome (LQTS)

Filter by Disease / Trait:
Arrhythmia

Regression

Filter by Task Type:
Regression

CNN (2D)

Filter by Architecture:
Convolutional (CNN)

PyTorch

Filter by Framework:
PyTorch


Model ID: 0135

AIRE (AI-ECG Risk Estimation)

Imperial College London (Sau, Ng et al.) · 2024

lock

Code & model weights private

Multi-outcome AI-ECG risk-estimation platform that turns a single 12-lead ECG into a patient-specific survival curve, using a residual convolutional neural network trained with a discrete-time survival loss to predict not just risk but time-to-event. Beyond all-cause and cardiovascular mortality, separately fine-tuned heads predict future ventricular arrhythmia, complete heart block, atrial fibrillation, atherosclerotic cardiovascular disease, heart failure, and (in follow-up work) hypertension -- all from a single resting ECG, including in ECGs a cardiologist would read as normal. Derived on 1.16 million ECGs from 189,539 Beth Israel Deaconess Medical Center patients and externally validated across transnational cohorts in the USA, Brazil (CODE), and the UK (UK Biobank). A variational autoencoder and genome/phenome-wide association analyses were used to show the model's predictions track biologically plausible features (QRS morphology, LV structure/function, and loci linked to cardiac structure, QT interval, and biological aging). NHS trials of AIRE were planned for late 2025. No public code or model weights have been released.

12-lead ECG

Filter by Modality:
ECG

Mortality

Filter by Disease / Trait:
Prognosis & Aging

Atrial fibrillation

Filter by Disease / Trait:
Arrhythmia

Atherosclerotic cardiovascular disease (ASCVD) risk

Filter by Disease / Trait:
Prognosis & Aging

Heart failure

Filter by Disease / Trait:
Cardiac Function & Hemodynamics

Regression

Filter by Task Type:
Regression

Binary classification

Filter by Task Type:
Classification

CNN (1D)

Filter by Architecture:
Convolutional (CNN)


Model ID: 0089

·

Subject Count: 189,539

DeepECG-SL

Montreal Heart Institute (HeartWise.AI) (Avram et al.) · EfficientNetV2 (supervised) · 2026

graph_1

Code & model weights public

Supervised EfficientNetV2-based 12-lead ECG model trained on over 1 million ECGs from the Montreal Heart Institute to predict 77 cardiac conditions derived from American Heart Association recommendations, plus fine-tuned digital-biomarker heads for reduced LVEF, 5-year atrial-fibrillation risk, and long-QT-syndrome (LQTS) detection/genotyping. Validated on 881,403 ECGs across 11 geographically diverse cohorts (4 public, 7 private health systems), achieving AUROCs above 0.98 for the 77-condition interpretation task while being 60x smaller and 29x faster at inference than its self-supervised DeepECG-SSL counterpart, with up to 9.7x lower CO2 emissions on equivalent tasks.

12-lead ECG

Filter by Modality:
ECG

General Purpose / Multi-task

Filter by Disease / Trait:
General / Foundation

LV systolic dysfunction (LVSD)

Filter by Disease / Trait:
Cardiac Function & Hemodynamics

Atrial fibrillation

Filter by Disease / Trait:
Arrhythmia

Long QT syndrome (LQTS)

Filter by Disease / Trait:
Arrhythmia

Multi-label classification

Filter by Task Type:
Classification

Binary classification

Filter by Task Type:
Classification

Multi-class classification

Filter by Task Type:
Classification

CNN (2D)

Filter by Architecture:
Convolutional (CNN)

PyTorch

Filter by Framework:
PyTorch

Apache 2.0

Filter by License:
Permissive


Model ID: 0070

·

Subject Count: 184,210

DeepECG-SSL

Montreal Heart Institute (HeartWise.AI) (Avram et al.) · EfficientNetV2 (self-supervised) · 2026

graph_1

Code & model weights public

Self-supervised EfficientNetV2-based 12-lead ECG foundation model pretrained via contrastive learning and masked-lead modeling on 1.9 million ECGs (Montreal Heart Institute plus CODE-15% and MIMIC-IV), then fine-tuned for the same 77-condition ECG interpretation task and digital-biomarker extraction as DeepECG-SL. Outperforms the supervised counterpart on label-scarce digital-biomarker tasks, with the largest gains on LQTS genotype classification (AUROC 0.931 vs. 0.850, n=127 ECGs) and 5-year atrial-fibrillation risk (AUROC 0.742 vs. 0.734, n=132,050 ECGs), and outperforms ECG-FM and ECGFounder on shared external diagnostic classes.

12-lead ECG

Filter by Modality:
ECG

General Purpose / Multi-task

Filter by Disease / Trait:
General / Foundation

LV systolic dysfunction (LVSD)

Filter by Disease / Trait:
Cardiac Function & Hemodynamics

Atrial fibrillation

Filter by Disease / Trait:
Arrhythmia

Long QT syndrome (LQTS)

Filter by Disease / Trait:
Arrhythmia

Multi-label classification

Filter by Task Type:
Classification

Binary classification

Filter by Task Type:
Classification

Multi-class classification

Filter by Task Type:
Classification

CNN (2D)

Filter by Architecture:
Convolutional (CNN)

PyTorch

Filter by Framework:
PyTorch


Model ID: 0071

·

Subject Count: 345,562

ECG-XPLAIM

National and Kapodistrian University of Athens / ETH Zurich (Pantelidis, Ruiperez-Campillo et al.) · 2025

graph_1

Code & model weights public

Explainable Inception-style 1D CNN for multi-label arrhythmia detection from 12-lead ECGs, integrating Grad-CAM visualization to highlight the waveform segments driving each prediction. Trained on MIMIC-IV-ECG and externally validated on PTB-XL across atrial fibrillation, sinus tachycardia, conduction disturbances (RBBB/LBBB/LAFB), long QT, Wolff-Parkinson-White pattern, and paced-rhythm detection, with all metrics exceeding 90% internally and strong generalization on external validation.

12-lead ECG

Filter by Modality:
ECG

Atrial fibrillation

Filter by Disease / Trait:
Arrhythmia

Multi-class classification

Filter by Task Type:
Classification

Multi-label classification

Filter by Task Type:
Classification

CNN (1D)

Filter by Architecture:
Convolutional (CNN)

TensorFlow

Filter by Framework:
TensorFlow / Keras

CC BY 4.0

Filter by License:
Open — Attribution


Model ID: 0064

·

Subject Count: 161,352

TolerantECG

FPT Software AI Center / University of Arkansas (Nguyen et al.) · 2025

graph_1

Code & model weights public

ECG foundation model designed to remain accurate when leads are missing or signals are noisy. A 1D ConvNeXt V2 encoder is trained with a dual-mode self-distillation objective (separate lead-missing and noise "teachers") alongside contrastive alignment to detailed diagnostic-criteria text reports retrieved via a lightweight, LLM-free "Cardiac Feature Retrieval" module. Consistently ranks best or second-best across PTB-XL diagnostic tasks and MIT-BIH arrhythmia classification under original, noisy, lead-missing, and combined-corruption conditions. Developed by FPT Software AI Center and the University of Arkansas.

12-lead ECG

Filter by Modality:
ECG

General Purpose / Multi-task

Filter by Disease / Trait:
General / Foundation

Multi-label classification

Filter by Task Type:
Classification

Embedding

Filter by Task Type:
Representation Learning

CNN (1D)

Filter by Architecture:
Convolutional (CNN)

PyTorch

Filter by Framework:
PyTorch

CC BY-NC-SA 4.0

Filter by License:
Non-commercial / Research-only


Model ID: 0057

·

Subject Count: 180,237

ECG2AF

Broad Institute (ML4H) · ecg2af_quintuplet_v2024_01_13 (updated 2025) · 2022

graph_1

Code & model weights public

Multi-task 12-lead ECG model with output heads for incident atrial-fibrillation risk (as a survival curve), incident mortality risk, prevalent AF classification, sex classification, and age regression. Built on a 1D CNN over the raw waveform, and developed by the Broad Institute's ML4H group as a successor to their ECG-AI model published in Circulation. Trained on ECGs from UK Biobank and Massachusetts General Hospital, neither of which is publicly released.

12-lead ECG

Filter by Modality:
ECG

Atrial fibrillation

Filter by Disease / Trait:
Arrhythmia

Mortality

Filter by Disease / Trait:
Prognosis & Aging

Multi-label classification

Filter by Task Type:
Classification

CNN (1D)

Filter by Architecture:
Convolutional (CNN)

TensorFlow

Filter by Framework:
TensorFlow / Keras

GPL 3.0

Filter by License:
Copyleft


Model ID: 0016

·

Subject Count: 45,770