CVAI Catalog

·

View Catalog

tune

3 models found

·

2 public code

·

2 public weights

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

ECG-age ResNet (ecg-age-prediction)

Uppsala University / UFMG (Lima et al.) · 2021

graph_1

Code & model weights public

1D residual neural network that predicts a patient's age directly from a 12-lead ECG; the gap between this predicted 'ECG age' and true chronological age is used as a biomarker of cardiovascular risk and mortality. Trained on the CODE-15% Brazilian ECG dataset, with the original R² of 0.71 later reproduced (R² = 0.70) in an independent German validation cohort. Developed by researchers at Uppsala University and UFMG.

12-lead ECG

Filter by Modality:
ECG

Cardiac aging / biological age

Filter by Disease / Trait:
Prognosis & Aging

Regression

Filter by Task Type:
Regression

CNN (1D)

Filter by Architecture:
Convolutional (CNN)

PyTorch

Filter by Framework:
PyTorch

CC BY 4.0

Filter by License:
Open — Attribution


Model ID: 0011

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