Next Article in Journal
Reduced Graphene Oxide and Polyaniline Nanofibers Nanocomposite for the Development of an Amperometric Glucose Biosensor
Next Article in Special Issue
Classification with a Deferral Option and Low-Trust Filtering for Automated Seizure Detection
Previous Article in Journal
Follower: A Novel Self-Deployable Action Recognition Framework
Previous Article in Special Issue
Comprehensive Review of Vision-Based Fall Detection Systems
 
 
Font Type:
Arial Georgia Verdana
Font Size:
Aa Aa Aa
Line Spacing:
Column Width:
Background:
Article

A Hybrid Deep CNN Model for Abnormal Arrhythmia Detection Based on Cardiac ECG Signal

1
Software Engineering Department, University of Engineering and Technology Taxila, Punjab 47050, Pakistan
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
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
6
Telecommunication Engineering Department, University of Engineering and Technology Taxila, Punjab 47050, Pakistan
*
Author to whom correspondence should be addressed.
Sensors 2021, 21(3), 951; https://doi.org/10.3390/s21030951
Submission received: 13 December 2020 / Revised: 9 January 2021 / Accepted: 15 January 2021 / Published: 1 February 2021
(This article belongs to the Special Issue Healthcare Monitoring and Management with Artificial Intelligence)

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 algorithm 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 down-sampling layers, and a fully connected layer is proposed in this work. The same 1D data was transformed into two-dimensional (2D) images to improve the model’s classification accuracy. Then, we applied the 2D CNN model consisting of input and output layers, three 2D-convolutional layers, three down-sampling layers, and a fully connected layer. The classification accuracy of 97.38% and 99.02% is achieved with the proposed 1D and 2D model when tested on the publicly available Massachusetts Institute of Technology-Beth Israel Hospital (MIT-BIH) arrhythmia database. Both proposed 1D and 2D CNN models outperformed the corresponding state-of-the-art classification algorithms for the same data, which validates the proposed models’ effectiveness.

1. Introduction

