CVAI Catalog

·

View Catalog

tune

19 models found

·

18 public code

·

8 public weights

Multilabel CNN for Atrial Fibrosis Assessment (CemrgApp)

King's College London (Niederer Lab / CEMRG) · 2020

code

Training code public

Fully automatic, open-source deep learning pipeline for estimating left atrial fibrosis from late gadolinium enhancement (LGE) cardiac MRI, built to remove the operator-dependent steps that limit reproducibility of conventional atrial LGE analysis. A multilabel convolutional neural network delineates the left atrial blood pool, pulmonary veins, and mitral valve; these structures are then used to automatically calculate fibrosis burden via established image-intensity-ratio thresholds, without manual tracing. Validated on a 3D LGE-CMR dataset of 207 scans, the pipeline's automatic segmentation achieved a 91% Dice score against manual tracing, and its fully automatic fibrosis quantification closely matched semi-automatic reference methods. The CNN and pipeline are distributed as part of the open-source CemrgApp platform.

Cardiac MRI

Filter by Modality:
Cardiac MRI

LGE scar burden

Filter by Disease / Trait:
Structural Heart & Cardiomyopathy

Segmentation

Filter by Task Type:
Segmentation & Detection

CNN (2D)

Filter by Architecture:
Convolutional (CNN)


Model ID: 0134

·

Subject Count: 207

EchoNet-TR

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

graph_1

Code & model weights public

Automated deep learning workflow that detects and grades tricuspid regurgitation (TR) severity from full transthoracic echocardiography studies. The pipeline first identifies apical-4-chamber (A4C) video clips with color Doppler across the tricuspid valve from a full echo study, then applies a dedicated R(2+1)D video classifier to grade TR severity. Trained on over 2 million echo videos from 47,312 studies at Cedars-Sinai Medical Center and externally validated at Stanford Healthcare, the model identified color-Doppler A4C views with AUC ≥0.999 and detected clinically significant (moderate-or-severe) TR with AUC 0.951 and severe TR with AUC 0.980. Code and trained model weights are released to support prospective evaluation of AI-assisted TR screening.

Echocardiography video

Filter by Modality:
Echocardiography

Valvular disease

Filter by Disease / Trait:
Structural Heart & Cardiomyopathy

Multi-class classification

Filter by Task Type:
Classification

CNN (3D)

Filter by Architecture:
Convolutional (CNN)

PyTorch

Filter by Framework:
PyTorch


Model ID: 0122

ACDC Segmenter

ETH Zurich (Baumgartner, Koch, Pollefeys, Konukoglu) · 2017

code

Training code public

One of the foundational baseline segmentation networks submitted to the 2017 Automated Cardiac Diagnosis Challenge (ACDC), comparing 2D and 3D convolutional network designs for segmenting the left ventricle cavity, myocardium, and right ventricle cavity from short-axis cine cardiac MRI at end-diastole and end-systole. The accompanying study systematically explored the tradeoffs between 2D and 3D convolutions for this task, finding that, due to the highly anisotropic voxel spacing typical of clinical cine cardiac MRI, 2D networks that treat each slice independently can match or exceed 3D networks while being far cheaper to train. The public code and pretrained weights for the best-performing configuration have served as a widely used, simple baseline for later cardiac MRI segmentation research (including for automatically deriving ventricular volumes and ejection fraction).

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)


Model ID: 0116

·

Subject Count: 150

CAMUS U-Net Baseline

CREATIS, University of Lyon (Leclerc et al.) / University of Sherbrooke (vitalab pretrained models) · 18,000,000 params · 2019

graph_1

Code & model weights public

The original U-Net baseline segmentation network introduced alongside the CAMUS (Cardiac Acquisitions for Multi-structure Ultrasound Segmentation) dataset, one of the largest fully open-access, expert-annotated 2D echocardiography benchmarks. The network segments the left ventricle endocardium (LVEndo), left ventricle epicardium/myocardium (LVEpi), and left atrium (LA) from apical 2-chamber and 4-chamber echo views at end-diastole (ED) and end-systole (ES). In the original ten-fold cross-validation benchmark comparing U-Net, U-Net++, Stacked Hourglass, Anatomically Constrained Neural Networks, and classical methods, the U-Net variant (18M parameters) achieved the best overall accuracy, reaching Dice scores of 0.939 (ED) / 0.916 (ES) for LVEndo and 0.954 (ED) / 0.945 (ES) for LVEpi, approaching inter-observer variability. A pretrained checkpoint of this baseline U-Net is distributed via the University of Sherbrooke's vitalab CASTOR project as part of a broader library for building anatomically-constrained cardiac segmentation pipelines.

Echocardiography video

Filter by Modality:
Echocardiography

