CVAI Catalog

·

View Catalog

tune

17 models found

·

17 public code

·

13 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

code

Training code public

3D convolutional autoencoder that filters reverberation clutter artifacts from transthoracic echocardiography (TTE) video sequences, improving downstream measurements such as speckle-tracking strain. Built on a 3D U-Net-style encoder-decoder with an input-output skip connection to preserve fine structures and attention-gate modules to focus on cluttered regions, the network was trained on synthetic clutter simulated across six ultrasound vendors and generalized well to real in vivo artifactual sequences, substantially reducing the discrepancy between cluttered and clutter-free strain profiles while running in a fraction of a second per sequence.

Echocardiography video

Filter by Modality:
Echocardiography

General Purpose / Multi-task

Filter by Disease / Trait:
General / Foundation

Generation

Filter by Task Type:
Generation

Hybrid

Filter by Architecture:
Hybrid / Multi-branch

Keras

Filter by Framework:
TensorFlow / Keras


Model ID: 0145

ECG Scalogram-Phasogram Fusion CNN

Sapienza University of Rome · 2024

code

Training code public

CNN-based arrhythmia classifier that fuses the magnitude (scalogram) and phase (phasogram) of the continuous wavelet transform (CWT) of ECG heartbeats, rather than relying on magnitude information alone as most prior 2D-representation approaches do. Several fusion strategies (input-level, intermediate-layer, and output-level fusion) were compared on the public PhysioNet MIT-BIH Arrhythmia database. Despite a simple CNN architecture, the best fusion strategy achieved about 98.5% overall accuracy, 98.5% sensitivity and 95.6% specificity, competitive with more complex state-of-the-art approaches.

Single-lead ECG

Filter by Modality:
ECG

Multi-label classification

Filter by Task Type:
Classification

CNN (2D)

Filter by Architecture:
Convolutional (CNN)

Keras

Filter by Framework:
TensorFlow / Keras


Model ID: 0143

·

Subject Count: 47

OHFFDRL

2026

code

Training code public

Optimized Hierarchical Fused Fuzzy Deep Reinforcement Learning model for predicting non-sinus (higher-risk) cardiac rhythms from PQRST-analyzed 12-lead ECG data. The three-stage approach combines data preprocessing, reinforcement learning, and fuzzy deep learning to classify sinus vs. non-sinus rhythms. Evaluated on a 12-lead ECG dataset of 10,646 patients, OHFFDRL achieved 94% accuracy, an AUC of 0.91, and was interpreted using SHAP, LIME, calibration curves, adversarial vulnerability analysis, and integrated gradients; TAxis (ventricular repolarization movement range) was found to be the most important distinguishing feature.

12-lead ECG

Filter by Modality:
ECG

Binary classification

Filter by Task Type:
Classification

Hybrid

Filter by Architecture:
Hybrid / Multi-branch

TensorFlow

Filter by Framework:
TensorFlow / Keras


Model ID: 0157

·

Subject Count: 10,646

SDNet (Spatial Decomposition Network)

University of Edinburgh · 2019

code

Training code public

Disentangled representation learning model for cardiac image analysis that factorises 2D medical images (MRI, CT) into a spatial 'anatomy factor' (a semantically meaningful multi-channel map, produced by a U-Net-style anatomy encoder) and a non-spatial 'modality factor' (a latent vector capturing imaging-specific characteristics). This disentangled representation supports semi-supervised segmentation using only a fraction of labeled images (matching fully supervised performance), multi-task learning (e.g. jointly regressing cardiac indices), multimodal pooling of MRI and CT data, and image-to-image synthesis between modalities via latent-space arithmetic (swapping modality factors). SDNet also demonstrates that its modality factor alone can predict the input imaging modality with high accuracy.

Cardiac MRI

Filter by Modality:
Cardiac MRI

Cardiac chamber segmentation

Filter by Disease / Trait:
Structural Heart & Cardiomyopathy

Segmentation

Filter by Task Type:
Segmentation & Detection

Hybrid

Filter by Architecture:
Hybrid / Multi-branch

Keras

Filter by Framework:
TensorFlow / Keras


Model ID: 0163

SSL-ECGv2 (Maternal/Fetal Stress Detection)

Queen's University / Technical University of Munich / University of Washington · 2021

graph_1

Code & model weights public

Self-supervised learning (SSL) model that identifies chronically stressed mother-fetus dyads from raw maternal abdominal ECG (aECG), which contains both maternal and fetal cardiac signals. Built on a self-supervised representation-learning approach originally developed for ECG-based emotion recognition, the model is pretrained on public ECG datasets and evaluated on a cohort of pregnant women with chronic stress exposure validated by psychological inventory, maternal hair cortisol, and the fetal stress index (FSI). Using maternal ECG alone with the publicly pretrained model, it detected the chronic-stress-exposure group with AUROC 0.982 and predicted psychological stress score (R2 0.943), FSI (R2 0.946), and maternal hair cortisol (R2 0.931).

Fetal ECG

Filter by Modality:
ECG

Fetal / maternal cardiac monitoring

Filter by Disease / Trait:
Other Conditions

Binary classification

Filter by Task Type:
Classification

Hybrid

Filter by Architecture:
Hybrid / Multi-branch

TensorFlow

Filter by Framework:
TensorFlow / Keras

CC BY-NC 4.0

Filter by License:
Non-commercial / Research-only


Model ID: 0154

·

Subject Count: 103

SVS-net

Shanghai Jiao Tong University (Qin Lab) / University of Texas Southwestern Medical Center · 2020

graph_1

Code & model weights public

