24 models found
·
23 public code
·
16 public weights
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.
Model ID: 0137
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.
Model ID: 0148
·
Subject Count: 60
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.
Model ID: 0160
Reinforcement-learning-based unsupervised domain adaptation framework for spatio-temporal (2D+time) echocardiography segmentation, extending the authors' earlier RL4Seg work to full-length video sequences. RL4Seg3D uses a sliding-window approach supporting high-resolution, full-sized inputs, and fuses multiple reward mechanisms to improve segmentation reliability without requiring additional expert annotations in the target domain. Trained and evaluated on a large dataset of over 30,000 echocardiography videos, it outperforms baselines and foundation models on overall segmentation accuracy as well as echocardiography-specific metrics including anatomical/temporal validity and mitral-valve-commissure landmark precision, and supports test-time optimization via calibrated uncertainty estimates.
Model ID: 0159
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).
Model ID: 0133
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.
Model ID: 0130
·
Subject Count: 59
Pediatric-specific extension of EchoNet-Dynamic: a video-based deep learning model that segments the left ventricle and estimates ejection fraction (EF) from apical-4-chamber (A4C) and parasternal short-axis (PSAX) pediatric echocardiogram clips. Because adult-trained echo models generalize poorly to children (who vary widely in heart size, rate, and image quality), EchoNet-Peds was trained from scratch on a dedicated pediatric video dataset. It segments the LV with a Dice similarity coefficient of 0.89 in both views, estimates EF with a mean absolute error of 3.66%, and identifies pediatric systolic dysfunction with an AUC of 0.95, significantly outperforming an adult-trained model applied to the same pediatric data.
Model ID: 0126
·
Subject Count: 1,923
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.
Model ID: 0113
·
Subject Count: 263
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.
Model ID: 0115
·
Subject Count: 500
Self-supervised model that performs single-frame digital-subtraction-angiography-style vessel/background separation directly from a single live (non-subtracted) coronary angiogram frame, then supports fine-tuned coronary vessel segmentation. A U-Net-style network is pretrained via an image-to-image translation objective on 58,128 unannotated angiography DICOM series (3,756 patients), then fine-tuned for vessel segmentation on just 40 expert-annotated frames, reaching a Dice of 0.828 on the held-out fine-tuning set and a new state-of-the-art Dice of 0.755 on the public XCAD benchmark. Intended to help clinicians visualize potential stenosis sites without requiring true two-frame digital subtraction acquisition.
Model ID: 0105
·
Subject Count: 3,796
nnU-Net-based segmentation network that detects and delineates stenotic lesions directly from X-ray coronary angiography frames, developed for the ARCADE (MICCAI 2023) stenosis-detection challenge. A companion model (YOLO-Angio, same team) handles vessel-tree segmentation; StenUNet focuses specifically on pixel-wise localization of stenotic regions. Placed 3rd overall among ARCADE challenge entrants with an F1 score of 0.5348 on the hold-out test set, within 0.0005 of the 2nd-place team.
Model ID: 0101
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.
Model ID: 0099
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.
Model ID: 0076
·
Subject Count: 129
Framework for training an echocardiography left-ventricle segmentation model purely by data-free knowledge distillation: a ConvLSTM-based student network learns to reproduce the masks produced by an EchoNet-Dynamic (DeepLabV3-ResNet50) teacher on entirely synthetic echo videos, with no real labeled data or even real videos required. Achieves state-of-the-art results identifying end-diastolic/end-systolic frames, reaching segmentation quality close to real-data training with substantially fewer weights; also introduces a human-annotation-free evaluation method using a large auxiliary model.
Model ID: 0065
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.
Model ID: 0074
·
Subject Count: 8,104
Adapts the Segment Anything Model (SAM) to echocardiography video segmentation by giving it a space-time memory that carries both spatial and temporal cues, so that only the first frame of a video needs an external point prompt and every subsequent frame is segmented from a propagated memory prompt instead. A memory reinforcement mechanism uses each frame's predicted mask to suppress speckle-noise features before they are written back into memory, addressing a key failure mode of naively adapting video object segmentation (e.g. XMem) to noisy ultrasound. Built on SAMUS (an ultrasound-adapted SAM) with a frozen SAM backbone and only the image-encoder adapter layers trained. On the semi-supervised CAMUS and EchoNet-Dynamic benchmarks (only end-diastole/end-systole frames labeled), MemSAM reaches 93.3% and 92.8% mean Dice respectively, outperforming UNet, SwinUNet, H2Former, and prior medical-SAM adaptations (MedSAM, MSA, SAMed, SonoSAM, SAMUS) with far fewer prompts, and derives LVEF (via Simpson's biplane method of disks) with a Pearson correlation of 78.9% against ground truth on CAMUS. Training/inference code is public (MIT license); only the starting SAM ViT-B checkpoint is linked for download, not a separately released fine-tuned MemSAM checkpoint.
Model ID: 0098
·
Subject Count: 10,530
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.
Model ID: 0081
·
Subject Count: 10,030
Multimodal cardiac MRI foundation model that fuses 3D+T cine CMR (short-axis and long-axis views) with tabular patient health records (demographics, metabolic, and lifestyle factors) from 42,000 UK Biobank participants. Two-stage self-supervised pretraining -- masked-image reconstruction, then imaging-tabular contrastive alignment -- produces representations that transfer to whole-heart segmentation, cardiac phenotype/physiological-feature regression, and cardiac/metabolic disease classification within one unified framework.
Model ID: 0062
·
Subject Count: 74,916
Open-source pipeline that classifies aortic stenosis (AS) severity from transthoracic echocardiography by combining structural and functional information. Video-based R(2+1)D convolutional networks read six B-mode and color Doppler views while a segmentation model measures peak aortic-jet velocity, and an ensemble integrates these into a final severity prediction. Trained on 210,193 images from Kaiser Permanente Northern California and validated across held-out, temporally distinct, and external Stanford and Cedars-Sinai cohorts, reaching AUCs up to 0.96–0.99 for severe AS. Developed by the Ouyang lab.
Model ID: 0054
Foundation model for cine cardiac MRI, self-supervised via masked autoencoding on nearly 75,000 UK Biobank scans. Uses a Vision Transformer with a convolutional stem, unified across long-axis and short-axis views. Fine-tuned checkpoints are released for ventricle and myocardium segmentation, ejection-fraction regression, cardiovascular disease classification, and landmark localization across several public benchmark datasets (ACDC, M&Ms, M&Ms2, EMIDEC, and others).
Model ID: 0003
·
Subject Count: 74,916
End-to-end pipeline for apical-4-chamber echocardiogram videos that segments the left ventricle, estimates ejection fraction on a beat-to-beat basis, and classifies cardiomyopathy with reduced ejection fraction. Combines a DeepLabV3-ResNet50 segmentation model with a 3D CNN (R2+1D/R3D/MC3) initialized on the Kinetics-400 video dataset. Trained on the public EchoNet-Dynamic dataset released alongside it, and one of the most widely reused open echocardiography models since its 2020 Nature publication. Developed by Stanford University.
Model ID: 0036
·
Subject Count: 10,030
Ensemble of ten self-configuring nnU-Net models (five 2D, five 3D) that segments the left ventricle, right ventricle, and myocardium from short-axis cardiac cine MRI. Won the 2020 M&Ms challenge, a multi-centre, multi-vendor, multi-disease benchmark spanning scanners from four vendors and three countries, demonstrating strong generalization across acquisition protocols. Developed by DKFZ, the group behind the widely used nnU-Net framework.
Model ID: 0005
·
Subject Count: 350
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.
Model ID: 0004
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.
Model ID: 0001