CVAI Catalog

·

View Catalog

tune

17 models found

·

14 public code

·

4 public weights

3D nnU-Net Aortic Dissection Pipeline

General Hospital of Northern Theater Command / Northeastern University, China · 2025

graph_1

Code & model weights public

Fully automatic four-module deep learning pipeline for identifying, segmenting, and Stanford-subtyping aortic dissection (AD) from CT angiography (CTA). A 3D full-resolution nnU-Net first segments the aorta; the segmented boundary is then multi-view projected for AD identification; for AD-positive cases, a second 3D nnU-Net segments the true lumen (TL) and false lumen (FL); finally, a classifier performs Stanford subtyping from multi-view maximum-density projections of the TL/FL. On 386 CTA scans, the pipeline achieved 0.979 accuracy for AD identification, Dice of 0.968 (TL) and 0.971 (FL) for lumen segmentation, and 0.990 accuracy for Stanford subtyping.

Aortic CT

Filter by Modality:
Cardiac CT

Aortic aneurysm / dissection

Filter by Disease / Trait:
Vascular Disease

Segmentation

Filter by Task Type:
Segmentation & Detection

CNN (2D)

Filter by Architecture:
Convolutional (CNN)

Keras

Filter by Framework:
TensorFlow / Keras


Model ID: 0144

·

Subject Count: 386

AortaExplorer

Technical University of Denmark / Capital Region of Denmark · 2026

graph_1

Code & model weights public

Open-source, fully automated framework for end-to-end aortic analysis from CT angiography (CTA), built on top of TotalSegmentator baseline segmentations with additional refinement. AortaExplorer extracts established biomarkers such as diameters across anatomical segments defined by the European Society of Cardiology, and introduces new metrics including aortic tortuosity. Diameter measurements were validated against expert manual readings in more than 10,000 CTA scans from Danish population cohorts, and the tortuosity index's increase with age is consistent with prior literature; the tool reduces per-case analysis time from about 15 minutes to under 5 minutes.

Aortic CT

Filter by Modality:
Cardiac CT

Aortic anatomy segmentation / measurement

Filter by Disease / Trait:
Vascular Disease

Regression

Filter by Task Type:
Regression

CNN (2D)

Filter by Architecture:
Convolutional (CNN)

PyTorch

Filter by Framework:
PyTorch

CC BY 4.0

Filter by License:
Open — Attribution


Model ID: 0151

code

Training code public

Cross-modality cardiac image segmentation model that addresses spatial-temporal confounding -- where the anatomy and imaging-modality elements of cardiac images are intertwined across space and time. DCL performs multi-dimensional causal intervention, modeling causal relationships between images and labels as well as causality along the time and space dimensions, integrating historical optimal interventions to transfer knowledge across temporal contexts. A diffusion mechanism further keeps extracted anatomical elements causally invariant across modalities. On cross-modality cardiac images (MR, CT, and ultrasound), DCL achieved a mean Dice of 0.951, outperforming other advanced segmentation methods.

Echocardiography video

Filter by Modality:
Echocardiography

Cardiac MRI

Filter by Modality:
Cardiac MRI

CT angiography

Filter by Modality:
Cardiac CT

Multimodal

Filter by Modality:
Multimodal

Cardiac chamber segmentation

Filter by Disease / Trait:
Structural Heart & Cardiomyopathy

Segmentation

Filter by Task Type:
Segmentation & Detection

Hybrid

Filter by Architecture:
Hybrid / Multi-branch

PyTorch

Filter by Framework:
PyTorch


Model ID: 0148

·

Subject Count: 60

Deep Vectorised Operators for Coronary Hemodynamics

University of Twente / Politecnico di Milano · 2025

code

Training code public

Machine-learning surrogate model for estimating pulsatile hemodynamic fields (velocity, pressure) in coronary arteries from a steady-state computational fluid dynamics (CFD) prior, avoiding the high computational cost of full pulsatile CFD. The model, a neural field conditioned on hemodynamic boundary conditions, is discretisation-independent and can be parametrised with message-passing or self-attention layers by relaxing point-wise action to permutation-equivariance. Evaluated on 74 stenotic coronary arteries from coronary CT angiography (CCTA) with patient-specific pulsatile CFD as ground truth, the model produced accurate, discretisation-independent estimates of pulsatile velocity and pressure fields.

