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.

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

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

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

Multi-label Classification of Abnormalities in 12-Lead ECG Using Deep Learning.

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

Automatic Classification Method of Arrhythmias Based on 12-Lead Electrocardiogram

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

ECG-based multi-class arrhythmia detection using spatio-temporal attention-based convolutional recurrent neural network

Automated inter-patient arrhythmia classification with dual attention neural network

ECG signal classification based on deep CNN and BiLSTM

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

Automatic Arrhythmia Detection Based on Convolutional Neural Networks

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

Attention-assisted hybrid CNN-BILSTM-BiGRU model with SMOTE–Tomek method to detect cardiac arrhythmia based on 12 - lead electrocardiogram signals

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

Two-Dimensional ECG-based Cardiac Arrhythmia Classi�cation using DSE-ResNet

Automatic Detection of ECG Abnormalities by using an Ensemble of Deep Residual Networks with Attention

Visualization deep learning model for automatic arrhythmias classification

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

A Neural Network for High-Precise and Well-Interpretable Electrocardiogram Classification

Automatic Detection for Multi-Labeled Cardiac Arrhythmia Based on Frame Blocking Preprocessing and Residual Networks