Cardiac chamber segmentation

Filter by Disease / Trait:
Structural Heart & Cardiomyopathy

Segmentation

Filter by Task Type:
Segmentation & Detection

CNN (2D)

Filter by Architecture:
Convolutional (CNN)

PyTorch

Filter by Framework:
PyTorch

Apache 2.0

Filter by License:
Permissive


Model ID: 0115

·

Subject Count: 500

DBRes (Dual Bayesian ResNet)

University of Oxford / University of Surrey · 2022

code

Training code public

Dual Bayesian ResNet model for heart murmur detection from multi-location phonocardiogram (PCG) recordings, developed for the George B. Moody PhysioNet Challenge 2022. Each patient's PCG recordings are segmented into overlapping log-mel spectrograms, which are passed through two Bayesian ResNet binary classifiers running simultaneously (present vs. unknown-or-absent, and unknown vs. present-or-absent); the two outputs are aggregated into a patient-level present/unknown/absent murmur classification. An optional second-stage XGBoost model integrates the DBRes output with demographic data and hand-crafted signal features. On the Challenge's official hidden test set, DBRes achieved a weighted accuracy of 0.771 for the murmur-detection task, placing 4th among all teams.

Phonocardiogram (PCG) / heart sounds

Filter by Modality:
Heart Sounds / Phonocardiography

Heart murmur detection

Filter by Disease / Trait:
Structural Heart & Cardiomyopathy

Multi-class classification

Filter by Task Type:
Classification

CNN (2D)

Filter by Architecture:
Convolutional (CNN)

PyTorch

Filter by Framework:
PyTorch


Model ID: 0108

·

Subject Count: 1,568

RHD Video Diagnosis Network

Universidade Federal de Minas Gerais (UFMG) / Children's National Hospital (Martins et al.) · 2021

lock

Code & model weights private

Video-based deep learning system for automated screening of rheumatic heart disease (RHD) from echocardiography, aimed at low-resource settings where RHD -- the most common acquired heart disease in children and young adults worldwide -- is endemic but echocardiography expertise is scarce. A 3D convolutional neural network (C3D) classifies each echo video clip, explicitly modeling temporal information across frames; a supervised meta-classifier then aggregates the per-video predictions from an exam (which may contain many video clips from different views) into a single exam-level RHD diagnosis. Evaluated on 11,646 echocardiography videos from 912 screening exams collected in underserved areas of Brazil and Uganda, the 3D C3D network significantly outperformed a comparison 2D CNN (VGG16) that ignores temporal information, and the learned aggregation model reached 72.77% exam-level diagnostic accuracy, exceeding simple majority voting across a patient's videos.

Echocardiography video

Filter by Modality:
Echocardiography

Rheumatic heart disease (RHD)

Filter by Disease / Trait:
Structural Heart & Cardiomyopathy

Binary classification

Filter by Task Type:
Classification

CNN (3D)

Filter by Architecture:
Convolutional (CNN)


Model ID: 0120

·

Subject Count: 912

DeepHeart (Pediatric Tricuspid Valve Segmentation)

Children's Hospital of Philadelphia / MIT / Queen's University (Herz, Jolley et al.) · 2021

code

Training code public

Deep learning framework, developed in collaboration with the MONAI community, for automatic segmentation of tricuspid valve leaflets from transthoracic 3D echocardiograms in children with hypoplastic left heart syndrome (HLHS) and other forms of single-ventricle congenital heart disease, integrated into 3D Slicer via MONAILabel for interactive clinical/research use. Addresses a modality (pediatric 3D echocardiography) and population (single-ventricle congenital heart disease) largely absent from adult-focused cardiac AI models.

Echocardiography video

Filter by Modality:
Echocardiography

Valvular disease

Filter by Disease / Trait:
Structural Heart & Cardiomyopathy

Segmentation

Filter by Task Type:
Segmentation & Detection

CNN (3D)

Filter by Architecture:
Convolutional (CNN)

PyTorch

Filter by Framework:
PyTorch


Model ID: 0076

·

Subject Count: 129

EchoCLR

Yale School of Medicine (CarDS Lab) · 2024

code

Training code public

Multi-instance contrastive self-supervised learning framework for echocardiography video representation learning: two distinct videos from the same patient exam are treated as positive pairs (rather than augmentations of a single clip), and a frame-reordering pretext task on temporally shuffled frames encourages the 3D-CNN backbone to learn temporally coherent representations. After self-supervised pretraining on unlabeled echocardiograms, the backbone is efficiently fine-tuned for cardiac disease classification (severe aortic stenosis, left ventricular hypertrophy) using very few labeled examples, outperforming SimCLR, standard multi-instance SimCLR, and Kinetics-400-initialized baselines across a range of training-data ratios.

