CVAI Catalog

·

View Catalog

tune

11 models found

·

10 public code

·

5 public weights

ADTEP (Adversarial Deep Treatment Effect Prediction)

Zhejiang University / Chinese PLA General Hospital · 2020

code

Training code public

Adversarial deep learning model that predicts treatment effects for cardiology patients from structured electronic health record (EHR) data, aiming to forecast expected clinical outcomes of specific treatment choices given a patient's clinical status. Two autoencoders separately learn representations of patient characteristics and of the treatments given; an adversarial loss then encourages these representations to capture the correlational structure between a patient's status and the treatment received, improving downstream outcome prediction over non-adversarial baselines. Evaluated on two private cardiology EHR cohorts from a Chinese hospital, ADTEP modestly outperformed a non-adversarial ablation (DTEP) and classical baselines (logistic regression, SVM) at predicting major adverse cardiac events (MACE) after acute coronary syndrome (AUC 0.662 vs. 0.653/0.648/0.621) and at heart-failure outcome prediction.

Structured EHR

Filter by Modality:
Text & EHR

Major adverse cardiovascular events (MACE)

Filter by Disease / Trait:
Prognosis & Aging

Binary classification

Filter by Task Type:
Classification

Hybrid

Filter by Architecture:
Hybrid / Multi-branch


Model ID: 0136

Interpretable LightGBM CHD Risk Model

Huzhou Central Hospital (Zhejiang Chinese Medical University / Huzhou University) · 2025

code

Training code public

Interpretable coronary heart disease (CHD) risk prediction model based on the LightGBM gradient-boosting algorithm, combined with SHAP (SHapley Additive exPlanations) values to make individual risk predictions explainable to clinicians. Trained on the public BRFSS_2015 survey dataset and externally validated on the Framingham and Z-Alizadeh Sani datasets, the model reached 90.60% accuracy and 81.06% AUROC on BRFSS_2015, with SHAP analysis identifying age, smoking status, diabetes, hypertension, and high cholesterol as the most influential risk features. A companion CHD scoring system was built from the model to give clinicians a user-friendly risk-assessment tool.

Structured EHR

Filter by Modality:
Text & EHR

Atherosclerotic cardiovascular disease (ASCVD) risk

Filter by Disease / Trait:
Prognosis & Aging

Binary classification

Filter by Task Type:
Classification


Model ID: 0142

·

Subject Count: 253,680

CXR CVD-Risk

Massachusetts General Hospital / Harvard Medical School (Weiss, Raghu, Lu, Aerts et al.) · 2024

graph_1

Code & model weights public

Deep-learning model that estimates a patient's 10-year risk of major adverse cardiovascular events (MACE) directly from a single routine chest radiograph (CXR), intended as an opportunistic risk-assessment tool when the inputs needed for the standard ASCVD risk calculator (lipids, blood pressure, smoking status, etc.) are missing. A 2D convolutional network takes the CXR image alone as input and outputs a continuous 10-year MACE risk estimate. Developed on 147,801 CXRs from 40,718 participants in the PLCO cancer-screening trial and externally validated in 8,869 outpatients with unknown ASCVD risk and 2,132 with known risk at Mass General Brigham, CXR CVD-Risk identified people at elevated MACE risk (adjusted hazard ratio 1.73 for a >=7.5% predicted risk) and provided added discrimination beyond the traditional ASCVD score.

Chest Radiography

Filter by Modality:
Other Imaging

Major adverse cardiovascular events (MACE)

Filter by Disease / Trait:
Prognosis & Aging

Regression

Filter by Task Type:
Regression

CNN (2D)

Filter by Architecture:
Convolutional (CNN)

PyTorch

Filter by Framework:
PyTorch


Model ID: 0125

·

Subject Count: 40,718

CTO-PCI Success Predictor (Patch-UCTNet + Swin Transformer)

Beijing Anzhen Hospital, Capital Medical University / Sun Yat-Sen University · 2023

code

Training code public

End-to-end deep learning framework that predicts the procedural outcome of percutaneous coronary intervention (PCI) for chronic total occlusion (CTO) lesions directly from preprocedural coronary CT angiography, aiming to replace slower manual scoring systems (J-CTO, CT-RECTOR, KCCT). The pipeline first segments the coronary artery tree (Patch-UCTNet), detects candidate CTO lesions along the delineated vessel, extracts pathological lesion features with a Swin Transformer, and classifies two outcomes: successful guidewire crossing within 30 minutes and overall PCI success. In the original study, the model completed reconstruction and analysis 85% faster than manual scores (73.7s vs. 418-467s) and was more accurate than the manual CT-RECTOR, KCCT, and J-CTO_CCTA_ scores, reaching an AUROC of 0.97 on the internal test set and 0.96 on an independent external validation cohort (186 patients, 189 CTO lesions).

CT angiography

Filter by Modality:
Cardiac CT

Procedural planning / outcome (PCI, TAVI)

Filter by Disease / Trait:
Prognosis & Aging

Binary classification

Filter by Task Type:
Classification

Hybrid

Filter by Architecture:
Hybrid / Multi-branch

PyTorch

Filter by Framework:
PyTorch


Model ID: 0107

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

CSFM (Cardiac Sensing Foundation Model)

University of Oxford (Gu et al.) · 2026