CT angiography

Filter by Modality:
Cardiac CT

Fractional flow reserve (FFR) / coronary physiology

Filter by Disease / Trait:
Coronary & Ischemic Disease

Regression

Filter by Task Type:
Regression

Hybrid

Filter by Architecture:
Hybrid / Multi-branch

PyTorch

Filter by Framework:
PyTorch


Model ID: 0146

·

Subject Count: 74

3D_CAS (Feature-Fusion-and-Rectification 3D-UNet)

Northeastern University, China (Song, Xu, Yang et al.) · 2022

code

Training code public

Automatic coronary artery segmentation pipeline for coronary CT angiography (CCTA). A 2D DenseNet classifier first screens out CT slices that don't contain coronary artery, then a 3D-UNet -- enhanced with dense blocks in the encoder for richer feature extraction and residual, feature-rectifying blocks in the decoder -- segments the coronary artery tree in the remaining slices. A Gaussian-weighted merging scheme combines overlapping 3D patch predictions, up-weighting the more reliable predictions near each patch's center. On the authors' in-house CCTA dataset, the method achieved a Dice similarity coefficient of 0.826.

CT angiography

Filter by Modality:
Cardiac CT

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


Model ID: 0127

CIS-UNet

University of Florida (Cooper, Shao Labs) · 2024

code

Training code public

Deep learning model for multi-class 3D segmentation of the aorta and its thirteen branches from CT angiography, intended to support planning of endovascular aortic interventions. CIS-UNet combines a CNN encoder with a symmetric decoder and a novel Context-aware Shifted Window Self-Attention (CSW-SA) bottleneck block that adapts the Swin transformer's patch-merging mechanism to more efficiently capture global spatial context. Trained and evaluated via 4-fold cross-validation on the first public multi-branch aorta CTA dataset (59 patients), CIS-UNet outperformed the state-of-the-art SwinUNETR baseline, achieving a mean Dice of 0.713 vs. 0.697 and mean surface distance of 2.78mm vs. 3.39mm, while being more computationally efficient.

CT angiography

Filter by Modality:
Cardiac CT

Aortic anatomy segmentation / measurement

Filter by Disease / Trait:
Vascular Disease

Segmentation

Filter by Task Type:
Segmentation & Detection

Hybrid

Filter by Architecture:
Hybrid / Multi-branch

PyTorch

Filter by Framework:
PyTorch


Model ID: 0130

·

Subject Count: 59

Aortic Dissection Detection nnU-Net

German Cancer Research Center (DKFZ) / University Medical Centre Mannheim, Heidelberg University · 2025

lock

Code & model weights private

nnU-Net-based pipeline for automated detection and sub-classification of acute thoracic aortic dissection (AD) on heterogeneous CT imaging, formulated as a semantic segmentation task rather than direct image classification. The model segments the false lumen (ascending and descending) and the dissection membrane -- along with optional indirect signs such as hemopericardium, aortic wall hematoma, and supra-aortic branch dissection -- and a patient is classified as AD-positive if at least two of the three primary segmented regions exceed a volume threshold tuned via Youden's index; the same pipeline additionally flags Stanford type A dissections. Trained on 157 heterogeneous internal CT studies (not restricted to a single contrast protocol) from Mannheim University Medical Centre and evaluated on an internal held-out test set as well as public external datasets (ImageTBAD and AVT), the model reached an AUROC of 98.7% internally and 97.0% externally, and correctly flagged 93.3% of dissection cases that had not been clinically suspected before imaging. The authors state the trained network will be made publicly available as a non-medical device for further scientific research.

Aortic CT

Filter by Modality:
Cardiac CT

Aortic aneurysm / dissection

Filter by Disease / Trait:
Vascular Disease

Segmentation

Filter by Task Type:
Segmentation & Detection

Binary classification

Filter by Task Type:
Classification

CNN (3D)

Filter by Architecture:
Convolutional (CNN)

PyTorch

Filter by Framework:
PyTorch


Model ID: 0113

·

Subject Count: 263

code

Training code public

