sensors

Article
A Hybrid Deep CNN Model for Abnormal Arrhythmia
Detection Based on Cardiac ECG Signal
Amin Ullah 1,2 , Sadaqat ur Rehman 3,4 , Shanshan Tu 3, *, Raja Majid Mehmood 5 , Fawad 6 and
Muhammad Ehatisham-ul-haq 1

1 Software Engineering Department, University of Engineering and Technology Taxila, Punjab 47050, Pakistan;
[email protected] (A.U.); [email protected] (M.E.-u.-h.)
2 Center for Research in Computer Vision Lab (CRCV Lab), College of Engineering and Computer Science,
University of central Florida (UCF), Orlando, FL 32816, USA
3 Engineering Research Center of Intelligent Perception and Autonomous Control, Faculty of Information
Technology, Beijing University of Technology, Beijing 100124, China; [email protected]
4 Department of Computer Science, Namal Institute, Mianwali 42250, Pakistan
5 Information and Communication Technology Department, School of Electrical and Computer Engineering,
Xiamen University Malaysia, Sepang 43900, Malaysia; [email protected]
6 Telecommunication Engineering Department, University of Engineering and Technology Taxila,
Punjab 47050, Pakistan; [email protected]
* Correspondence: [email protected]

Abstract: Electrocardiogram (ECG) signals play a vital role in diagnosing and monitoring patients
suffering from various cardiovascular diseases (CVDs). This research aims to develop a robust algo-
rithm that can accurately classify the electrocardiogram signal even in the presence of environmental





 noise. A one-dimensional convolutional neural network (CNN) with two convolutional layers, two
Citation: Ullah, A.; Rehman, S.u.; Tu,
down-sampling layers, and a fully connected layer is proposed in this work. The same 1D data
S.; Mehmood, R.M.; Fawad; was transformed into two-dimensional (2D) images to improve the model’s classification accuracy.
Ehatisham-ul-haq, M. A Hybrid Deep Then, we applied the 2D CNN model consisting of input and output layers, three 2D-convolutional
CNN Model for Abnormal layers, three down-sampling layers, and a fully connected layer. The classification accuracy of 97.38%
Arrhythmia Detection Based on and 99.02% is achieved with the proposed 1D and 2D model when tested on the publicly available
Cardiac ECG Signal. Sensors 2021, 21, Massachusetts Institute of Technology-Beth Israel Hospital (MIT-BIH) arrhythmia database. Both
951. https://doi.org/10.3390/ proposed 1D and 2D CNN models outperformed the corresponding state-of-the-art classification
s21030951 algorithms for the same data, which validates the proposed models’ effectiveness.

Academic Editors: Zongyuan Ge,

Keywords: electrocardiogram signal; arrhythmia; classification; 2D CNN; MIT-BIH; arrhythmia
Yingying Zhu and Xiaofeng Zhu
database
Received: 13 December 2020
Accepted: 15 January 2021
Published: 1 February 2021

1. Introduction
Publisher’s Note: MDPI stays neutral
with regard to jurisdictional claims in
Cardiovascular issues are currently the primary cause of human morbidity, causing
published maps and institutional affil-
more than 17 million deaths each year. The World Heart Federation report witness about
iations. three fourth of the total cardiovascular disease (CVD) patients reside inside low-income
regions across the globe [1]. Electrocardiogram (ECG) records the electrical activity gen-
erated by heart muscle depolarizations, which propagate in pulsating electrical waves
towards the skin. Although the electricity amount is, in fact, very small, it can be picked
up reliably with ECG electrodes attached to the skin (in microvolts, or uV) [2]. ECG signals
Copyright: © 2021 by the authors.
contain no less than two critical pieces of statistics, including correlated to biomedicine’s
Licensee MDPI, Basel, Switzerland.
This article is an open access article
healthiness [3–5] and associated with personal credentials or biometrics [6–8]. As a result
distributed under the terms and
of its easiness, several ECG categorizations processes have been established, counting
conditions of the Creative Commons manuals methods [9,10] and machine learning approaches [11–16]. The manual process is
Attribution (CC BY) license (https:// complicated. It is used for transient signals like ECG, often necessary for machine learning
creativecommons.org/licenses/by/ procedures with excessive computer assets. For better classification accuracy, machine
4.0/).

Sensors 2021, 21, 951. https://doi.org/10.3390/s21030951 https://www.mdpi.com/journal/sensors

Sensors 2021, 21, 951 2 of 13