code

Training code public

Multimodal cardiac-sensing foundation model pretrained with generative masked pretraining on ECG, PPG, and paired clinical/machine-generated text reports from roughly 1.7 million individuals across three large-scale critical-care and outpatient ECG datasets. A channel-embedding scheme lets the same model accept any combination of 12-lead ECG, single-lead/wearable ECG, and PPG. The resulting embeddings transfer to diagnostic classification, demographic recognition, vital-sign measurement, clinical-outcome prediction, and ECG question answering. Pretrained weights require a signed academic-access agreement rather than an open download.

12-lead ECG

Filter by Modality:
ECG

Single-lead ECG

Filter by Modality:
ECG

PPG / wearable

Filter by Modality:
PPG / Wearable

Clinical text

Filter by Modality:
Text & EHR

Multimodal

Filter by Modality:
Multimodal

General Purpose / Multi-task

Filter by Disease / Trait:
General / Foundation

Cardiac aging / biological age

Filter by Disease / Trait:
Prognosis & Aging

Embedding

Filter by Task Type:
Representation Learning

Regression

Filter by Task Type:
Regression

Transformer

Filter by Architecture:
Transformer

PyTorch

Filter by Framework:
PyTorch


Model ID: 0058

·

Subject Count: 161,352

DeepCORO-CLIP

Montreal Heart Institute / UCSF / Cedars-Sinai (Harrabi, Avram, Tison, Ouyang et al.) · 2026

graph_1

Code & model weights public

Multi-view foundation model for coronary angiography trained with video-text contrastive learning on 203,808 angiography videos from 28,117 patients across 32,473 studies at the Montreal Heart Institute, externally validated on 4,249 studies from UCSF. Integrates multiple angiographic projections with attention-based pooling for study-level assessment spanning diagnostic, prognostic, and disease-progression tasks: significant-stenosis detection (AUROC 0.888 internal / 0.89 external), stenosis-percentage estimation (MAE 13.6% vs. 19.0% for clinical reports), chronic total occlusion, intracoronary thrombus, and coronary calcification detection. Transfer learning further enables one-year MACE prediction (AUROC 0.79) and LVEF estimation (MAE 7.3%) from the same angiography embeddings, with a mean in-hospital inference time of 4.2 seconds.

Coronary angiography

Filter by Modality:
Invasive Coronary & Intracoronary Imaging

Coronary artery disease / stenosis

Filter by Disease / Trait:
Coronary & Ischemic Disease

LVEF estimation

Filter by Disease / Trait:
Cardiac Function & Hemodynamics

Major adverse cardiovascular events (MACE)

Filter by Disease / Trait:
Prognosis & Aging

Binary classification

Filter by Task Type:
Classification

Regression

Filter by Task Type:
Regression

Hybrid

Filter by Architecture:
Hybrid / Multi-branch

PyTorch

Filter by Framework:
PyTorch


Model ID: 0075

·

Subject Count: 28,117

FedKD-SwinUNETR (Cardiac CT)

Heidelberg University Hospital (AICM) (Toelle, Engelhardt et al.) · 2025

code

Training code public

Largest federated cardiac CT analysis to date (n=8,104 scans) across a real-world federation of German university hospitals, addressing partially-labeled data across sites via a two-step semi-supervised knowledge-distillation strategy: task-specific CNNs first predict on unlabeled data per label type, then a SWIN-UNETR transformer learns from these predictions with label-specific heads. Learns a single federated model that simultaneously predicts TAVI-relevant landmarks (aortic hinge points, coronary ostia, membranous septum) and calcification from cardiac CT, improving generalizability over UNet-based baselines on downstream tasks.

CT angiography

Filter by Modality:
Cardiac CT

Procedural planning / outcome (PCI, TAVI)

Filter by Disease / Trait:
Prognosis & Aging

Coronary artery calcium (CAC) scoring

Filter by Disease / Trait:
Coronary & Ischemic Disease

Detection / localization

Filter by Task Type:
Segmentation & Detection

Segmentation

Filter by Task Type:
Segmentation & Detection

Transformer

Filter by Architecture:
Transformer

PyTorch

Filter by Framework:
PyTorch


Model ID: 0074

·

Subject Count: 8,104

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

EchoNet-Aging

Cedars-Sinai Medical Center (Smidt Heart Institute) / Ouyang Lab · 2025

graph_1

Code & model weights public

Predicts a patient's age from echocardiogram videos across four standard views (PLAX, A2C, A4C, and subcostal), trained on a private multi-site cohort of over 2.6 million videos from more than 166,000 studies across roughly 90,000 patients. The gap between this AI-predicted age and true chronological age is studied as a marker of accelerated or delayed cardiovascular aging and its relationship to all-cause mortality. Uses a 3D CNN (R(2+1)D) with a separate pretrained model per view. Developed by Cedars-Sinai Medical Center and Stanford's Ouyang lab.

Echocardiography video

Filter by Modality:
Echocardiography

Cardiac aging / biological age

Filter by Disease / Trait:
Prognosis & Aging

Regression

Filter by Task Type:
Regression

CNN (3D)

Filter by Architecture:
Convolutional (CNN)

PyTorch

Filter by Framework:
PyTorch


Model ID: 0034

·

Subject Count: 90,738