Cardiovascular issues are currently the primary cause of human morbidity, causing more than 17 million deaths each year. The World Heart Federation report witness about three fourth of the total cardiovascular disease (CVD) patients reside inside low-income regions across the globe [1]. Electrocardiogram (ECG) records the electrical activity generated 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 contain no less than two critical pieces of statistics, including correlated to biomedicine’s healthiness [3,4,5] and associated with personal credentials or biometrics [6,7,8]. As a result of its easiness, several ECG categorizations processes have been established, counting manuals methods [9,10] and machine learning approaches [11,12,13,14,15,16]. The manual process is complicated. It is used for transient signals like ECG, often necessary for machine learning procedures with excessive computer assets. For better classification accuracy, machine 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 heartbeats 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,19,20,21,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 arrhythmia. 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 arrhythmia [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,34,35,36,37,38,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, such as muscle contraction, baseline wanders, and power line interference using different algorithms [40,41,42,43,44,45,46,47].
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 arrhythmia 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.

2.1.1. Pre-processing of Data

Pre-processing is a significant and essential part of applying machine learning techniques that are used to completeness and remove noise from the dataset. These processes 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.

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.
Z = 1 m x ln a + 1 x ln 1 β
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.

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.
X C W T r ,   c = j = 0 M 1 x j h r j e i 2 π n j / M
where M represents the window length, x j denote the input signal. The log values of X C W T r ,   c represents as spectrogram (512 × 512) images.

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 convolution 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.
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 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.
R e L U x = m a x 0 ; x

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.
C = 1 n c = 1 N N y c * l n a c + 1 y c l n 1 a c
where C represent the cost function that is desired to be minimized. The y c 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].

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.
A = T P + T N T P + T N + F P + F N
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.

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,53,54,55,56,57,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.

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.
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,63,64,65,66,67,68,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 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 Commission 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. [Google Scholar] [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. [Google Scholar] [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. [Google Scholar] [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. [Google Scholar] [CrossRef] [Green Version]
  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. [Google Scholar] [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. [Google Scholar] [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. [Google Scholar] [CrossRef]
  8. Kim, H.; Chun, S.Y. Cancelable ECG Biometrics Using Compressive Sensing-Generalized Likelihood Ratio Test. IEEE Access 2019, 7, 9232–9242. [Google Scholar] [CrossRef]
  9. Dokur, Z.; Olmez, T.; Yazgan, E. Comparison of discrete wavelet and Fourier transforms for ECG beat classification. Electron. Lett. 1999, 35, 1502. [Google Scholar] [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. [Google Scholar] [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. [Google Scholar] [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. [Google Scholar] [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. [Google Scholar] [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. [Google Scholar] [CrossRef]
  15. Pasolli, E.; Melgani, F. Active Learning Methods for Electrocardiographic Signal Classification. IEEE Trans. Inf. Technol. Biomed. 2010, 14, 1405–1416. [Google Scholar] [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. [Google Scholar] [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. [Google Scholar]
  18. Parhi, K.K. VLSI Digital Signal Processing Systems: Design and Implementation; John Wiley & Sons: Hoboken, NJ, USA, 2007. [Google Scholar]
  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. [Google Scholar] [CrossRef] [Green Version]
  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. [Google Scholar] [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. [Google Scholar] [CrossRef] [Green Version]
  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. [Google Scholar] [CrossRef] [Green Version]
  23. Kononenko, I. Machine learning for medical diagnosis: History, state of the art and perspective. Artif. Intell. Med. 2001, 23, 89–109. [Google Scholar] [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. [Google Scholar]
  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. [Google Scholar]
  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. [Google Scholar]
  27. Das, M.K.; Ari, S. ECG Beats Classification Using Mixture of Features. Int. Sch. Res. Not. 2014, 2014, 1–12. [Google Scholar] [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. [Google Scholar] [CrossRef] [Green Version]
  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. [Google Scholar]
  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. [Google Scholar] [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. [Google Scholar] [CrossRef]
  32. Pyakillya, B.; Kazachenko, N.; Mikhailovsky, N. Deep Learning for ECG Classification. J. Phys. Conf. Ser. 2017, 913, 012004. [Google Scholar] [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. [Google Scholar] [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. [Google Scholar]
  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. [Google Scholar] [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. [Google Scholar] [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. [Google Scholar] [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. [Google Scholar] [CrossRef]
  39. Längkvist, M.; Karlsson, L.; Loutfi, A. Sleep Stage Classification Using Unsupervised Feature Learning. Adv. Artif. Neural Syst. 2012, 2012, 1–9. [Google Scholar] [CrossRef] [Green Version]
  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. [Google Scholar] [CrossRef] [PubMed] [Green Version]
  41. Tracey, B.H.; Miller, E.L. Nonlocal Means Denoising of ECG Signals. IEEE Trans. Biomed. Eng. 2012, 59, 2383–2386. [Google Scholar] [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. [Google Scholar] [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. [Google Scholar] [CrossRef] [Green Version]
  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. [Google Scholar] [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. [Google Scholar] [CrossRef] [Green Version]
  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. [Google Scholar] [CrossRef]
  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. [Google Scholar]
  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. [Google Scholar] [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. [Google Scholar]
  51. Guler, I.; Ubeylı, E.D. ECG beat classifier designed by combined neural network model. Pattern Recognit. 2005, 38, 199–208. [Google Scholar] [CrossRef]
  52. Jiang, W.; Kong, S.G. Block-based neural networks for personalized ECG signal classification. IEEE Trans. Neural Netw. 2007, 18, 1750–1761. [Google Scholar] [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. [Google Scholar] [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. [Google Scholar]
  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. [Google Scholar] [CrossRef] [Green Version]
  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. [Google Scholar]
  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. [Google Scholar] [CrossRef]
  58. Zadeh, A.E.; Khazaee, A. High Efficient System for Automatic Classification of the Electrocardiogram Beats. Ann. Biomed. Eng. 2011, 39, 996–1011. [Google Scholar] [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. [Google Scholar] [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. [Google Scholar]
  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. [Google Scholar]
  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. [Google Scholar] [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. [Google Scholar] [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. [Google Scholar]
  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. [Google Scholar] [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. [Google Scholar] [CrossRef]
  67. Zhai, X.; Tin, C. Automated ECG Classification Using Dual Heartbeat Coupling Based on Convolutional Neural Network. IEEE Access 2018, 6, 27465–27472. [Google Scholar] [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. [Google Scholar] [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. [Google Scholar]
Figure 1. A model of an ECG signal. A different portion of the ECG signal is labeled to identify the functionality.
Figure 1. A model of an ECG signal. A different portion of the ECG signal is labeled to identify the functionality.
Sensors 21 00951 g001
Figure 2. Process of ECG signal classification.
Figure 2. Process of ECG signal classification.
Sensors 21 00951 g002
Figure 3. Proposed 1D CNN Model Architecture.
Figure 3. Proposed 1D CNN Model Architecture.
Sensors 21 00951 g003
Figure 4. Complete procedures in ECG signal classification.
Figure 4. Complete procedures in ECG signal classification.
Sensors 21 00951 g004
Figure 5. The proposed architecture of 2D CNN model.
Figure 5. The proposed architecture of 2D CNN model.
Sensors 21 00951 g005
Figure 6. Illustration of confusion Matrix of 1D CNN model.
Figure 6. Illustration of confusion Matrix of 1D CNN model.
Sensors 21 00951 g006
Figure 7. Illustration of confusion Matrix of 2D CNN model.
Figure 7. Illustration of confusion Matrix of 2D CNN model.
Sensors 21 00951 g007
Table 1. The architecture of the proposed 1D CNN Model.
Table 1. The architecture of the proposed 1D CNN Model.
Layer #TypeKernel SizeStride# Kernel
11D Conv51128
2Pooling22-
31D Conv51128
4Pooling22-
5Fully Connected--520
6Output Layer--5
Table 2. The architecture of the proposed 2D CNN model.
Table 2. The architecture of the proposed 2D CNN model.
Layer #TypeKernel SizeStride# KernelInput Size
12D Conv3 × 3164512 × 512 × 1
1Pooling2 × 22 512 × 512 × 64
22D Conv3 × 31128256 × 256 × 64
2Pooling2 × 22 256 × 256 × 128
32D Conv3 × 31256128 × 128 × 128
3Pooling2 × 22 128 × 128 × 256
42D Conv3 × 3151264 × 64 × 256
4Pooling2 × 22 64 × 64 × 512
5Fully Connected 409616 × 16 × 512
6Output Layer 84096
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).
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.ReferenceYear#ClassMethodsAccuracy
1[52]20075NN96.70%
2[11]20086SVM89.72%
3[53]201551D-CNN95.14%
4[54]201651D-CNN92.60%
5[55]20175PNN92.80%
6[56]201851D-CNN90.00%
7[57]201951D-CNN93.60%
8[58]20115SVM, GA97.30%
9[59]20135NN, SVM93.00%
10[60]20145SVM86.40%
11Proposed Technique202051D-CNN97.38%
Table 4. Summarized performance evaluation results of different deep learning models on the MIT-BIH arrhythmia database.
Table 4. Summarized performance evaluation results of different deep learning models on the MIT-BIH arrhythmia database.
No.ReferenceYear#ClassMethodsAccuracy
1[11]20086SVM91.67%
2[61]20094FFNN96.94%
3[62]20095PCA, ANN98.30%
4[63]20103LS-SVM95.82%
5[64]20123RFT92.16%
6[65]201651D-CNN96.40%
7[66]20182DWT, DNN98.33%
8[67]201852D-CNN96.05%
9[68]20182KNN98.40%
10[69]201952D CNN97.42%
11[57]201971D CNN93.60%
12Proposed Technique202082D-CNN99.02%
Publisher’s Note: MDPI stays neutral with regard to jurisdictional claims in published maps and institutional affiliations.

Share and Cite

MDPI and ACS Style

Ullah, A.; Rehman, S.u.; Tu, S.; Mehmood, R.M.; Fawad; Ehatisham-ul-haq, M. A Hybrid Deep CNN Model for Abnormal Arrhythmia Detection Based on Cardiac ECG Signal. Sensors 2021, 21, 951. https://doi.org/10.3390/s21030951

AMA Style

Ullah A, Rehman Su, Tu S, Mehmood RM, Fawad, Ehatisham-ul-haq M. A Hybrid Deep CNN Model for Abnormal Arrhythmia Detection Based on Cardiac ECG Signal. Sensors. 2021; 21(3):951. https://doi.org/10.3390/s21030951

Chicago/Turabian Style

Ullah, Amin, Sadaqat ur Rehman, Shanshan Tu, Raja Majid Mehmood, Fawad, and Muhammad Ehatisham-ul-haq. 2021. "A Hybrid Deep CNN Model for Abnormal Arrhythmia Detection Based on Cardiac ECG Signal" Sensors 21, no. 3: 951. https://doi.org/10.3390/s21030951

APA Style

Ullah, A., Rehman, S. u., Tu, S., Mehmood, R. M., Fawad, & Ehatisham-ul-haq, M. (2021). A Hybrid Deep CNN Model for Abnormal Arrhythmia Detection Based on Cardiac ECG Signal. Sensors, 21(3), 951. https://doi.org/10.3390/s21030951

Note that from the first issue of 2016, this journal uses article numbers instead of page numbers. See further details here.

Article Metrics

Back to TopTop