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.

Inter-patient ECG classification with intra-class coherence based weighted kernel extreme learning machine

Heartbeat Classification Using Discrete Wavelet Transform and Kernel Principal Component Analysis

Inter-patient ECG heartbeat classification for arrhythmia classification: a new approach of multi-layer perceptron with weight capsule and sequence-to-sequence combination

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

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

Inter-patient Heartbeat Classification Based on Region Feature Extraction and Ensemble Classifier

ECG arrhythmia detection in an inter-patient setting using Fourier decomposition and machine learning

ECG Heartbeat Classification Using Deep Transfer Learning with Convolutional Neural Network and STFT Technique

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

A robust multiple heartbeats classification with weight-based loss based on convolutional neural network and bidirectional long short-term memory

An Energy-Efficient ECG Processor with Weak-Strong Hybrid Classifier for Arrhythmia Detection

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

Interpatient ECG Heartbeat Classification with an Adversarial Convolutional Neural Network

Imbalanced heartbeat classification using EasyEnsemble technique and global heartbeat information

ECG beat classification using machine learning and pre-trained convolutional neural networks

A Novel Imbalanced Dataset Mitigation Method and ECG Classification Model Based on Combined 1D_cbam-Autoencoder and Lightweight CNN Model.

Multilayer extreme learning machine-based unsupervised deep feature representation for heartbeat classification

Automated inter-patient arrhythmia classification with dual attention neural network

rECGnition_v1.0: Arrhythmia detection using cardiologist-inspired multi-modal architecture incorporating demographic attributes in ECG

Interpatient Heartbeat Classification Using Modified Residual Attention Network with Two-Phase Training and Assistant Decision

Multi-Label Classification of Arrhythmia for Long-Term Electrocardiogram Signals with Feature Learning