Echocardiography video

Filter by Modality:
Echocardiography

Valvular disease

Filter by Disease / Trait:
Structural Heart & Cardiomyopathy

Left ventricular hypertrophy (LVH)

Filter by Disease / Trait:
Structural Heart & Cardiomyopathy

Binary classification

Filter by Task Type:
Classification

CNN (3D)

Filter by Architecture:
Convolutional (CNN)


Model ID: 0077

EchoNet-CMR

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

graph_1

Code & model weights public

Predicts cardiac MRI (CMR) tissue-characterization findings -- wall-motion abnormalities and myocardial scar -- directly from standard transthoracic echocardiography videos (A4C/A2C/PLAX views), using a factorized 3D R2+1D convolutional network. Trained and validated on a single-institution cohort of more than 1,400 patients with paired echo and CMR studies within 30 days of each other. Released weights and inference code cover the two binarized outcomes (wall motion, scar); the continuous CMR tissue markers (native T1, T2, ECV) evaluated in the paper are not part of the public release.

Echocardiography video

Filter by Modality:
Echocardiography

LGE scar burden

Filter by Disease / Trait:
Structural Heart & Cardiomyopathy

Structural heart disease (composite)

Filter by Disease / Trait:
Structural Heart & Cardiomyopathy

Binary classification

Filter by Task Type:
Classification

CNN (3D)

Filter by Architecture:
Convolutional (CNN)

PyTorch

Filter by Framework:
PyTorch

Research use only

Filter by License:
Non-commercial / Research-only


Model ID: 0060

PLAX Severe-AS Ensemble

Yale School of Medicine (CarDS Lab) · 2023

code

Training code public

Ensemble of 3D convolutional neural networks that detects severe aortic stenosis directly from single-view 2D parasternal long-axis (PLAX) transthoracic echocardiogram videos, without requiring Doppler imaging. Representations are first pretrained with patient-level contrastive self-supervised learning on PLAX clips, then fine-tuned for binary AS classification; the ensemble is externally validated across a temporally-distinct cohort and geographically-distinct cohorts in California and New England. No pretrained weights are released; only the training/evaluation pipeline is public.

Echocardiography video

Filter by Modality:
Echocardiography

Valvular disease

Filter by Disease / Trait:
Structural Heart & Cardiomyopathy

Binary classification

Filter by Task Type:
Classification

CNN (3D)

Filter by Architecture:
Convolutional (CNN)


Model ID: 0059

SimLVSeg

Mohamed Bin Zayed University of Artificial Intelligence (BioMedIA) · 2024

graph_1

Code & model weights public

Self- and weakly-supervised pipeline for left-ventricle segmentation across the full cardiac cycle in apical-4-chamber echocardiography videos. A video segmentation network (2D super-image or 3D U-Net encoder) is first pretrained with a self-supervised temporal-masking objective on largely unannotated echo frames, then fine-tuned with weak supervision from the sparse end-diastole/end-systole frame labels that most echo datasets provide. Achieves 93.3% Dice on EchoNet-Dynamic, outperforming nnU-Net and non-SSL baselines, and generalizes to the external CAMUS dataset. Developed by the BioMedIA group at MBZUAI.

Echocardiography video

Filter by Modality:
Echocardiography

Cardiac chamber segmentation

Filter by Disease / Trait:
Structural Heart & Cardiomyopathy

Segmentation

Filter by Task Type:
Segmentation & Detection

CNN (3D)

Filter by Architecture:
Convolutional (CNN)

PyTorch

Filter by Framework:
PyTorch

CC BY-NC 4.0

Filter by License:
Non-commercial / Research-only


Model ID: 0081

·

Subject Count: 10,030

EchoNet-AR

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

graph_1

Code & model weights public

Video-based deep learning model that grades aortic regurgitation (AR) severity—none/trace, mild, moderate, or severe—from color Doppler echocardiography. View-specific R(2+1)D 3D-CNNs analyze five standard transthoracic views (PLAX, PLAX-AV, A3C, A3C-AV, A5C) and their outputs are combined by a maximum-severity rule at the study level. Trained on ~47,600 color Doppler videos from Cedars-Sinai and externally validated at Stanford Healthcare, reaching AUCs of 0.95 for at-least-moderate AR and 0.97 for severe AR internally. Developed by the Ouyang lab at Cedars-Sinai Medical Center.

Echocardiography video

Filter by Modality:
Echocardiography

Valvular disease

Filter by Disease / Trait:
Structural Heart & Cardiomyopathy

Multi-class classification

Filter by Task Type:
Classification

CNN (3D)

Filter by Architecture:
Convolutional (CNN)

