Abstract:Electrocardiography (ECG) plays a pivotal role in monitoring cardiac health, yet the manual analysis of ECG signals is challenging due to the complex task of identifying and categorizing various waveforms and morphologies within the data. Additionally, ECG datasets often suffer from a significant class imbalance issue, which can lead to inaccuracies in detecting minority class samples. To address these challenges and enhance the effectiveness and efficiency of cardiac arrhythmia detection from imbalanced ECG datasets, this study proposes a novel approach. This research leverages the MIT-BIH arrhythmia dataset, encompassing a total of 109,446 ECG beats distributed across five classes following the Association for the Advancement of Medical Instrumentation (AAMI) standard. Given the dataset's inherent class imbalance, a 1D generative adversarial network (GAN) model is introduced, incorporating the Bi-LSTM model to synthetically generate the two minority signal classes, which represent a mere 0.73% fusion (F) and 2.54% supraventricular (S) of the data. The generated signals are rigorously evaluated for similarity to real ECG data using three key metrics: mean squared error (MSE), structural similarity index (SSIM), and Pearson correlation coefficient (r). In addition to addressing data imbalance, the work presents three deep learning models tailored for ECG classification: SkipCNN (a convolutional neural network with skip connections), SkipCNN+LSTM, and SkipCNN+LSTM+Attention mechanisms. To further enhance efficiency and accuracy, the test dataset is rigorously assessed using an ensemble model, which consistently outperforms the individual models. The performance evaluation employs standard metrics such as precision, recall, and F1-score, along with their average, macro average, and weighted average counterparts. Notably, the SkipCNN+LSTM model emerges as the most promising, achieving remarkable precision, recall, and F1-scores of 99.3%, which were further elevated to an impressive 99.60% through ensemble techniques. Consequently, with this innovative combination of data balancing techniques, the GAN-SkipNet model not only resolves the challenges posed by imbalanced data but also provides a robust and reliable solution for cardiac arrhythmia detection. This model stands poised for clinical applications, offering the potential to be deployed in hospitals for real-time cardiac arrhythmia detection, thereby benefiting patients and healthcare practitioners alike.

Fully automatic electrocardiogram classification system based on generative adversarial network with auxiliary classifier

Automated Heartbeat Classification Exploiting Convolutional Neural Network with Channel-Wise Attention

Generative Adversarial Network with Transformer Generator for Boosting ECG Classification

A Novel Temporal Generative Adversarial Network for Electrocardiography Anomaly Detection

Patient-Specific ECG Classification with Integrated Long Short-Term Memory and Convolutional Neural Networks

Arrhythmia Classification using CGAN-augmented ECG Signals

Electrocardiogram generation with a bidirectional LSTM-CNN generative adversarial network

ECG data enhancement method using generate adversarial networks based on BiLSTM and CBAM

An Automatic Cardiac Arrhythmia Classification System with Wearable Electrocardiogram

[An image classification method for arrhythmias based on Gramian angular summation field and improved Inception-ResNet-v2]

Hierarchical deep learning with Generative Adversarial Network for automatic cardiac diagnosis from ECG signals

GAN-SkipNet: A Solution for Data Imbalance in Cardiac Arrhythmia Detection Using Electrocardiogram Signals from a Benchmark Dataset

ECG Arrhythmias Detection Using Auxiliary Classifier Generative Adversarial Network and Residual Network

Research on ECG Signal Classification Based on Data Enhancement of Generative Adversarial Network

Arrhythmia Detection Using ECG-Based Classification with Prioritized Feature Subset Vector-Associated Generative Adversarial Network

Interpatient ECG Heartbeat Classification with an Adversarial Convolutional Neural Network

Electrocardiogram Quality Assessment with a Generalized Deep Learning Model Assisted by Conditional Generative Adversarial Networks

A hybrid deep learning network for automatic diagnosis of cardiac arrhythmia based on 12-lead ECG

ECG-Adv-GAN: Detecting ECG Adversarial Examples with Conditional Generative Adversarial Networks

ECG-ATK-GAN: Robustness against Adversarial Attacks on ECGs using Conditional Generative Adversarial Networks

SLC-GAN: An automated myocardial infarction detection model based on generative adversarial networks and convolutional neural networks with single-lead electrocardiogram synthesis