learning methods are preferred compared to manual processes, though, a useful algorithm
needed to diminish it.
One of the most common cardiovascular conditions is arrhythmias, where the heart-
beats pattern deviates from its routine. These irregular patterns require classification into
their subclasses; this information can be used to precisely suggest cure the patients. The
ECG is widely used to diagnose and predict the irregular pattern of the human heart
to diagnose cardiological diseases. The analysis of arrhythmia is primarily liable on the
ECG. It is a significant current medical instrument that can record cardiac excitability, the
process of transmission, and recovery. ECG is an essential and reliable diagnostic tool in
modern medicine—the reliable automation of the interpretation of ECG signals is extremely
beneficial for clinical routine and patient safety. Arrhythmia is an issue that deals with the
irregular activity and pattern of the human heart [17].
Artificial intelligence, especially the machine learning technique, is one of the most
useful tools in the prediction and diagnosis [18–22] of different types of fatal diseases,
specifically cardiovascular diseases. Its results even sometimes perform better than medical
experts [23]. Different types of medical data in the form of medical records, usually used
for medical purposes in various hospitals. Moreover, the medical data can be regenerate
quickly for different types of machine learning algorithms for various purposes. The first
machine is trained on the given data; then, it learns from the provided data. After training,
the device can detect and classify a patient, whether it is healthy or not, by visualizing the
different attributes of patient records. Hence, in this way, a machine discovers the patterns
in the given records that generally a human being cannot discover, because of deficiency of
time or a deficiency of view.
Many approaches have been used to perform the arrangement of ECG signals like k-
nearest neighbors (KNN), support vector machines (SVM), neural networks (NN), decision
trees, linear discriminant analysis (LDA), Bayesian classifiers, etc. SVM is one of the best
algorithms of machine learning technique (supervised classifier) [24,25], which is used in
the taxonomy of the ECG wave in the discovery of Arrhythmia disease. The combination
of SVM and LDA to classify six types of arrhythmia is presented in Reference [26]. An
efficient classification model, which is based on the NN and Machine Language Program
(MLP), gives better performance as compared to other feature extraction methods [27].
Deep Learning, such as Artificial Neural Networks (ANN), is successfully applied in
applications, such as information retrieval [28], image recognition, object tracking, and
language processing [29]. In Reference [30], a two-dimensional convolution neural network
(2D CNN) model is developed to classify the ECG signals for the diagnosis of arrhyth-
mia. However, in Reference [30], only a 2D CNN method is applied, while the 1D CNN
model is completely ignored. The whole objective of Deep Learning is to solve problems
characterized by high dimensionality, which have no rules.
CNN is used in many studies to isolate the best characteristics from the ECG wave and
analyze the extracted features for various determinations, i.e., detection of QRS wave, ST
segment, as shown in Figure 1 or classification of a heartbeat. 1D CNN is trained to extract
the best features from the ECG signal then classified these characteristics into five various
types of arrhythmia [31]. To remove noise from the wave, it uses the wavelet method. The
output layer is the last layer of the CNN model. It is also called a fully connected layer,
contains five neurons that classify five different types of arrhythmias.
In the same way, the 1D CNN model is used to classify four various types of arrhyth-
mia [32]. It also removes noise from the ECG signal in the pre-processing step, and then
the denoised signal is input to the CNN model. The softmax layer is the last layer of the
model, which gives four different outputs.
In the last few years, different pattern recognition techniques have been used in the
prediction and classification of arrhythmia disease [33–39]. Typically, these methods have
three main necessary steps; (i) pre-processing; (ii) feature extraction (iii) classification.
Initially, the ECG wave is cleaned by eliminating the different types of noise and outliers,
Sensors 2021, 21, 951 3 of 13

such as muscle contraction, baseline wanders, and power line interference using different
algorithms [40–47].

Figure 1. A model of an ECG signal. A different portion of the ECG signal is labeled to identify
the functionality.

In this research paper, a 1D and 2D CNN model are applied on 1D and 2D ECG signal,
which classified five (N (normal), VPC (Ventricular premature complexes), LBB (left bundle
branch), APC (atrial premature contraction), and RBB (right bundle branch)) and eight
(NOR (normal), VFW (ventricular flutter wave), PVC (premature ventricular contraction),
VEB (ventricular escape beat), RBB (right bundle branch block beat), LBB (left bundle
branch block beat), PAB (paced beat) and APC (atrial premature contraction)) different
types of arrhythmias, respectively. At last, the contributions of the proposed model are
as follows.
• The suggested technique did not demand the post-processing of the ECG signal.
• It does not need handcrafted feature extraction.
• The proposed model has lower computational complexity than the previous models
used to classify arrhythmia types.
It has better performance accuracy than the state-of-the-art algorithms for the arrhyth-
mia classification.

2. Materials and Methods

2.1. Proposed System for Arrhythmia Classification Using 1D CNN
There are three crucial steps in the whole process, i.e., signal pre-processing, extraction
of features from the data, and its classification. The overall process is shown in Figure 2.

Figure 2. Process of ECG signal classification.

2.1.1. Pre-processing of Data

Pre-processing is a significant and essential part of applying machine learning tech-
niques that are used to completeness and remove noise from the dataset. These processes
Sensors 2021, 21, 951 4 of 13

are used to perform an eloquent examination of the data, and last it provides an optimum
result. Firstly, we can apply the method of wavelet threshold and reconstruction algorithm
of wavelet decomposition together to eliminate the noise from the original ECG wave. The
technique of wavelet threshold can reduce electromyography noise, as well as power line
noise interference. In contrast, the reconstruction algorithm of wavelet decomposition
reduces the baseline drift noise from the noisy ECG signal. These two essential methods
are initially used to eliminate noise from the ECG wave so that the wave can be used for
further processing. Finally, the heartbeat signal as input data is put forward directly for the
CNN model so that the best features are extracted, and ECG signals can be classified.

2.1.2. 1D CNN Architecture

There are two main parts of CNN, extraction of features, and classification of features.
The feature extraction part of CNN is responsible for the automatic extraction of the best
characteristics from the ECG wave. These extracted characteristics are used for the accurate
categorization of the ECG signal. In other words, these two parts of CNN accomplish
the primary function of CNN [47]. The features extraction part of the CNN consists of a
convolution layer and sampling layer. The fuzzy filter of the convolution layer is used
to diminish the noise from the ECG wave, and then features of the ECG signal can be
enhanced. The convolution process is done between the upper layer feature vector and the
current layer convolution kernel layer. The activation function of CNN finally completes
convolution process calculations. The overall architecture of the proposed 1D CNN is
visualized in Figure 3 and outline in Table 1.

Figure 3. Proposed 1D CNN Model Architecture.

Table 1. The architecture of the proposed 1D CNN Model.

Layer # Type Kernel Size Stride # Kernel

1 1D Conv 5 1 128
2 Pooling 2 2 -
3 1D Conv 5 1 128
4 Pooling 2 2 -
5 Fully Connected - - 520
6 Output Layer - - 5

2.1.3. Cost Function