Coronary artery calcium (CAC) scoring model that transfers a CNN trained for calcium scoring on non-contrast CT (NCCT) to coronary CT angiography (CCTA), where iodinated contrast otherwise confounds calcium detection and large annotated CCTA training sets are scarce. The CAC-scoring CNN is split into a feature generator and a classifier; the feature generator is trained on the NCCT source domain and adapted to the CCTA target domain via adversarial learning combined with a maximum-mean-discrepancy loss, while the source-domain classifier is reused unchanged for the target domain. Builds directly on the authors' earlier non-contrast CT calcium-scoring network.

CT angiography

Filter by Modality:
Cardiac CT

Coronary artery calcium (CAC) scoring

Filter by Disease / Trait:
Coronary & Ischemic Disease

Multi-class classification

Filter by Task Type:
Classification

Hybrid

Filter by Architecture:
Hybrid / Multi-branch

PyTorch

Filter by Framework:
PyTorch


Model ID: 0106

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

Calcium scoring in low-dose chest CT (dilated CNN)

University Medical Center Utrecht (Lessmann et al.) · 2018

code

Training code public

Two-stage convolutional neural network that automatically detects and anatomically labels coronary artery, thoracic aorta, and cardiac-valve calcifications in low-dose chest CT acquired for lung-cancer screening. A first CNN with a large receptive field (via dilated convolutions) identifies and anatomically labels candidate calcifications; a second CNN filters true positives from the candidates. Trained and evaluated on 1,744 CT scans from the National Lung Screening Trial (NLST), reaching an F1 of 0.89 (soft-filter reconstructions) / 0.84 (sharp-filter reconstructions) for coronary artery calcifications and a linearly-weighted kappa of 0.90-0.91 for per-subject cardiovascular risk categorization versus the manual reference standard.

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

Multi-class classification

Filter by Task Type:
Classification

CNN (2D)

Filter by Architecture:
Convolutional (CNN)


Model ID: 0102

·

Subject Count: 1,744

DeepAAA

Massachusetts General Hospital / Brigham and Women's Hospital (Center for Clinical Data Science) · 2019

lock

Code & model weights private

Deep learning pipeline for detection and quantification of abdominal aortic aneurysm (AAA) -- a typically asymptomatic condition often missed incidentally by radiologists -- from abdominal-pelvic CT. A modified 3D U-Net segments the aorta on both contrast and non-contrast CT volumes with a variable number of images, after which an ellipse-fitting post-processing step measures the aortic cross-sectional diameter along the vessel to detect aneurysmal dilation. Trained and validated on 321 abdominal-pelvic CT examinations from Massachusetts General Hospital, the model reached a sensitivity/specificity of 0.91/0.95 on the primary validation set, and 0.85/1.0 on a separate 57-exam generalization test set with different patient demographics and acquisition characteristics; the authors report that DeepAAA exceeded literature-reported radiologist performance for incidental AAA detection.

Aortic CT

Filter by Modality:
Cardiac CT

Aortic aneurysm / dissection

Filter by Disease / Trait:
Vascular Disease

Segmentation

Filter by Task Type:
Segmentation & Detection

Binary classification

Filter by Task Type:
Classification

CNN (3D)

Filter by Architecture:
Convolutional (CNN)


Model ID: 0114

·

Subject Count: 321

AI-CAC

Veterans Affairs Long Beach Healthcare System / UC Irvine / Mass General Brigham (Hagopian, Strebel, Bernatz, Aerts et al.) · 2025

graph_1

Code & model weights public

U-Net-variant segmentation model that identifies and quantifies coronary artery calcium (CAC) directly from routine non-gated, non-contrast chest CT scans -- the kind ordered for lung-cancer screening or unrelated indications rather than a dedicated cardiac scan -- so that the tens of millions of such scans performed annually can be opportunistically screened for cardiovascular risk without any extra imaging. Predicted calcium masks are combined with the CT's Hounsfield units to compute an Agatston-equivalent score. Trained on 446 expert-segmented scans from 98 medical centers across the U.S. Department of Veterans Affairs national health system (capturing substantial heterogeneity in scanners and protocols) and benchmarked against 795 patients with a paired same-year gated CAC study: nongated AI-CAC differentiates zero-vs-nonzero and <100-vs->=100 Agatston categories with 89.4% (F1 0.93) and 87.3% (F1 0.89) accuracy respectively, and its score stratifies 10-year all-cause mortality (CAC 0 vs. >400: 25.4% vs. 60.2%, hazard ratio 3.49) and composite stroke/MI/death risk (33.5% vs. 63.8%, hazard ratio 3.00). In a simulated opportunistic-screening run across 8,052 low-dose CT scans, cardiologists confirmed 99.2% of patients flagged with AI-CAC >400 would benefit from lipid-lowering therapy. Code and trained model weights are both public under an MIT license.

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

