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.

A Dual-Branch Convolutional Neural Network with Domain-Informed Attention for Arrhythmia Classification of 12-Lead Electrocardiograms

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

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

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

A Novel Multi-Scale Convolutional Neural Network for Arrhythmia Classification on Reduced-Lead ECGs

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

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

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

Automated inter-patient arrhythmia classification with dual attention neural network

Automated Heartbeat Classification Using 3-D Inputs Based on Convolutional Neural Network with Multi-Fields of View.

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

A Novel Interpretable Method Based on Dual-Level Attentional Deep Neural Network for Actual Multilabel Arrhythmia Detection

A novel multi-scale 2D CNN with weighted focal loss for arrhythmias detection on varying-dimensional ECGs

Vaccine development for group A streptococcus infections and associated disease.

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

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

Advanced Neural Network Architecture for Enhanced Multi-Lead ECG Arrhythmia Detection through Optimized Feature Extraction

Classification of Cardiac Arrhythmias from Single Lead ECG with a Convolutional Recurrent Neural Network

Automatic Arrhythmia Detection Based on Convolutional Neural Networks

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

ECG-Based Heart Arrhythmia Diagnosis Through Attentional Convolutional Neural Networks