CVAI Catalog

·

View Catalog

tune

9 models found

·

9 public code

·

9 public weights

AngioPy

Lausanne University Hospital / EPFL (Ando, Thanou Labs) · 2025

graph_1

Code & model weights public

Open-source, user-guided deep learning tool for coronary artery segmentation from invasive coronary angiography (ICA), designed to improve on traditional quantitative coronary angiography (QCA) edge-detection algorithms that typically require manual correction. Rather than segmenting the whole coronary tree indiscriminately, AngioPy lets the user click a handful of ground-truth points along a specific target vessel (including side branches), and predicts a binary mask for that single artery at the chosen cardiac-cycle time-step. Evaluated against an established QCA system on angiograms from the FAME 2 trial, AngioPy achieved an average F1 score of 0.927 (internal) and 0.924 (external validation), with vessel-diameter and lesion minimal-lumen-diameter measurements showing excellent agreement with QCA (r=0.93-0.96).

Coronary angiography

Filter by Modality:
Invasive Coronary & Intracoronary Imaging

Coronary artery segmentation / anatomy

Filter by Disease / Trait:
Coronary & Ischemic Disease

Segmentation

Filter by Task Type:
Segmentation & Detection

Hybrid

Filter by Architecture:
Hybrid / Multi-branch

PyTorch

Filter by Framework:
PyTorch

GPL 3.0

Filter by License:
Copyleft


Model ID: 0133

ENCASE

Peking University (Hong et al.) · 2019

graph_1

Code & model weights public

Ensemble classifier combining hand-engineered expert features with a deep convolutional neural network for classifying cardiac rhythm from a single-lead ECG recording into normal sinus rhythm, atrial fibrillation, another rhythm, or too noisy to classify. A large set of expert features (from time-, frequency-, and template-based analysis) is fed into a gradient-boosted tree classifier (AdaBoost), and its output is combined with a separate deep CNN operating directly on the raw waveform; combining both feature families measurably outperformed either alone. ENCASE won 1st place in the PhysioNet/Computing in Cardiology Challenge 2017 (single-lead AF classification) with an overall F1 score of 0.83 on the official hidden test set, and remains a widely cited example of combining classical signal-processing features with deep representations for ECG classification.

Single-lead ECG

Filter by Modality:
ECG

Atrial fibrillation

Filter by Disease / Trait:
Arrhythmia

Multi-class classification

Filter by Task Type:
Classification

Hybrid

Filter by Architecture:
Hybrid / Multi-branch

GPL 3.0

Filter by License:
Copyleft


Model ID: 0117

4Dsegment (bi-ventricular segmentation + motion tracking)

Imperial College London (Duan et al., UK Digital Heart Project) · 2019

graph_1

Code & model weights public

Automated pipeline that segments both heart ventricles and tracks their motion throughout the cardiac cycle from short-axis cine cardiac MRI, producing 3D bi-ventricular models with per-vertex wall-thickness and curvature measurements over time. Built on a shape-refined multi-task fully convolutional network, followed by non-rigid registration and mesh-based motion tracking. Trained on roughly 400 manually annotated pulmonary hypertension patients as part of Imperial College London's UK Digital Heart Project, and underlies the related 4Dsurvival cardiac-motion survival-prediction study.

Cardiac MRI

Filter by Modality:
Cardiac MRI

Cardiac chamber segmentation

Filter by Disease / Trait:
Structural Heart & Cardiomyopathy

Segmentation

Filter by Task Type:
Segmentation & Detection

CNN (2D)

Filter by Architecture:
Convolutional (CNN)

TensorFlow

Filter by Framework:
TensorFlow / Keras

GPL 3.0

Filter by License:
Copyleft


Model ID: 0002

·

Subject Count: 400

DeepBeat

Stanford University (Ashley Lab) · 2020

graph_1

Code & model weights public

Multi-task model that jointly assesses signal quality and detects atrial fibrillation from wrist-worn wearable photoplethysmography (PPG), pretrained on roughly one million simulated unlabeled signals before fine-tuning on labeled wearable data. Uses a 1D CNN with separate output heads for signal quality and arrhythmia detection. Developed by Stanford's Ashley Lab.

