CVAI Catalog

·

View Catalog

tune

7 models found

·

7 public code

·

6 public weights

CM-UNet (Contrastive Masked UNet)

EPFL / Lausanne University Hospital (CHUV) · 2025

graph_1

Code & model weights public

Self-supervised deep learning model for coronary artery segmentation from invasive X-ray coronary angiography (ICA), designed to reduce reliance on large annotated datasets. CM-UNet combines a Contrastive Masked Autoencoder (CMAE) with a UNet backbone: an online encoder-decoder branch reconstructs masked image patches while a momentum branch produces contrastive embeddings, jointly pretraining the network on unannotated angiography images before fine-tuning on a small labeled set. Fine-tuning with only 18 annotated images (instead of 500) led to just a 15.2% drop in Dice score, versus a 46.5% drop for baseline models trained without this self-supervised pretraining -- demonstrating strong label efficiency for coronary segmentation.

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


Model ID: 0137

DOSTA-Net

Northwestern University (Advanced AI in Medicine and Physics Lab) · 2026

graph_1

Code & model weights public

Domain-Shuffle Temporal Attention Network for coronary vessel extraction from X-ray coronary angiography (XCA), trained entirely on synthetic temporal XCA data without requiring manual vessel annotations. By leveraging synthetic data generation and a domain-shuffle temporal attention mechanism, DOSTA-Net avoids the need for costly expert-labeled real angiography sequences while still learning temporally consistent vessel segmentation across frames of an XCA sequence.

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


Model ID: 0160

SegmentMIL

Technical University of Munich (TUM University Hospital) · 2026

code

Training code public

Transformer-based multi-view multiple-instance learning (MIL) framework for patient-level coronary stenosis classification from multi-view invasive coronary angiography. Rather than requiring expensive view-level stenosis annotations, SegmentMIL is trained end-to-end on real-world clinical data using only patient-level labels already present in hospital systems, and jointly predicts stenosis presence while localizing the affected artery (left/right) and segment. It captures temporal dynamics and dependencies across the multiple angiographic views per patient (which prior view-level models ignore), and outperforms both single-view models and classical MIL baselines on internal and external clinical evaluations.

Coronary angiography

Filter by Modality:
Invasive Coronary & Intracoronary Imaging

Coronary artery disease / stenosis

Filter by Disease / Trait:
Coronary & Ischemic Disease

Binary classification

Filter by Task Type:
Classification

Hybrid

Filter by Architecture:
Hybrid / Multi-branch

PyTorch

Filter by Framework:
PyTorch


Model ID: 0155

·

Subject Count: 2,003

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

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

DeepCoro

Montreal Heart Institute (HeartWise.AI) (Avram et al.) · 2024

graph_1

Code & model weights public

AI-driven pipeline for quantitative coronary-stenosis assessment from routine DICOM coronary angiography videos, combining vessel tracking with a video Swin3D transformer trained and validated on 182,418 angiography videos spanning 5 years at the Montreal Heart Institute. Achieves a mean absolute error of 20.15% and a classification AUROC of 0.8294 for stenosis-percentage prediction against cardiologist assessment, with lower inter-rater variability than two expert interventional cardiologists, and can be fine-tuned to quantitative coronary angiography (QCA) data for even lower error (MAE 7.75%).

Coronary angiography

Filter by Modality:
Invasive Coronary & Intracoronary Imaging

Coronary artery disease / stenosis

Filter by Disease / Trait:
Coronary & Ischemic Disease

Coronary artery segmentation / anatomy

Filter by Disease / Trait:
Coronary & Ischemic Disease

Regression

Filter by Task Type:
Regression

Multi-class classification

Filter by Task Type:
Classification

Hybrid

Filter by Architecture:
Hybrid / Multi-branch

PyTorch

Filter by Framework:
PyTorch


Model ID: 0072

SAM-VMNet

Ocean University of China / Shandong University · 2025

graph_1

Code & model weights public

Segments coronary vessels from invasive X-ray angiography images and automatically quantifies the degree of stenosis along the extracted centerlines. Combines MedSAM, a Segment-Anything-style vision model, with a Mamba-based VM-UNet segmentation branch for efficient long-range feature modeling. Trained and evaluated on the ARCADE, DCA1, and GH angiography datasets by researchers at Ocean University of China and Shandong University.

Coronary angiography

Filter by Modality:
Invasive Coronary & Intracoronary Imaging

Coronary artery segmentation / anatomy

Filter by Disease / Trait:
Coronary & Ischemic Disease

Coronary artery disease / stenosis

Filter by Disease / Trait:
Coronary & Ischemic Disease

Segmentation

Filter by Task Type:
Segmentation & Detection

Detection / localization

Filter by Task Type:
Segmentation & Detection

Hybrid

Filter by Architecture:
Hybrid / Multi-branch

PyTorch

Filter by Framework:
PyTorch

MIT

Filter by License:
Permissive


Model ID: 0001