The efficiency of training a neural network is apparent using a cost function, which
signifies the proportion between the training sample and the gained output. The cost
function is controlled by exploiting the optimizer function. There are various kinds of cost
functions, though we use the cross-entropy function for loss measurement. To bound the
cost function, along with the learning rate, a gradient descent-based optimizer function
is applied.
−1
m ∑
Z= [ x ln a + (1 − x ) ln(1 − β)] (1)
Sensors 2021, 21, 951 5 of 13

where m is the number of training data, x is an expected value, and β is an actual value
from the output layer.

2.2. Method for Arrhythmia Classification Using 2D CNN

The accuracy obtained for arrhythmia classification through the 1D CNN model will
be improved by transforming this 1D ECG signal into 2D ECG images, which is given as
input to the 2D CNN model. The architecture of the proposed 2D CNN model is given in
Figure 4. The whole process contains four steps, i.e., signal processing, generation of 2D
images, augmentation of data, extraction of features from the data, and its categorization.
The overall procedures are shown in Figure 4.

Figure 4. Complete procedures in ECG signal classification.

2.2.1. Pre-Processing
The pre-processing step converts 1D ECG signal into 2D images, which are given
to the 2D CNN model as input to isolate different characteristics contained in the data.
Feature extraction helps in classifying various ECG types. This research study transforms
a single 1D ECG beat wave into a single 512 × 512 grey-scale image. ECG waves of the
MIT-BIH database are sliced into unique ECG beats based on Q-wave peak time and then
also labeled with arrhythmia type on the same criterion. Due to this reason, the first and
last 20 ECG waves before and after the Q-signal peak waves are chopped, and 2D images of
ECG are defined by considering the Q-wave peak time. We use Equation (2) to determine
the range of each ECG wave.

Generation of 2D Images
The one-dimensional convolutional neural network is mostly used for 1D signal,
which suffers from the issue of flexibility. To resolve such issues, we have transformed
the input signals to a two-dimensional form. The 2D CNN models are less prone to errors
and can adopt the 2D kernels to represent time series data. Moreover, a vast set of robust
methods available for the distinctive representation of 2D data. The 2D CNN can be
used to capture the microstructural detail of the input data. The ECGs are converted
in the 2D spectrogram by adapting the Continuous wave Transform method. The non-
stationary details with varying instantaneous frequency can efficiently be represented in
the frequency domain of continuous wavelet transform (CWT). The CWT can describe the
frequency along with the localized amplitude of the time-varying signals. To describe such
signals, we assume the signal to be stationary for the period of temporal window of finite
support. The transformation of 1D signal into 2D (512 × 512) images is presented in the
following expressions.

M −1
XCWT [r, c] = ∑ x [ j]h[r − j]e−i2πnj/M (2)
j =0

where M represents the window length, x [ j] denote the input signal. The log values of
XCWT [r, c] represents as spectrogram (512 × 512) images.
Sensors 2021, 21, 951 6 of 13

2.2.2. Augmentation of Data

When we supplement additional relevant information to our base data, this process
is called augmentation. It reduces the human effort required in adding other data and
improves the quality of data sometimes. This step might be helpful when 2D images
are used as input to CNN models. The effectiveness of data augmentation can vary for
different models. Some previous works achieved better results with augmentation, while
some underwent a decrease in performance.

2.2.3. Architecture of 2D CNN Model

This article uses the CNN model to classify ECG signals, which was introduced in
1989 to recognize handwritten zip codes. Due to the increase in free parameters of the
raw image, the existing feed-forward networks are not feasible to classify 2D images. But
embedding of multiple nonlinear filters in CNN models has made it possible to extract
various local features of 2D images by correlating spatially adjacent pixels.
Since it is feasible to extract and filter the spatial vicinity of 2D images through convo-
lution and down-sampling layers, therefore this research introduces the transformation of
1D ECG signal into 2D ECG images. This scheme increases the classification accuracy of
ECG signals to a level of not less than obtained via physician inspection. Hence, 2D CNN
model can replace the traditional diagnosis process of ECG arrhythmia through eye vision
inspection by medical experts. The architecture of the proposed model is given in Figure 5,
and an explanation of different layers is shown in Table 2.

Figure 5. The proposed architecture of 2D CNN model.

Table 2. The architecture of the proposed 2D CNN model.

Layer # Type Kernel Size Stride # Kernel Input Size

1 2D Conv 3×3 1 64 512 × 512 × 1
1 Pooling 2×2 2 512 × 512 × 64
2 2D Conv 3×3 1 128 256 × 256 × 64
2 Pooling 2×2 2 256 × 256 × 128
3 2D Conv 3×3 1 256 128 × 128 × 128
3 Pooling 2×2 2 128 × 128 × 256
4 2D Conv 3×3 1 512 64 × 64 × 256
4 Pooling 2×2 2 64 × 64 × 512
5 Fully Connected 4096 16 × 16 × 512
6 Output Layer 8 4096

The following steps are incorporated to make the algorithm work more accurately.

2.2.4. Activation Function

The activation function plays a crucial part in the classification process by assigning
kernel size and weighting the output of the CNN model [48]. Nonlinear activation functions
like leakage rectified linear units (LReLU), rectified linear units (ReLU), and exponential
Sensors 2021, 21, 951 7 of 13

linear units (ELU) have gained popularity in recent days. ReLU activation function is
amongst the commonly used activation functions [49]. It is approximately used in all CNN
algorithms to set all negative values equal to zero. These zero settings inhibit several nodes
from participating in the learning process. Other functions, like LReLU and ELU, which
provide small negative values, are rarely used in classification techniques. ReLU activation
function exhibits better results than LReLU activation function in arrhythmia classification;
that is why we have employed it in our classification model. The given equation can
mathematically denote the ReLU activation function.

ReLU ( x ) = max (0; x ) (3)

2.2.5. Cost Function

