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

Vollständiger Abstract

Worum geht es in dieser Arbeit?

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.