Abstract:BACKGROUND AND OBJECTIVES: Arrhythmia classification based on electrocardiograms (ECG) can enhance clinical diagnostic efficiency. However, due to the significant differences in the number of different categories of heartbeats, the performance of classes with fewer samples in arrhythmia classification have not met expectations under the inter-patient paradigm. This paper aims to mitigate the adverse effects of category imbalance and improve arrhythmia classification performance.METHODS: We constructed a novel dual attention hybrid network (DA-Net) for arrhythmia classification under sample imbalance, based on modified convolutional networks with channel attention (MCC-Net) and sequence-to-sequence network with global attention (Seq2Seq). The refined local features of the input heartbeat are first extracted by MCC-Net and then sent to Seq2Seq for further feature fusion. By applying local and global attention in the feature extraction and fusion parts, respectively, the method fully fuses low-level feature details and high-level context information and enhances the ability to extract discriminative features.RESULTS: Based on the MIT-BIH arrhythmia database, under the inter-patient paradigm without any data augmentation methods, the proposed method achieved 99.98% accuracy (ACC) for five categories. The various performance indicators are as follows: Class N: sensitivity (SEN) = 99.96%, specificity (SPEC) = 99.93%, positive predictive value (PPV) = 99.99%; Class S: SEN = 99.67%, SPEC = 99.98%, PPV = 99.56%; Class V: SEN = 100%, SPEC = 99.99%, PPV = 99.91%; Class F: SEN = 100%, PPV = 99.98%, SPEC = 97.17%. In further experiments simulating extreme cases, the model still achieved ACC of 99.54% and 98.91% in the three-category and five-category categories when the training sample size was much smaller than the test sample.CONCLUSIONS: Without any data augmentation methods, the proposed model not only alleviates the negative impact of class imbalance and achieves excellent performance in all categories but also provides a new approach for dealing with class imbalance in arrhythmia classification. Additionally, our method demonstrates potential in conditions with fewer samples.

Research on Arrhythmia Classification Method Using Optimized Probabilistic Neural Network

HADLN: Hybrid Attention-Based Deep Learning Network for Automated Arrhythmia Classification

Cardiac Arrhythmia Classification Based on 3D Recurrence Plot Analysis and Deep Learning.

Arrhythmia classification based on multi-feature multi-path parallel deep convolutional neural networks and improved focal loss

Genetic algorithm for the optimization of features and neural networks in ECG signals classification

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

Arrhythmia Classification Based on Multiple Features Fusion and Random Forest Using ECG

ECG-Based Heartbeat Classification Using Two-Level Convolutional Neural Network and RR Interval Difference

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

Automated inter-patient arrhythmia classification with dual attention neural network

Automatic Classification System of Arrhythmias Using 12-Lead ECGs with a Deep Neural Network Based on an Attention Mechanism

A deep neural network based on multi-model and multi-scale for arrhythmia classification

Multi-classification method of arrhythmia based on multi-scale residual neural network and multi-channel data fusion

Optimal Multi-Stage Arrhythmia Classification Approach

Automatic electrocardiogram detection and classification using bidirectional long short-term memory network improved by Bayesian optimization

Automatic Arrhythmia Detection Based on Convolutional Neural Networks

Automated arrhythmia classification using depthwise separable convolutional neural network with focal loss

Automatic Detection of Arrhythmia Based on Multi-Resolution Representation of ECG Signal

Optimized Convulutional Neural Network by Genetic Alorithm for the Classification of Complex Arrhythmia.

ECG-Based Multiclass Arrhythmia Classification Using Beat-Level Fusion Network

Arrhythmia classification of LSTM autoencoder based on time series anomaly detection