The cost function represents the difference between the targets and estimated labels.
The optimizer function is employed to reduce the gap. Various cost functions are used in
the neural network; however, the cross-entropy function the most frequently is used.

−1 N
n c=∑
C= ([yc ∗ ln ( ac ) + (1 − yc )ln (1 − ac )]) (4)
1N

where C represent the cost function that is desired to be minimized. The yc is the target
value, c is the class index. N is the total number of classes and a is true value. A gradient
descent with a learning rate of 0.001 is used as an optimization method. CNN model and it
reached the optimal point in less iteration.
Where m is the number of training data, x is an expected value, and β is an actual value
from the output layer. Gradient descent is used as an optimizer function with a learning
rate to reduce the error of cost function. Adagrad [50], Adam [51], and Adadelta [51] are
famous optimizer functions. However, we used the Adam optimizer function in our CNN
model, because in our experiments, it reaches optimal points quickly.

2.2.6. Validation of Data

Validation of data will determine any model which gains maximum accuracy with
the training data. If we did not use the validation data technique, then the model will fall
into the problem of overfitting. Broadly, the loss value of the CNN model is the validation
standard. Moreover, based on our notice, in different arrhythmia ECG signal classification,
maximum sensitivity will not be received if we have stopped the CNN model according to
lose value. Hence, we place the sensitivity mean of the validation data as the validation
standard. We halt the learning technique when the weighted sensitivity means is no more
increasing, and the evaluation of test data starts.

3. Experiments and Results

3.1. Data Set
The MIT-BIH database consists of 48 thirty minutes portions or pieces of 2-channel
ambulatory ECG records, received from 47 patients analyzed by the BIH Arrhythmia
Laboratory from 1975 to 1979. Twenty-three recordings were selected at random from
a range of 4000 24-h ambulatory ECG recordings composed of a diverse inhabitant of
indoor patients (60 out of 100) and outdoor patients (40 out of 100). The persist twenty-five
recordings were chosen from a similar set to contains significant, nevertheless, clinically
vital arrhythmias that will not be good symbolize in a less simple example. These recordings
were digitized at 360 samples/sec per channel with an 11-bit determination over a series of
10 mV. Independently more than two cardiologists explain each record; dispute solves to get
the computer-readable situation explanations for each record (about 110,000 interpretations)
are contained in this database [48].
Sensors 2021, 21, 951 8 of 13

3.2. Experimental Procedure

The proposed CNN classifier is implemented in one of the most essential and famous
machines-learning languages, Python, with an open-source library Tensor Flow developed
by Google for deep learning. To train the CNN model, it needs a lot of computational
power and training time. Therefore, we strongly recommended the Graphical processing
unit (GPU) system to decrease the learning time for the CNN model.

3.3. Performance Evaluation

The proposed model diagnoses and classifies the following five different types of
arrhythmia, i.e., N (normal), VPC (ventricular premature contraction), LBB (left bundle
branch block), APC (atrial premature contraction), and RBB (right bundle branch block)
with final accuracy 97.38%.
The precision-recall is a vital measurement technique used to assess recovery accuracy.

TP + TN
A= (5)
TP + TN + FP + FN

where Tp denotes the number of cases rightly classified as needed, Fp represents the number
of cases wrongly classified as needed, TN denotes the number of cases rightly classified as
not needed and FN represents the number of cases wrongly classified as not needed.

4. Discussions of Both Models

A detailed performance comparison between the proposed 1D CNN and 2D CNN
models is presented using confusion matrices for five and all eight classes, respectively.
The diagonal elements show the correctly classified classes, whereas anything off-diagonal
represents an incorrect classification. For the 1D ECG and 2D ECG data used in experiments,
results are presented for the proposed 1D CNN model (Figure 6) and the proposed 2D CNN
model (Figure 7). The average accuracy of these three models is presented by averaging
the diagonal values.

Figure 6. Illustration of confusion Matrix of 1D CNN model.

Sensors 2021, 21, 951 9 of 13

Figure 7. Illustration of confusion Matrix of 2D CNN model.

4.1. 1-D CNN Comparison with Other Algorithms

Table 3 shows the contrast between the optimized 1D CNN. It should be noted that there
are various methods that used 1D data directly to classify arrhythmia [52–58]. Among these
methods, SVM models achieved a lower classification accuracy (Reference [11]—89.72% and
Reference [54]—86.40%) when compared with the proposed model. We also used 1D ECG
signals as input to the 1D CNN model used in experiments and achieved a classification
accuracy of 97.38%. The features extraction part of the 1D CNN consists of a convolution layer
and sampling layer. The fuzzy filter of the convolution layer is used to diminish the noise from
the ECG wave, and then features of the ECG signal can be enhanced. The convolution process
is done between the upper layer feature vector and the current layer convolution kernel layer.
The activation function of CNN finally completes convolution process calculations. Therefore,
the proposed 1D CNN proved effective in classifying arrhythmia ECG signals, compared
with other state-of-the-art techniques.

Table 3. Comparison with other state-of-the-art algorithms. The proposed method is compared with
different approaches on the same benchmark dataset (MIT-BIH arrhythmia database).

S. No. Reference Year #Class Methods Accuracy

1 [52] 2007 5 NN 96.70%
2 [11] 2008 6 SVM 89.72%
3 [53] 2015 5 1D-CNN 95.14%
4 [54] 2016 5 1D-CNN 92.60%
5 [55] 2017 5 PNN 92.80%
6 [56] 2018 5 1D-CNN 90.00%
7 [57] 2019 5 1D-CNN 93.60%
8 [58] 2011 5 SVM, GA 97.30%
9 [59] 2013 5 NN, SVM 93.00%
10 [60] 2014 5 SVM 86.40%
11 Proposed Technique 2020 5 1D-CNN 97.38%
Sensors 2021, 21, 951 10 of 13

