CVAI Catalog

·

View Catalog

tune

4 models found

·

4 public code

·

2 public weights

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

CardioIncNet (PVC Beats Localization Pipeline)

Ukrainian Catholic University (Machine Learning Lab) (Petryshak, Kachko, Maksymenko, Dobosevych) · 2021

code

Training code public

Two-stage deep learning pipeline that localizes premature ventricular contraction (PVC) beats directly from raw, unsegmented ECG signal, without relying on hand-crafted features or pre-existing R-peak annotations. An encoder-decoder network first localizes the R-peak of every heartbeat (normal or anomalous); the resulting R-peak positions are then passed to CardioIncNet, a 1D InceptionTime-based classifier, which delineates each beat as healthy or PVC. Evaluated with both single-dataset and cross-dataset protocols across three public ECG databases, the pipeline reached F1 scores above 0.99 (single-dataset) and 0.979 (cross-dataset) for R-peak localization, and above 0.96 and 0.85 respectively for PVC beat classification.

Single-lead ECG

Filter by Modality:
ECG

Detection / localization

Filter by Task Type:
Segmentation & Detection

Hybrid

Filter by Architecture:
Hybrid / Multi-branch


Model ID: 0128

·

Subject Count: 47

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

ECG-DualNet++ XL

TU Darmstadt (Rohr, Reich, Hoog Antink et al.) · 2022

graph_1

Code & model weights public

Dual-encoder single-lead ECG classifier for atrial fibrillation detection that fuses a raw-signal branch with a spectrogram branch via axial attention and a Transformer. Originally developed as a graduate-course project at TU Darmstadt for the 2017 PhysioNet/CinC Challenge, and later extended in a 2023 follow-up study. Released in four sizes up to 130M parameters (S/M/L/XL), alongside a simpler CNN+LSTM variant.

Single-lead ECG

Filter by Modality:
ECG

Atrial fibrillation

Filter by Disease / Trait:
Arrhythmia

Binary classification

Filter by Task Type:
Classification

Hybrid

Filter by Architecture:
Hybrid / Multi-branch

PyTorch

Filter by Framework:
PyTorch

MIT

Filter by License:
Permissive


Model ID: 0015

·

Subject Count: 11,000