Compact neural network algorithm for electrocardiogram classification.
Phys Eng Sci Med 2026 Jun 11. [Online ahead of print]

Abstract

In this paper, we present a powerful, compact electrocardiogram (ECG) classification algorithm for cardiac arrhythmia diagnosis that addresses the current reliance on deep learning and convolutional neural networks (CNNs) in ECG analysis. This work aims to reduce the demand for deep learning, which often requires extensive computational resources and large labeled datasets. Our approach introduces an artificial neural network (ANN) with a simple architecture combined with a compact, interpretable feature-engineering pipeline. A key contribution of this work is the incorporation of 17 engineered features that enable the extraction of critical patterns from raw ECG signals. By integrating mathematical transformations, signal processing methods, and data extraction algorithms, our model captures the morphological and physiological characteristics of ECG signals with high efficiency, without requiring deep learning. Our method demonstrates a similar performance to other state-of-the-art models in classifying five rhythm classes-normal sinus rhythm, sinus bradycardia, sinus tachycardia, ventricular flutter, and atrial fibrillation. Our algorithm achieved an accuracy of [Formula: see text] on the MIT-BIH and St. Petersburg INCART arrhythmia databases, with a Cohen's kappa of [Formula: see text] and a Matthews correlation coefficient of [Formula: see text]. The compactness of the model and its low inference latency on a standard CPU suggest that the approach is a promising candidate for deployment on resource-constrained platforms.

Authors+Show Affiliations

Frausto-Avila CMCentro de Física Aplicada y Tecnología Avanzada, Universidad Nacional Autónoma de México, Boulevard Juriquilla 3001, 76230, Querétaro, Querétaro, México.
Manriquez-Amavizca JPInstituto Tecnológico y de Estudios Superiores de Monterrey, Universidad Nacional Autónoma de México, Epigmenio González 500, Fracc, San Pablo, 76130, Querétaro, Querétaro, México.
Rocha-Robledo AKSCentro de Física Aplicada y Tecnología Avanzada, Universidad Nacional Autónoma de México, Boulevard Juriquilla 3001, 76230, Querétaro, Querétaro, México.
Quiroz-Juarez MACentro de Física Aplicada y Tecnología Avanzada, Universidad Nacional Autónoma de México, Boulevard Juriquilla 3001, 76230, Querétaro, Querétaro, México. maqj@fata.unam.mx.
U'Ren ABInstituto de Ciencias Nucleares, Universidad Nacional Autónoma de México, 04510, Querétaro, Querétaro, México.

Pub Type(s)

Journal Article

Language

eng

PubMed ID

42274990