4.2. 2-D CNN Comparison with Other Algorithms

Table 4 displays the performance evaluation results of the proposed CNN algorithm
with other CNN algorithms. Our proposed 2D CNN model gained 99.02% average accuracy.
Recently, 2D CNN models have been employed to categorize the input ECG signals into
their respective classes. The input 1D ECGs are transformed into 2D before the feature
extraction process [16]. The 2D CNN models offer more distinctiveness and robustness
towards noise in the input signals. The proposed model employs the CWT to convert
the input 1D ECG into a 2D signal before extracting robust features. The proposed 2D
CNN model offers better accuracy than the FFNN [61], moreover, our proposed model
can categorize more classes as compare to FFNN. The proposed model consisting of three
convolutional layers, followed by three pooling layers and a fully-connected layer, is lower
in-depth than the AlexNet model. The VGGNet model consisting of ten convolution layer
requires more computational resources than our proposed 2D CNN model. The DenseNet
and ResNet are deeper networks that require more computational resources in comparison
to our model.

Table 4. Summarized performance evaluation results of different deep learning models on the
MIT-BIH arrhythmia database.

No. Reference Year #Class Methods Accuracy

1 [11] 2008 6 SVM 91.67%
2 [61] 2009 4 FFNN 96.94%
3 [62] 2009 5 PCA, ANN 98.30%
4 [63] 2010 3 LS-SVM 95.82%
5 [64] 2012 3 RFT 92.16%
6 [65] 2016 5 1D-CNN 96.40%
7 [66] 2018 2 DWT, DNN 98.33%
8 [67] 2018 5 2D-CNN 96.05%
9 [68] 2018 2 KNN 98.40%
10 [69] 2019 5 2D CNN 97.42%
11 [57] 2019 7 1D CNN 93.60%
12 Proposed Technique 2020 8 2D-CNN 99.02%

In Table 4, we have compared the accuracy of our proposed model with that of recently
reported works. The table shows our model outperforms most state-of-the-art model in
term of classification performance. The classification accuracy of our model is 99.02% better
than the techniques mentioned in References [11,57,62–69].

5. Conclusions and Future Work

Accurate categorization of ECG waves is very supportive in the prevention and
diagnosis of CVDs. Deep CNN has proved to be very operative in amplifying the accuracy
of diagnosis algorithms in the fusion of medicine and modern machine learning technology.
Noise removal in source ECG signals using the Wavelet algorithm enhances the quality
of the original ECG signal, improves the feature selection process, and hence, reduces the
classification error. Our suggested 1D CNN model applied to the ECG signal, which learns
useful attributes from the given data. Finally, it classifies the ECG signal automatically
based on the extracted features. Using a 1D CNN model for categorizing the ECG signal
provides an accuracy of 97.38%, which is better than the other algorithms used in previous
works. Likewise, implementing efficient and practical techniques to classify arrhythmia
ECG signal is achievable by transforming 1D ECG signal into 2D ECG images, and then
using this as an input to the 2D CNN algorithm. Our proposed technique can 99.02%
accurately classify arrhythmia by employing 2D images. This performance is a clear
Sensors 2021, 21, 951 11 of 13

indication of the effectiveness of the proposed 2D CNN model to classify arrhythmia with
transformed 2D ECG images.
In future work, we will design an integrated system to classify arrhythmia ECG
signals, which will monitor and scan the patient’s ECG via the internal camera of the robot
and will predict and diagnose the arrhythmia ECG signal to advise the medical expert. The
current research relies on the use of a single ECG signal. The use of multi-channel data for
categorizing ECG data will be useful in the future.

Author Contributions: Conceptualization, A.U.; Methodology, A.U., S.u.R., S.T.; Validation, A.U., F.,
R.M.M.; Formal Analysis, A.U., S.u.R.; Writing-Original Draft Preparation, A.U.; Writing-Review &
Editing, S.u.R., A.U., M.E.-u.-h.; Funding Acquisition, S.T. All authors have read and agreed to the
published version of the manuscript.
Funding: This work was supported in part by the China National Key Research and Development
Program under Grant 2018YFB0803600, in part by the Natural Science Foundation of China under
Grant 61801008, in part by the Scientific Research Common Program of Beijing Municipal Commis-
sion of Education under Grant KM201910005025, and in part by the Chinese Postdoctoral Science
Foundation under Grant 2020M670074.
Conflicts of Interest: The author declares that there is no conflict between authors.

