CVAI Catalog

·

View Catalog

tune

6 models found

·

5 public code

·

2 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

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

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

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