CVAI Catalog

·

View Catalog

tune

5 models found

·

5 public code

·

5 public weights

ENCASE

Peking University (Hong et al.) · 2019

graph_1

Code & model weights public

Ensemble classifier combining hand-engineered expert features with a deep convolutional neural network for classifying cardiac rhythm from a single-lead ECG recording into normal sinus rhythm, atrial fibrillation, another rhythm, or too noisy to classify. A large set of expert features (from time-, frequency-, and template-based analysis) is fed into a gradient-boosted tree classifier (AdaBoost), and its output is combined with a separate deep CNN operating directly on the raw waveform; combining both feature families measurably outperformed either alone. ENCASE won 1st place in the PhysioNet/Computing in Cardiology Challenge 2017 (single-lead AF classification) with an overall F1 score of 0.83 on the official hidden test set, and remains a widely cited example of combining classical signal-processing features with deep representations for ECG classification.

Single-lead ECG

Filter by Modality:
ECG

Atrial fibrillation

Filter by Disease / Trait:
Arrhythmia

Multi-class classification

Filter by Task Type:
Classification

Hybrid

Filter by Architecture:
Hybrid / Multi-branch

GPL 3.0

Filter by License:
Copyleft


Model ID: 0117

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

ESI (ECG Semantic Integrator)

Rice University · convnextv2_base · 2024

graph_1

Code & model weights public

Multimodal ECG model that pairs a 1D ConvNeXtV2 signal encoder with a BioLinkBERT text encoder, trained with a joint contrastive-and-captioning objective using LLM-generated descriptions of ECG demographics and waveform patterns in place of raw clinical reports. Validated on arrhythmia diagnosis and ECG-based subject identification, reaching an AUROC of 0.938 fine-tuned and 0.812 zero-shot on PTB-XL diagnostic classification. Developed at Rice University.

12-lead ECG

Filter by Modality:
ECG

Clinical text

Filter by Modality:
Text & EHR

Multimodal

Filter by Modality:
Multimodal

General Purpose / Multi-task

Filter by Disease / Trait:
General / Foundation

Multi-label classification

Filter by Task Type:
Classification

Embedding

Filter by Task Type:
Representation Learning

Hybrid

Filter by Architecture:
Hybrid / Multi-branch

PyTorch

Filter by Framework:
PyTorch

GPL 3.0

Filter by License:
Copyleft


Model ID: 0019

·

Subject Count: 64,037

PTB-XL benchmark ECG classifier (xresnet1d101)

Fraunhofer HHI (Strodthoff et al.) · 2021

graph_1

Code & model weights public

Reference benchmark suite for the PTB-XL ECG dataset, providing pretrained xresnet1d, InceptionTime, LSTM, and CPC-pretrained models for 71-label, diagnostic, sub-diagnostic, and super-diagnostic classification of ECG findings. Widely used as a standardized baseline for comparing new ECG classification methods. Developed by Strodthoff et al. at Fraunhofer HHI.

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)

PyTorch

Filter by Framework:
PyTorch

GPL 3.0

Filter by License:
Copyleft


Model ID: 0029

·

Subject Count: 18,885