References
1. Lackland, D.T.; Weber, M.A. Global Burden of Cardiovascular Disease and Stroke: Hypertension at the Core. Can. J. Cardiol. 2015,
31, 569–571. [CrossRef]
2. Evans, G.F.; Shirk, A.; Muturi, P.; Soliman, E.Z. Feasibility of using mobile ECG recording technology to detect atrial fibril-lation
in low-resource settings. Glob. Heart 2017, 12, 285–289. [CrossRef] [PubMed]
3. Sun, W.; Zeng, N.; He, Y. Morphological Arrhythmia Automated Diagnosis Method Using Gray-Level Co-Occurrence Matrix
Enhanced Convolutional Neural Network. IEEE Access 2019, 7, 67123–67129. [CrossRef]
4. Laguna, P.; Cortes, J.P.M.; Pueyo, E. Techniques for Ventricular Repolarization Instability Assessment From the ECG. Proc. IEEE
2016, 104, 392–415. [CrossRef]
5. Satija, U.; Ramkumar, B.; Manikandan, M.S. A New Automated Signal Quality-Aware ECG Beat Classification Method for
Unsupervised ECG Diagnosis Environments. IEEE Sensors J. 2018, 19, 277–286. [CrossRef]
6. Lynn, H.M.; Pan, S.B.; Kim, P. A Deep Bidirectional GRU Network Model for Biometric Electrocardiogram Classification Based
on Recurrent Neural Networks. IEEE Access 2019, 7, 145395–145405. [CrossRef]
7. Chu, Y.; Shen, H.; Huang, K. ECG Authentication Method Based on Parallel Multi-Scale One-Dimensional Residual Network
with Center and Margin Loss. IEEE Access 2019, 7, 51598–51607. [CrossRef]
8. Kim, H.; Chun, S.Y. Cancelable ECG Biometrics Using Compressive Sensing-Generalized Likelihood Ratio Test. IEEE Access 2019,
7, 9232–9242. [CrossRef]
9. Dokur, Z.; Olmez, T.; Yazgan, E. Comparison of discrete wavelet and Fourier transforms for ECG beat classification. Electron. Lett.
1999, 35, 1502. [CrossRef]
10. Banerjee, S.; Mitra, M. Application of Cross Wavelet Transform for ECG Pattern Analysis and Classification. IEEE Trans. Instrum.
Meas. 2014, 63, 326–333. [CrossRef]
11. Melgani, F.; Bazi, Y. Classification of electrocardiogram signals with support vector machines and particle swarm optimization.
IEEE Trans. Inf. Technol. Biomed. 2008, 12, 667–677. [CrossRef] [PubMed]
12. Venkatesan, C.; Karthigaikumar, P.; Paul, A.; Satheeskumaran, S.; Kumar, R. ECG signal preprocessing and SVM classifi-er-based
abnormality detection in remote healthcare applications. IEEE Access 2018, 6, 9767–9773. [CrossRef]
13. Zhu, W.; Chen, X.; Wang, Y.; Wang, L. Arrhythmia Recognition and Classification Using ECG Morphology and Segment Feature
Analysis. IEEE/ACM Trans. Comput. Biol. Bioinform. 2018, 16, 131–138. [CrossRef] [PubMed]
14. Raj, S.; Ray, K.C. ECG Signal Analysis Using DCT-Based DOST and PSO Optimized SVM. IEEE Trans. Instrum. Meas. 2017, 66,
470–478. [CrossRef]
15. Pasolli, E.; Melgani, F. Active Learning Methods for Electrocardiographic Signal Classification. IEEE Trans. Inf. Technol. Biomed.
2010, 14, 1405–1416. [CrossRef] [PubMed]
16. Li, Z.; Feng, X.; Wu, Z.; Yang, C.; Bai, B.; Yang, Q. Classification of Atrial Fibrillation Recurrence Based on a Convolution Neural
Network With SVM Architecture. IEEE Access 2019, 7, 77849–77856. [CrossRef]
17. Swathi, O.N.; Ganesan, M.; Lavanya, R. R peak detection and feature extraction for the diagnosis of heart diseases. In Proceedings
of the 2017 International Conference on Advances in Computing, Communications and Informatics (ICACCI), Udupi, India, 13–16 September
2017; Institute of Electrical and Electronics Engineers (IEEE): Piscataway Township, NJ, USA, 2017; pp. 2388–2391.
18. Parhi, K.K. VLSI Digital Signal Processing Systems: Design and Implementation; John Wiley & Sons: Hoboken, NJ, USA, 2007.
Sensors 2021, 21, 951 12 of 13