PyTorch

Filter by Framework:
PyTorch

Research use only

Filter by License:
Non-commercial / Research-only


Model ID: 0053

EchoNet-MS

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

graph_1

Code & model weights public

Deep learning model for automated phenotyping of mitral stenosis (MS) from echocardiography. Uses video-based R(2+1)D convolutional networks on color Doppler and B-mode views to identify and grade mitral stenosis severity, following the multi-view valvular-assessment approach of the EchoNet family. Trained and validated on large clinical echocardiography cohorts from Kaiser Permanente Northern California with external testing at Stanford Healthcare and Cedars-Sinai. Developed by the Ouyang lab.

Echocardiography video

Filter by Modality:
Echocardiography

Valvular disease

Filter by Disease / Trait:
Structural Heart & Cardiomyopathy

Multi-class classification

Filter by Task Type:
Classification

CNN (3D)

Filter by Architecture:
Convolutional (CNN)

PyTorch

Filter by Framework:
PyTorch


Model ID: 0055

AI-ECG Amyloid (Image)

Yale School of Medicine (CarDS Lab) · 2024

code

Training code public

CNN-based image classifier that screens 12-lead ECG images for a signature of transthyretin amyloid cardiomyopathy (ATTR-CM), producing a study-level probability score. Trained on a private Yale New Haven Health System cohort of nuclear-imaging-confirmed ATTR-CM cases and matched controls. Used alongside a companion echocardiography model to track pre-clinical ATTR-CM progression years before it would otherwise be confirmed by nuclear amyloid imaging. Developed by Yale's CarDS Lab and distributed as a packaged research-use executable rather than downloadable weights.

12-lead ECG image

Filter by Modality:
ECG

Cardiac amyloidosis

Filter by Disease / Trait:
Structural Heart & Cardiomyopathy

Binary classification

Filter by Task Type:
Classification

CNN (2D)

Filter by Architecture:
Convolutional (CNN)

PyTorch

Filter by Framework:
PyTorch


Model ID: 0024

AI-ECG HCM (ECGVision HCM)

Yale School of Medicine (CarDS Lab) · 2025

code

Training code public

EfficientNet-B3 CNN that detects hypertrophic cardiomyopathy directly from images of printed or scanned 12-lead ECGs, rather than from raw digital waveforms, enabling screening from a photo of a paper tracing. Initialized via self-supervised contrastive pretraining on patient identity, then fine-tuned at Yale New Haven Hospital on over 124,000 ECGs from about 67,000 patients, with HCM status confirmed by cardiac MRI or echocardiography. Externally validated on ECG images from MIMIC-IV, Amsterdam UMC, and UK Biobank. Developed by Yale's CarDS Lab.

12-lead ECG image

Filter by Modality:
ECG

Hypertrophic cardiomyopathy

Filter by Disease / Trait:
Structural Heart & Cardiomyopathy

Binary classification

Filter by Task Type:
Classification

CNN (2D)

Filter by Architecture:
Convolutional (CNN)

PyTorch

Filter by Framework:
PyTorch


Model ID: 0025

·

Subject Count: 66,987

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

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

MSWS Network + Boundary2Patches (LAScarQS2022)

Queen Mary University of London (Khan et al.) · 2022

code

Training code public

Two-stage pipeline that segments the left atrium and quantifies atrial scar tissue from 3D late-gadolinium-enhancement cardiac MRI, supporting atrial-fibrillation ablation planning. A Multi-Scale Weight Sharing network first delineates the atrial cavity, then a boundary-patch method segments scar tissue around the detected wall. Developed at Queen Mary University of London for the LAScarQS 2022 MICCAI/STACOM segmentation challenge.

Cardiac MRI

Filter by Modality:
Cardiac MRI

Cardiac chamber segmentation

Filter by Disease / Trait:
Structural Heart & Cardiomyopathy

LGE scar burden

Filter by Disease / Trait:
Structural Heart & Cardiomyopathy

Segmentation

Filter by Task Type:
Segmentation & Detection

CNN (3D)

Filter by Architecture:
Convolutional (CNN)

PyTorch

Filter by Framework:
PyTorch


Model ID: 0004

ukbb_cardiac segmentation network

Imperial College London (Bai et al.) · 2018

graph_1

Code & model weights public

Long-standing toolbox for automated segmentation of the ventricles and atria and derivation of cardiac imaging phenotypes from short- and long-axis cine cardiac MRI. Built on a fully convolutional network trained per slice, and widely reused across UK Biobank cardiac imaging studies since its 2018 publication. Developed at Imperial College London.

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

Apache 2.0

Filter by License:
Permissive


Model ID: 0006

·

Subject Count: 74,916