Deep channel-attention network for segmenting the full coronary vessel tree from sequential X-ray coronary angiography (XCA) frames, rather than a single static image. An encoder-decoder architecture fuses temporal-spatial feature maps across the XCA sequence via skip connections, then uses channel-attention blocks in the decoder to refine features and separate thin vessel structures from complex, noisy backgrounds; a Dice loss addresses the severe foreground/background class imbalance typical of XCA. The authors report that SVS-net outperforms prior 2D and video-based baselines on both quantitative vessel-segmentation metrics and visual validation.

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

Keras

Filter by Framework:
TensorFlow / Keras


Model ID: 0129

·

Subject Count: 120

DeepIVUS

Emory University (Molony & Samady) · 2019

graph_1

Code & model weights public

Deep learning platform for fully automatic segmentation and phenotyping of coronary intravascular ultrasound (IVUS) pullbacks, packaged with a desktop GUI and CLI. A convolutional encoder-decoder network delineates the internal (lumen) and external elastic lamina borders on each cross-sectional IVUS frame; downstream rule-based analysis derives lumen area, plaque area, plaque burden, automatically flags lesions with plaque burden exceeding 40%, and reports minimum lumen area and maximum plaque burden along the pullback. Also supports end-diastolic gating and manual contour editing. Trained on 305 clinical IVUS pullbacks (270 train / 35 validation) from Philips and Boston Scientific catheters at Emory University; downstream evaluations have applied DeepIVUS to tasks such as automated detection of stent underexpansion.

Intravascular ultrasound (IVUS)

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

CNN (2D)

Filter by Architecture:
Convolutional (CNN)

TensorFlow

Filter by Framework:
TensorFlow / Keras

Apache 2.0

Filter by License:
Permissive


Model ID: 0104

·

Subject Count: 305

ECG-XPLAIM

National and Kapodistrian University of Athens / ETH Zurich (Pantelidis, Ruiperez-Campillo et al.) · 2025

graph_1

Code & model weights public

Explainable Inception-style 1D CNN for multi-label arrhythmia detection from 12-lead ECGs, integrating Grad-CAM visualization to highlight the waveform segments driving each prediction. Trained on MIMIC-IV-ECG and externally validated on PTB-XL across atrial fibrillation, sinus tachycardia, conduction disturbances (RBBB/LBBB/LAFB), long QT, Wolff-Parkinson-White pattern, and paced-rhythm detection, with all metrics exceeding 90% internally and strong generalization on external validation.

12-lead ECG

Filter by Modality:
ECG

Atrial fibrillation

Filter by Disease / Trait:
Arrhythmia

Multi-class classification

Filter by Task Type:
Classification

Multi-label classification

Filter by Task Type:
Classification

CNN (1D)

Filter by Architecture:
Convolutional (CNN)

TensorFlow

Filter by Framework:
TensorFlow / Keras

CC BY 4.0

Filter by License:
Open — Attribution


Model ID: 0064

·

Subject Count: 161,352

DeepStrain

Massachusetts General Hospital / Harvard-MIT HST (Morales et al.) · 2021

graph_1

Code & model weights public

Fully automated deep learning workflow for characterizing cardiac mechanics from balanced steady-state free-precession (bSSFP) cine cardiac MRI. It decouples two convolutional networks—a segmentation net (CarSON) and a 3D motion-estimation net (CarMEN)—to derive left- and right-ventricular volumes plus global and regional myocardial strain and strain rate without manual tracing. Trained and validated on healthy and cardiovascular-disease subjects and shown to be robust across MRI vendors, with excellent intra-scanner repeatability for strain. Developed at Massachusetts General Hospital and the Harvard-MIT Division of Health Sciences and Technology.

Cardiac MRI

Filter by Modality:
Cardiac MRI

Cardiac chamber segmentation

Filter by Disease / Trait:
Structural Heart & Cardiomyopathy

Myocardial strain (global/regional)

Filter by Disease / Trait:
Cardiac Function & Hemodynamics

Segmentation

Filter by Task Type:
Segmentation & Detection

Regression

Filter by Task Type:
Regression

Hybrid

Filter by Architecture:
Hybrid / Multi-branch

TensorFlow

Filter by Framework:
TensorFlow / Keras

Public Domain

Filter by License:
Permissive


Model ID: 0056

·

Subject Count: 150

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

Automatic 12-lead ECG diagnosis DNN

Universidade Federal de Minas Gerais (UFMG) · 2020

graph_1

Code & model weights public

1D residual neural network that screens 10-second, 12-lead ECG tracings for six common abnormalities: first- and second-degree AV block patterns, right and left bundle branch block, sinus bradycardia, atrial fibrillation, and sinus tachycardia. Reported F1 scores above 80% and specificity over 99% when benchmarked against cardiology residents. Developed at Universidade Federal de Minas Gerais and trained on the large Brazilian CODE-15% ECG dataset.

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)

TensorFlow

Filter by Framework:
TensorFlow / Keras

CC BY 4.0

Filter by License:
Open — Attribution


Model ID: 0010

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

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

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

PPG2ABP

BUET (Ibtehaz & Rahman) · 2020

graph_1

Code & model weights public

Translates a raw PPG waveform into a full continuous arterial blood-pressure waveform, from which systolic, diastolic, and mean arterial pressure are derived. Uses a two-stage cascaded 1D convolutional network in a U-Net style, with a coarse approximation stage followed by a refinement stage. Meets BHS Grade A and AAMI accuracy standards for diastolic and mean arterial pressure. Developed at BUET.

PPG / wearable

Filter by Modality:
PPG / Wearable

Blood pressure estimation

Filter by Disease / Trait:
Cardiac Function & Hemodynamics

Generation

Filter by Task Type:
Generation

Regression

Filter by Task Type:
Regression

CNN (1D)

Filter by Architecture:
Convolutional (CNN)

Keras

Filter by Framework:
TensorFlow / Keras

MIT

Filter by License:
Permissive


Model ID: 0049

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