Classification of Electrocardiogram Signals Based on Hybrid Deep Learning Models

Abstract

According to the analysis of the World Health Organization (WHO), the diagnosis and treatment of heart diseases is the most difficult task. Several algorithms for the classification of arrhythmic heartbeats from electrocardiogram (ECG) signals have been developed over the past few decades, using computer-aided diagnosis systems. Deep learning architecture adaption is a recent effective advancement of deep learning techniques in the field of artificial intelligence. In this study, we developed a new deep convolutional neural network (CNN) and bidirectional long-term short-term memory network (BLSTM) model to automatically classify ECG heartbeats into five different groups based on the ANSI-AAMI standard. End-to-end learning (feature extraction and classification work together) is done in this hybrid model without extracting manual features. The experiment is performed on the publicly accessible PhysioNet MIT-BIH arrhythmia database, and the findings are compared with results from the other two hybrid deep learning models, which are a combination of CNN and LSTM and CNN and Gated Recurrent Unit (GRU). The performance of the model is also compared with existing works cited in the literature. Using the SMOTE approach, this database was artificially oversampled to address the class imbalance problem. This new hybrid model was trained on the oversampled ECG database and validated using tenfold cross-validation on the actual test dataset. According to experimental observations, the developed hybrid model outperforms in terms of recall, precision, accuracy and F-score performance of the hybrid model are 94.36%, 89.4%, 98.36% and 91.67%, respectively, which is better than the existing methods.