19. Medjahed, S.A.; Ouali, M.; Saadi, T.A.; Benyettou, A. An Optimization-Based Framework for Feature Selection and Parameters
Determination of SVMs. Int. J. Inf. Technol. Comput. Sci. 2015, 7, 1–9. [CrossRef]
20. Huertas-Fernandez, I.; Garcia-Gomez, F.J.; Garcia-Solis, D.; Benitez-Rivero, S.; Marin-Oyaga, V.A.; Jesus, S.; Cáceres-Redondo,
M.T.; Lojo, J.A.; Martín-Rodríguez, J.F.; Carrillo, F.; et al. Machine learning models for the differential diagnosis of vascular
parkinsonism and Parkin-son’s disease using [123 I] FP-CIT SPECT. Eur. J. Nucl. Med. Mol. Imaging 2015, 42, 112–119. [CrossRef]
21. Salvatore, C.; Cerasa, A.; Battista, P.; Gilardi, M.C.; Quattrone, A.; Castiglioni, I. Magnetic resonance imaging biomarkers for the
early diagnosis of Alzheimer’s disease: A machine learning approach. Front. Neurosci. 2015, 9, 307. [CrossRef]
22. Dreiseitl, S.; Ohno-Machado, L.; Kittler, H.; Vinterbo, S.; Billhardt, H.; Binder, M. A Comparison of Machine Learning Methods
for the Diagnosis of Pigmented Skin Lesions. J. Biomed. Inform. 2001, 34, 28–36. [CrossRef]
23. Kononenko, I. Machine learning for medical diagnosis: History, state of the art and perspective. Artif. Intell. Med. 2001, 23, 89–109.
[CrossRef]
24. Maglogiannis, I.G. Emerging Artificial Intelligence Applications in Computer Engineering: Real Word ai Systems with Applications in
Ehealth, Hci, Information Retrieval and Pervasive Technologies; Ios Press: Amsterdam, The Netherlands, 2007; Volume 160.
25. Prusty, M.R.; Chakraborty, J.; Jayanthi, T.; Velusamy, K. Performance comparison of supervised machine learning algorithms for
multiclass transient classification in a nuclear power plant. In Proceedings of the International Conference on Swarm, Evolu-tionary,
and Memetic Computing, Bhubaneswar, India,18–20 December 2014; Springer: Berlin/Heidelberg, Germany, 2014; pp. 111–122.
26. Song, M.H.; Lee, J.; Cho, S.P.; Lee, K.J.; Yoo, S.K. Support vector machine based arrhythmia classification using reduced features.
Int. J. Control Autom. Syst. 2005, 3, 571.
27. Das, M.K.; Ari, S. ECG Beats Classification Using Mixture of Features. Int. Sch. Res. Not. 2014, 2014, 1–12. [CrossRef]
28. Santur, Y.; Santur, S.G. Knowledge Mining Approach For Healthy Monitoring From Pregnancy Data With Big Volumes. Int. J.
Intell. Syst. Appl. Eng. 2016, 4, 141–145. [CrossRef]
29. Santur, Y.; Karakose, M.; Akin, E. Random Forest Based Diagnosis Approach for Rail Fault Inspection in Railways. In Proceedings
of the International Conference on Electrical and Electronics Engineering (Eleco 2015), Bursa, Turkey, 26–28 November 2015;
pp. 714–719.
30. Ullah, A.; Anwar, S.M.; Bilal, M.; Mahmoud, R.M. Classification of Arrhythmia by Using Deep Learning with 2-D ECG Spec-tral
Image Representation. Remote Sens. 2020, 12, 1685. [CrossRef]
31. Acharya, U.R.; Oh, S.L.; Hagiwara, Y.; Tan, J.H.; Adam, M.; Gertych, A.; San Tan, R. A Deep Convolutional Neural Network
Model to Classify Heartbeats. Comput. Biol. Med. 2017, 89, 389–396. [CrossRef]
32. Pyakillya, B.; Kazachenko, N.; Mikhailovsky, N. Deep Learning for ECG Classification. J. Phys. Conf. Ser. 2017, 913, 012004.
[CrossRef]
33. Alajlan, N.; Bazi, Y.; Melgani, F.; Malek, S.; Bencherif, M.A. Detection of premature ventricular contraction arrhythmias in
electrocardiogram signals with kernel methods. Signal Image Video Process. 2014, 8, 931–942. [CrossRef]
34. Rehman, S.U.; Waqas, M.; Tu, S.; Koubaa, A.; Rehman, O.; Ahmad, J.; Hanif, M.; Han, Z. Deep Learning Techniques for Future
Intelligent Cross-Media Retrieval. arXiv 2020, arXiv:2008.01191.
35. Alonso-Atienza, F.; Morgado, E.; Fernandez-Martinez, L.; García-Alberola, A.; Rojo-Álvarez, J.L. Detection of Life-Threatening
Arrhythmias Using Feature Selection and Support Vector Machines. IEEE Trans. Biomed. Eng. 2013, 61, 832–840. [CrossRef]
36. Alvarado, A.S.; Lakshminarayan, C.; Principe, J.C. Time-Based Compression and Classification of Heartbeats. IEEE Trans. Biomed.
Eng. 2012, 59, 1641–1648. [CrossRef] [PubMed]
37. Homaeinezhad, M.R.; Atyabi, S.A.; Tavakkoli, E.; Toosi, H.N.; Ghaffari, A.; Ebrahimpour, R. ECG arrhythmia recognition via a
neuro-SVM–KNN hybrid clas- sifier with virtual QRS image-based geometrical features. Expert Syst. Appl. 2012, 39, 2047–2058.
[CrossRef]
38. Javadi, M.; Arani, S.A.A.A.; Sajedin, A.; Ebrahimpour, R. Classification of ECG arrhythmia by a modular neural network based
on Mixture of Experts and Negatively Correlated Learning. Biomed. Signal Process. Control. 2013, 8, 289–296. [CrossRef]
39. Längkvist, M.; Karlsson, L.; Loutfi, A. Sleep Stage Classification Using Unsupervised Feature Learning. Adv. Artif. Neural Syst.
2012, 2012, 1–9. [CrossRef]
40. Sameni, R.; Shamsollahi, M.B.; Jutten, C.; Clifford, G.D. A Nonlinear Bayesian Filtering Framework for ECG Denoising. IEEE
Trans. Biomed. Eng. 2007, 54, 2172–2185. [CrossRef] [PubMed]
41. Tracey, B.H.; Miller, E.L. Nonlocal Means Denoising of ECG Signals. IEEE Trans. Biomed. Eng. 2012, 59, 2383–2386. [CrossRef]
42. Wang, J.; Ye, Y.; Pan, X.; Gao, X. Parallel-type fractional zero-phase filtering for ECG signal denoising. Biomed. Signal Process.
Control. 2015, 18, 36–41. [CrossRef]
43. Yadav, S.K.; Sinha, R.; Bora, P.K. Electrocardiogram signal denoising using non-local wavelet transform domain filtering.
IET Signal Process. 2015, 9, 88–96. [CrossRef]
44. Rehman, S.U.; Waqas, M.; Huang, Y.; Rehman, O.U.; Ahmad, B.; Ahmad, S. Unsupervised pre-trained filter learning approach for
efficient convolution neural network. Neurocomputing 2019, 365, 171–190. [CrossRef]
45. Rehman, S.U.; Tu, S.; Huang, Y.; Magurawalage, C.M.S.; Chang, C.-C. Optimization of CNN through Novel Training Strategy for
Visual Classification Problems. Entropy 2018, 20, 290. [CrossRef]
46. Rehman, S.U.; Tu, S.; Huang, G.; Liu, G. CSFL: A novel unsupervised convolution neural network approach for visual pattern
classification. AI Commun. 2017, 30, 311–324. [CrossRef]
Sensors 2021, 21, 951 13 of 13

