Abstract:Background: Analysis of electrocardiogram (ECG) provides a straightforward and non-invasive approach for cardiologists to diagnose and classify the nature and severity of variant cardiac diseases including cardiac arrhythmia. However, the interpretation and analysis of ECG are highly working-load demanding, and the subjective may lead to false diagnoses and heartbeats classification. In recent years, many deep learning works showed an excellent role in accurate heartbeats classification. However, the imbalance of heartbeat classes is universal in most of the available ECG databases since abnormal heartbeats are always relatively rare in real life scenarios. In addition, many existing approaches achieved prominent results by removing noise and extracting features in data preprocessing, which relies heavily on powerful computers. It is a pressing need to develop efficient and automatic light weighted algorithms for accurate heartbeats classification that can be used in portable ECG sensors. Objective: This study aims at developing a robust and efficient deep learning method, which can be embedded into wearable or portable ECG monitors for classifying heartbeats. Methods: We proposed a novel and light weighted deep learning architecture with weight-based loss based on a convolutional neural network (CNN) and bidirectional long short-term memory (Bi-LSTM) that can automatically identify five types of ECG heartbeats according to the AAMI EC57 standard. It was also true that the raw ECG signals were simply segmented without noise removal and other feature extraction processing. Moreover, to tackle the challenge of classification bias due to imbalanced ECG datasets for different types of arrhythmias, we introduced a weight-based loss function to reduce the influence of over-weighted categories in the ECG dataset. For avoiding the influence of the division of validation dataset, k-fold method was adopted to improve the reliability of the model. Results: The proposed algorithm is trained and tested on MIT-BIH Arrhythmia Database, and achieves an average of 99.33% accuracy, 93.67% sensitivity, 99.18% specificity, 89.85% positive prediction, and 91.65% F1 score.

A Multi-Scale Convolutional Neural Network for Heartbeat Classification

A Deep Multi-Scale Convolutional Neural Network For Classifying Heartbeats

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

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

Feature Enrichment Based Convolutional Neural Network for Heartbeat Classification from Electrocardiogram

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

Multi-Scale and Attention Based ResNet for Heartbeat Classification.

ECG Classification Using 1-D Convolutional Deep Residual Neural Network.

An Improved Convolutional Neural Network Based Approach for Automated Heartbeat Classification.

ECG Heartbeat Classification Using Convolutional Neural Networks.

A 1D Convolutional Neural Network for Heartbeat Classification from Single Lead ECG.

Automated Heartbeat Classification Based on Convolutional Neural Network with Multiple Kernel Sizes

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

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

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

Automated Heartbeat Classification Based on Deep Neural Network with Multiple Input Layers

An Electrocardiogram Classification Using a Multiscale Convolutional Causal Attention Network

Arrhythmia detection based on multi-scale fusion of hybrid deep models from single lead ECG recordings: A multicenter dataset study

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

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

Heartbeat classification method combining multi-branch convolutional neural networks and Transformer