Binary classification

Filter by Task Type:
Classification

CNN (2D)

Filter by Architecture:
Convolutional (CNN)

PyTorch

Filter by Framework:
PyTorch

MIT

Filter by License:
Permissive


Model ID: 0099

CT-LVEF

Columbia University Irving Medical Center / Weill Cornell Medicine / Cornell Tech (Raikhelkar, Bai, Sabuncu, Uriel et al.) · 2026

lock

Code & model weights private

Vision Transformer-based classifier that detects abnormal left-ventricular ejection fraction (LVEF < 50%) directly from static, non-gated, non-contrast chest CT scans -- an imaging modality ordered for unrelated indications (lung cancer screening, pulmonary embolism, trauma) in over 80 million US exams a year -- as a form of opportunistic heart-failure screening. Fine-tunes the encoder of the CT-ViT (GenerateCT) framework, with separate spatial (axial-plane) and z-axis (slice-wise) self-attention blocks, on 3D CT volumes paired with echocardiogram-derived LVEF labels from 25,948 Columbia University studies; reaches an AUROC of 0.786 on a held-out test set and 0.762 on external validation at Weill Cornell Medicine, clearly outperforming demographic/diagnosis-code-only baselines (Random Forest, XGBoost, AUROC 0.54-0.61). On a radiologist-comparison subset, the model's weighted F1 (0.80-0.81) exceeded two board-certified thoracic radiologists (0.62-0.80) at a small fraction of the interpretation time. Grad-CAM saliency maps highlighted clinically sensible correlates of reduced LVEF (cardiomegaly, dilated superior vena cava, calcified ascending aorta, pacemaker hardware, pulmonary edema). No public code or model weights have been released.

Non-contrast cardiac CT

Filter by Modality:
Cardiac CT

LV systolic dysfunction (LVSD)

Filter by Disease / Trait:
Cardiac Function & Hemodynamics

Binary classification

Filter by Task Type:
Classification

Vision Transformer

Filter by Architecture:
Transformer


Model ID: 0095

·

Subject Count: 19,410

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

MIL-Attention Coronary Stenosis Classifier

University of Gothenburg / Sahlgrenska University Hospital (Gupta et al.) · 2025

code

Training code public

Multi-instance-learning (MIL) model for detecting >=50% coronary stenosis directly from curved multiplanar reformation (CMR) images generated during routine coronary CT angiography (CCTA) reads, without requiring slice-level annotations. A multi-range Hounsfield-unit preprocessing pipeline (Sobel edge detection across five attenuation windows) highlights plaque and vessel-wall structures, which a VGG16-based encoder with positional encoding and multi-head attention aggregates across each patient's 'bag' of up to 36 CMR slices per artery to give an interpretable, attention-weighted patient-level prediction. Trained and five-fold cross-validated on 900 real-world CCTA cases (776 LAD / 694 RCA / 600 LCX) from Sahlgrenska University Hospital, reaching AUCs of 0.91-0.92 across the three major coronary arteries. Code (preprocessing + MIL training pipeline) is public; the clinical CMR dataset and trained weights are not released.

CT angiography

Filter by Modality:
Cardiac CT

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


Model ID: 0083

·

Subject Count: 900

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

code

Training code public

3D CNN that segments seven cardiac substructures - both ventricles, both atria, the LV myocardium, ascending aorta, and pulmonary artery trunk - from cardiac CT angiography. Trained with a hybrid loss function combining multiple segmentation objectives. Developed at CUHK for the MICCAI 2017 Multi-Modality Whole Heart Segmentation (MM-WHS) challenge.

CT angiography

Filter by Modality:
Cardiac CT

Cardiac chamber segmentation

Filter by Disease / Trait:
Structural Heart & Cardiomyopathy

Segmentation

Filter by Task Type:
Segmentation & Detection

CNN (3D)

Filter by Architecture:
Convolutional (CNN)


Model ID: 0009

·

Subject Count: 60