47. Rehman, S.; Tu, S.; Huang, Y.; Yang, Z. Face recognition: A novel un-supervised convolutional neural network meth-od. In
Proceedings of the 2016 IEEE International Conference of Online Analysis and Computing Science (ICOACS), Chongqing, China,
28–29 May 2016; IEEE: Piscataway Township, NJ, USA, 2016; pp. 139–144.
48. Available online: https://www.physionet.org/physiobank/database/mitdb/ (accessed on 15 October 2020).
49. Tu, S.; Rehman, S.U.; Waqas, M.; Rehman, O.U.; Yang, Z.; Ahmad, B.; Halim, Z.; Zhao, W. Optimisation-based training of
evolutionary convolution neural network for visual classification applications. IET Comput. Vis. 2020, 14, 259–267. [CrossRef]
50. Abadi, M.; Agarwal, A.; Barham, P.; Brevdo, E.; Chen, Z.; Citro, C.; Ghemawat, S. Tensorflow: Large-scale machine learning on
heterogeneous distributed systems. arXiv 2016, arXiv:1603.04467.
51. Guler, I.; Ubeylı, E.D. ECG beat classifier designed by combined neural network model. Pattern Recognit. 2005, 38, 199–208.
[CrossRef]
52. Jiang, W.; Kong, S.G. Block-based neural networks for personalized ECG signal classification. IEEE Trans. Neural Netw. 2007, 18,
1750–1761. [CrossRef]
53. Kiranyaz, S.; Ince, T.; Gabbouj, M. Real-Time Patient-Specific ECG Classification by 1-D Convolutional Neural Networks. IEEE
Trans. Biomed. Eng. 2015, 63, 664–675. [CrossRef]
54. Zubair, M.; Kim, J.; Yoon, C. An Automated ECG Beat Classification System Using Convolutional Neural Networks. In
Proceedings of the 2016 6th International Conference on IT Convergence and Security (ICITCS), Prague, Czech Republic, 26–26
September 2016; pp. 1–5.
55. Gutiérrez-Gnecchi, J.A.; Morfin-Magaña, R.; Lorias-Espinoza, D.; Tellez-Anguiano, A.D.C.; Reyes-Archundia, E.; Méndez-Patiño,
A.; Castañeda-Miranda, R. DSP-based arrhythmia classification using wavelet transform and probabilistic neural network. Biomed.
Signal Process. Control. 2017, 32, 44–56. [CrossRef]
56. Rajkumar, A.; Ganesan, M.; Lavanya, R. Arrhythmia classification on ECG using Deep Learning. In Proceedings of the 2019 5th
International Conference on Advanced Computing & Communication Systems (ICACCS), Coimbatore, India, 15–16 March 2019; Institute
of Electrical and Electronics Engineers (IEEE): Piscataway Township, NJ, USA, 2019; pp. 365–369.
57. Inan, O.T.; Giovangrandi, L.; Kovacs, G.T. Robust neural-network-based classification of premature ventricular contractions using
wavelet transform and timing interval features. IEEE Trans. Biomed. Eng. 2006, 53, 2507–2515. [CrossRef]
58. Zadeh, A.E.; Khazaee, A. High Efficient System for Automatic Classification of the Electrocardiogram Beats. Ann. Biomed. Eng.
2011, 39, 996–1011. [CrossRef]
59. Martis, R.J.; Acharya, U.R.; Mandana, K.; Ray, A.; Chakraborty, C. Cardiac decision making using higher order spectra. Biomed.
Signal Process. Control. 2013, 8, 193–203. [CrossRef]
60. Joshi, N.P.; Topannavar, P.S. Support vector machine-based heartbeat classification. In Proceedings of the 4th IRF International
Conference, Bangalore, India, 4 May 2014; pp. 140–144.
61. Ur Rehman, S.; Huang, Y.; Tu, S.; Basharat, A. Learning a Semantic Space for Modeling Images, Tags and Feelings in Cross-Media
Search. In Pacific-Asia Conference on Knowledge Discovery and Data Mining; Springer: Cham, Switzerland, 2019; pp. 65–76.
62. Ince, T.; Kiranyaz, S.; Gabbouj, M. A Generic and Robust System for Automated Patient-Specific Classification of ECG Signals.
IEEE Trans. Biomed. Eng. 2009, 56, 1415–1426. [CrossRef]
63. Dutta, S.; Chatterjee, A.; Munshi, S. Correlation technique and least square support vector machine combine for frequency
domain based ECG beat classification. Med Eng. Phys. 2010, 32, 1161–1169. [CrossRef]
64. Kumar, R.G.; Kumaraswamy, Y.S. Investigating cardiac arrhythmia in ECG using random forest 400 classification. Int. J. Comput.
Appl. 2012, 37, 31–34.
65. Ince, T.; Kiranyaz, S.; Eren, L.; Askar, M.; Gabbouj, M. Real-time motor fault detection by 1-D 404 convolutional neu-ral networks.
IEEE Trans. Ind. Electron. 2016, 6311, 7067–7075. [CrossRef]
66. Nanjundegowda, R.; Meshram, V.; Jain University. Dayananda Sagar University Arrhythmia Detection Based on Hybrid Features
of T-wave in Electrocardiogram. Int. J. Intell. Eng. Syst. 2018, 11, 153–162. [CrossRef]
67. Zhai, X.; Tin, C. Automated ECG Classification Using Dual Heartbeat Coupling Based on Convolutional Neural Network. IEEE
Access 2018, 6, 27465–27472. [CrossRef]
68. Rangappa, V.G.; Prasad, S.V.A.V.; Agarwal, A. Classification of Cardiac Arrhythmia stages using Hybrid Features Extraction with
K-Nearest Neighbour classifier of ECG Signals. Learning 2018, 11, 21–32. [CrossRef]
69. Izci, E.; Ozdemir, M.A.; Degirmenci, M.; Akan, A. Cardiac Arrhythmia Detection from 2D ECG Images by Using Deep Learning
Technique. In 2019 Medical Technologies Congress (TIPTEKNO); Institute of Electrical and Electronics Engineers (IEEE): Piscataway
Township, NJ, USA, 2019; pp. 1–4.