Addressing Class Imbalance in ECG Arrhythmia Classification Using Latent Diffusion and Quantum-Enhanced Generative Modeling

Class imbalance in clinical electrocardiogram (ECG) datasets limits the diagnostic sensitivity of automated arrhythmia classifiers, particularly for rare but clinically significant beat types. We propose a three-stage hybrid generative pipeline that combines a spectral-guided conditional variational autoencoder (cVAE), a class-conditional latent denoising diffusion probabilistic model (DDPM), and a Quantum Latent Refinement (QLR) module built on parameterized quantum circuits, implemented and evaluated using a classical quantum-circuit simulator, to augment minority arrhythmia classes, and present results based on the MIT-BIH Arrhythmia Database. The QLR module applies a bounded residual correction guided by Maximum Mean Discrepancy minimization to align synthetic latent distributions with real class-specific latent banks. A lightweight 1D MobileNetV2 classifier evaluated over ten independent random seeds and four augmentation ratios serves as the downstream benchmark. Our findings establish latent diffusion augmentation as an effective strategy for imbalanced ECG classification. To our knowledge, the proposed QLR module is the first use of a parameterized quantum circuit as a distributional refiner within a generative augmentation pipeline. While its performance is comparable to that of the classical latent diffusion framework under the present experimental conditions, the proposed approach demonstrates the feasibility of integrating quantum latent operators into generative medical AI pipelines and provides a foundation for future investigations on quantum-enhanced representation learning and data augmentation.

Authors

Institutions

Publication Details

Journal
AI in Medicine
Published
2026-08-24
DOI
https://doi.org/10.3390/aimed1030023
Primary Topic
ECG Monitoring and Analysis
Type
article
Field-Weighted Citation Impact
0.00

Funders

Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

Addressing Class Imbalance in ECG Arrhythmia Classification Using Latent Diffusion and Quantum-Enhanced Generative Modeling

Georgios D. Barmparis, Georgios Neofotistos, Giorgos P. Tsironis, Georgios Kritopoulos
AI in Medicine
ECG Monitoring and Analysis
article

Addressing Class Imbalance in ECG Arrhythmia Classification Using Latent Diffusion and Quantum-Enhanced Generative Modeling

Georgios D. Barmparis, Georgios Neofotistos, Giorgos P. Tsironis, Georgios Kritopoulos
article en

Abstract

Class imbalance in clinical electrocardiogram (ECG) datasets limits the diagnostic sensitivity of automated arrhythmia classifiers, particularly for rare but clinically significant beat types. We propose a three-stage hybrid generative pipeline that combines a spectral-guided conditional variational autoencoder (cVAE), a class-conditional latent denoising diffusion probabilistic model (DDPM), and a Quantum Latent Refinement (QLR) module built on parameterized quantum circuits, implemented and evaluated using a classical quantum-circuit simulator, to augment minority arrhythmia classes, and present results based on the MIT-BIH Arrhythmia Database. The QLR module applies a bounded residual correction guided by Maximum Mean Discrepancy minimization to align synthetic latent distributions with real class-specific latent banks. A lightweight 1D MobileNetV2 classifier evaluated over ten independent random seeds and four augmentation ratios serves as the downstream benchmark. Our findings establish latent diffusion augmentation as an effective strategy for imbalanced ECG classification. To our knowledge, the proposed QLR module is the first use of a parameterized quantum circuit as a distributional refiner within a generative augmentation pipeline. While its performance is comparable to that of the classical latent diffusion framework under the present experimental conditions, the proposed approach demonstrates the feasibility of integrating quantum latent operators into generative medical AI pipelines and provides a foundation for future investigations on quantum-enhanced representation learning and data augmentation.

AI in MedicineVol. 1(3)
Harvard University (US), University of Crete (GR), Foundation for Research and Technology Hellas (GR)
Office of Naval Research Global
Openalex Percentile: Top 10%
ECG Monitoring and Analysis
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.