CVAI Catalog

·

View Catalog

tune

5 models found

·

5 public code

·

3 public weights

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

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

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