An Electrocardiogram Classification Using a Multiscale Convolutional Causal Attention Network

Chaoqun Guo,Bo Yin,Jianping Hu

DOI: https://doi.org/10.3390/electronics13020326

IF: 2.9

2024-01-13

Electronics

Abstract:Electrocardiograms (ECGs) play a pivotal role in the diagnosis and prediction of cardiovascular diseases (CVDs). However, traditional methods for ECG classification involve intricate signal processing steps, leading to high design costs. Addressing this concern, this study introduces the Multiscale Convolutional Causal Attention network (MSCANet), which utilizes a multiscale convolutional neural network combined with causal convolutional attention mechanisms for ECG signal classification from the PhysioNet MIT-BIH Arrhythmia database. Simultaneously, the dataset is balanced by downsampling the majority class and oversampling the minority class using the Synthetic Minority Oversampling Technique (SMOTE), effectively categorizing the five heartbeat types in the test dataset. The experimental results showcase the classifier's performance, evaluated through accuracy, precision, sensitivity, and F1-score and culminating in an overall accuracy of 99.35%, precision of 96.55%, sensitivity of 96.73%, and an F1-recall of 96.63%, surpassing existing methods. Simultaneously, the application of this innovative data balancing technique significantly addresses the issue of data imbalance. Compared to the data before balancing, there was a significant improvement in accuracy for the S-class and the F-class, with increases of approximately 8% and 13%, respectively.

engineering, electrical & electronic,computer science, information systems,physics, applied

What problem does this paper attempt to address?

The paper attempts to address two major challenges in electrocardiogram (ECG) classification: 1. **Complexity and high cost of traditional methods**: Traditional ECG classification methods involve complex signal processing steps, leading to high design costs and difficulty in application in resource-constrained environments. 2. **Data imbalance problem**: In arrhythmia classification, datasets usually have a severe class imbalance problem, where the number of normal heartbeat samples far exceeds the number of abnormal heartbeat samples. This can cause the classifier to be biased towards the majority class, thereby affecting classification performance. To tackle these challenges, the authors propose a new deep learning model—Multiscale Convolutional Causal Attention Network (MSCANet), combined with data balancing techniques to improve classification performance. Specifically, the study addresses the aforementioned problems through the following methods: - **Multiscale Convolutional Neural Network**: Utilizes convolutional kernels of different sizes to extract spatial features of ECG signals. - **Causal Convolutional Attention Mechanism**: Captures the temporal features of the signals, enhancing the model's understanding of time dependencies. - **Data Balancing Techniques**: Balances the sample distribution in the dataset by downsampling the majority class and using SMOTE to upsample the minority class, thereby improving the classifier's ability to recognize the minority class. Experimental results show that the model outperforms existing methods on the MIT-BIH arrhythmia database, achieving an overall accuracy of 99.35%, a precision of 96.55%, a sensitivity of 96.73%, and an F1 score of 96.63%. Notably, for S-class and F-class samples, the accuracy after data balancing improved by approximately 8% and 13%, respectively.

An Electrocardiogram Classification Using a Multiscale Convolutional Causal Attention Network

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

EGCNet: a Hierarchical Graph Convolutional Neural Network for Improved Classification of Electrocardiograms

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

ECG signal classification based on deep CNN and BiLSTM

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

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

Multi-Scale and Multi-Channel Information Fusion for Exercise Electrocardiogram Feature Extraction and Classification

Automatic Recognition of Coronary Artery Disease and Congestive Heart Failure Using a Multi-Granularity Cascaded Hybrid Network

Intra-group and inter-group electrocardiograph coding image fusion and classification based on multi-scale group convolution feature fusion network

A convolutional neural network for ECG annotation as the basis for classification of cardiac rhythms

Deep Convolutional Neural Network Based ECG Classification System Using Information Fusion and One-Hot Encoding Techniques

Multimodal 12-lead ECG data classification using multi-label DenseNet for heart disease detection

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

Electrocardiogram Heartbeat Classification for Arrhythmias and Myocardial Infarction

CAT-Net: Convolution, attention, and transformer based network for single-lead ECG arrhythmia classification

SE-ECGNet: A Multi-scale Deep Residual Network with Squeeze-and-Excitation Module for ECG Signal Classification

ADCGNet: Attention-based Dual Channel Gabor Network Towards Efficient Detection and Classification of Electrocardiogram Images.

ECG Heartbeat Classification Using Convolutional Neural Networks.

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

Automatic Classification of Arrhythmias Using Multi-Branch Convolutional Neural Networks Based on Channel-Based Attention and Bidirectional LSTM.