PPG / wearable

Filter by Modality:
PPG / Wearable

Atrial fibrillation

Filter by Disease / Trait:
Arrhythmia

Binary classification

Filter by Task Type:
Classification

CNN (1D)

Filter by Architecture:
Convolutional (CNN)

Keras

Filter by Framework:
TensorFlow / Keras

GPL 3.0

Filter by License:
Copyleft


Model ID: 0046

·

Subject Count: 100

DeepCAC

Harvard AIM Lab (Zeleznik et al.) · 2021

graph_1

Code & model weights public

Fully automatic pipeline that localizes the heart, segments coronary calcium, and produces an Agatston-style coronary artery calcium score from gated and non-gated chest/cardiac CT. A three-stage 3D CNN performs each step in sequence. Validated across the Framingham, NLST, PROMISE, and ROMICAT-II cohorts, where the resulting calcium score predicted cardiovascular events with hazard ratios up to 4.3. Developed by the Harvard AIM Lab.

Non-contrast cardiac CT

Filter by Modality:
Cardiac CT

Coronary artery calcium (CAC) scoring

Filter by Disease / Trait:
Coronary & Ischemic Disease

Segmentation

Filter by Task Type:
Segmentation & Detection

CNN (3D)

Filter by Architecture:
Convolutional (CNN)

GPL 3.0

Filter by License:
Copyleft


Model ID: 0008

·

Subject Count: 3,380

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

ESI (ECG Semantic Integrator)

Rice University · convnextv2_base · 2024

graph_1

Code & model weights public

Multimodal ECG model that pairs a 1D ConvNeXtV2 signal encoder with a BioLinkBERT text encoder, trained with a joint contrastive-and-captioning objective using LLM-generated descriptions of ECG demographics and waveform patterns in place of raw clinical reports. Validated on arrhythmia diagnosis and ECG-based subject identification, reaching an AUROC of 0.938 fine-tuned and 0.812 zero-shot on PTB-XL diagnostic classification. Developed at Rice University.

12-lead ECG

Filter by Modality:
ECG

Clinical text

Filter by Modality:
Text & EHR

Multimodal

Filter by Modality:
Multimodal

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

Hybrid

Filter by Architecture:
Hybrid / Multi-branch

PyTorch

Filter by Framework:
PyTorch

GPL 3.0

Filter by License:
Copyleft


Model ID: 0019

·

Subject Count: 64,037

graph_1

Code & model weights public

Self-supervised ECG representation learned purely from patient identity: the model is trained so that ECGs from the same patient, recorded at different times, map to nearby points in latent space, with no other labels required. Linear models trained on these representations showed a 51% average performance gain over training from scratch across sex classification, age regression, LVH detection, and AF detection. Developed by the Broad Institute's ML4H group and trained on 3.2 million private ECGs from Massachusetts General Hospital; 12-lead, lead-I-only, and lead-II-only checkpoints are all released.

12-lead ECG

Filter by Modality:
ECG

General Purpose / Multi-task

Filter by Disease / Trait:
General / Foundation

Embedding

Filter by Task Type:
Representation Learning

CNN (1D)

Filter by Architecture:
Convolutional (CNN)

TensorFlow

Filter by Framework:
TensorFlow / Keras

GPL 3.0

Filter by License:
Copyleft


Model ID: 0028

·

Subject Count: 404,929

PTB-XL benchmark ECG classifier (xresnet1d101)

Fraunhofer HHI (Strodthoff et al.) · 2021

graph_1

Code & model weights public

Reference benchmark suite for the PTB-XL ECG dataset, providing pretrained xresnet1d, InceptionTime, LSTM, and CPC-pretrained models for 71-label, diagnostic, sub-diagnostic, and super-diagnostic classification of ECG findings. Widely used as a standardized baseline for comparing new ECG classification methods. Developed by Strodthoff et al. at Fraunhofer HHI.

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

CNN (1D)

Filter by Architecture:
Convolutional (CNN)

PyTorch

Filter by Framework:
PyTorch

GPL 3.0

Filter by License:
Copyleft


Model ID: 0029

·

Subject Count: 18,885