Skip to main content
Log in

Deep convolutional neural network application to classify the ECG arrhythmia

  • Original Paper
  • Published:
Signal, Image and Video Processing Aims and scope Submit manuscript

Abstract

The ECG signal is such a substantial means to reflect all the electrical activities of the cardiac system. Therefore, it is considered by the physician as the essential tools and materials to diagnose and treat heart diseases. To deal with different types of arrhythmia, the physician manually inspects the ECG heartbeat. Since there are tiny alternations in the amplitude, durations and therefore the morphology, the computer-based systems were needed to develop such solutions in order to help the physician to do their job. In this study, a novel tactic to automatically classify ten different arrhythmia types was developed depending on the deep learning theory. Consequently, the well-known convolutional neural network (CNN) approach was adopted to classify those different types of arrhythmia. The structure of the proposed model consists of 11 layers distributed as follows: four layers as convolution interchanged with other four layers of max pooling and finally three successfully connected layers. The experiment was conducted with the dataset which was downloaded from the Physionet in the Massachusetts Institute of Technology-Beth Israel Hospital database and then augmented to get sufficient and balanced dataset. To evaluate the performance of the proposed method and compare it with the previous algorithms, confusion matrix, sensitivity (SEN), specificity (SPE), precision (PRE), area under curve and receiver operating characteristic have been used and calculated. It has been found that performance from the proposed method is better than the existing methods based on CNN, and the accuracy is 99.84.

This is a preview of subscription content, log in via an institution to check access.

Access this article

Price excludes VAT (USA)
Tax calculation will be finalised during checkout.

Instant access to the full article PDF.

Fig. 1
Fig. 2
Fig. 3
Fig. 4
Fig. 5
Fig. 6

Similar content being viewed by others

References

  1. C. D. C. World Health Organization: https://www.who.int/mediacentre/factsheets/fs317/en. (2019)

  2. Abdalla, F.Y.O., Wu, L., Ullah, H., Ren, G., Noor, A., Zhao, Y.: ECG arrhythmia classification using artificial intelligence and nonlinear and nonstationary decomposition. Signal Image Video Process. 13, 1283–1291 (2019)

    Article  Google Scholar 

  3. Acharya, U.R., Sree, S.V., Swapna, G., Martis, R.J., Suri, J.S.: Automated EEG analysis of epilepsy: a review. Knowl. Based Syst. 45, 147–165 (2013)

    Article  Google Scholar 

  4. AAMI: Computer-aided diagnosis of cardiovascular disorders. Association for the Advancement of Medical Instrumentation (AAMI) (2008). Testing and reporting performance results of cardiac rhythm and ST segment measurement algorithms. American National Standards Institute (2008)

  5. Martis, R.J., Acharya, U.R., Adeli, H.: Current methods in electrocardiogram characterization. Comput. Biol. Med. 48, 133–149 (2014)

    Article  Google Scholar 

  6. Acharya, U.R., Oh, S.L., Hagiwara, Y., Tan, J.H., Adam, M., Gertych, A., et al.: A deep convolutional neural network model to classify heartbeats. Comput. Biol. Med. 89, 389–396 (2017)

    Article  Google Scholar 

  7. Acharya, U.R., Fujita, H., Lih, O.S., Adam, M., Tan, J.H., Chua, C.K.: Automated detection of coronary artery disease using different durations of ECG segments with convolutional neural network. Knowl. Based Syst. 132, 62–71 (2017)

    Article  Google Scholar 

  8. Fan, X., Yao, Q., Cai, Y., Miao, F., Sun, F., Li, Y.: Multiscaled fusion of deep convolutional neural networks for screening atrial fibrillation from single lead short ECG recordings. IEEE J. Biomed Health Inform. 22, 1744–1753 (2018)

    Article  Google Scholar 

  9. Kohler, B.U., Hennig, C., Orglmeister, R.: The principles of software QRS detection. IEEE Eng. Med. Biol. Mag. 21, 42–57 (2002)

    Article  Google Scholar 

  10. Clifford, G. D., Azuaje, F., McSharry, P.: Advanced methods and tools for ecg data analysis. Artech House Publishers (2006)

  11. Hamdi, S., Ben Abdallah, A., Bedoui, M.H.: Real time QRS complex detection using DFA and regular grammar. Biomed. Eng. Online 16, 31 (2017)

    Article  Google Scholar 

  12. Fukushima, K.: Neocognitron: a self organizing neural network model for a mechanism of pattern recognition unaffected by shift in position. Biol. Cybern 36, 193–202 (1980)

    Article  Google Scholar 

  13. Lecun, Y., Bottou, L., Bengio, Y., Haffner, P.: Gradient-based learning applied to document recognition. Proc. IEEE 86, 2278–2324 (1998)

    Article  Google Scholar 

  14. Zubair, M., Kim, J., Yoon, C., IEEE: An automated ECG beat classification system using convolutional neural networks. In: 2016 6th International Conference on it Convergence and Security, pp. 335–339 (2016)

  15. Bouvrie, J.: Notes on convolutional neural networks. In: Pract, pp. 47–60 (2006)

  16. Diederik, J.L.B., Kingma, P.: ADAM: a method for stochastic optimization. ICLR (2015)

  17. XavierGloro, Y.: Understanding the difficulty of training deep feedforward neural networks. In: International Conference on Artificial Intelligence and Statistics, AISTATS (2010)

  18. Zaplata, F., Kasal, M., IEEE: SDR Implementation for DCF77 (2013)

  19. Kutlu, Y., Kuntalp, D.: A multi-stage automatic arrhythmia recognition and classification system. Comput. Biol. Med. 41, 37–45 (2011)

    Article  Google Scholar 

  20. Das, M.K., Ari, S.: ECG beats classification using mixture of features. Int. Sch. Res. Not. 2014, 178436–178436 (2014)

    Google Scholar 

  21. Elhaj, F.A., Salim, N., Harris, A.R., Swee, T.T., Ahmed, T.: Arrhythmia recognition and classification using combined linear and nonlinear features of ECG signals. Comput. Methods Programs Biomed. 127, 52–63 (2016)

    Article  Google Scholar 

  22. Rajesh, K.N.V.P.S., Uhuli, R.: Classification of ECG heartbeats using nonlinear decomposition methods and support vector machine. Comput. Biol. Med. 87, 271–284 (2017)

    Article  Google Scholar 

  23. Rajesh, K.N.V.P.S., Dhuli, R.: Classification of imbalanced ECG beats using re-sampling techniques and AdaBoost ensemble classifier. Biomed. Signal Process. Control 41, 242–254 (2018)

    Article  Google Scholar 

  24. Martis, R.J., Acharya, U.R., Min, L.C.: ECG beat classification using PCA, LDA, ICA and Discrete Wavelet Transform. Biomed. Signal Process. Control 8, 437–448 (2013)

    Article  Google Scholar 

  25. Martis, R.J., Acharya, U.R., Lim, C.M., Suri, J.S.: Characterization of ECG beats from cardiac arrhythmia using discrete cosine transform in PCA framework. Knowl. Based Syst. 45, 76–82 (2013)

    Article  Google Scholar 

  26. Li, P., Liu, C., Wang, X., Zheng, D., Li, Y., Liu, C.: A low-complexity data-adaptive approach for premature ventricular contraction recognition. SIViP 8, 111–120 (2014)

    Article  Google Scholar 

  27. Alajlan, N., Bazi, Y., Melgani, F., Malek, S., Bencherif, M.A.: Detection of premature ventricular contraction arrhythmias in electrocardiogram signals with kernel methods. SIViP 8, 931–942 (2014)

    Article  Google Scholar 

  28. Hammad, M., Maher, A., Wang, K., Jiang, F., Amrani, M.: Detection of abnormal heart conditions based on characteristics of ECG signals. Measurement 125, 634–644 (2018)

    Article  Google Scholar 

  29. Sahoo, S., Kanungo, B., Behera, S., Sabut, S.: Multiresolution wavelet transform based feature extraction and ECG classification to detect cardiac abnormalities. Measurement 108, 55–66 (2017)

    Article  Google Scholar 

  30. Yang, W., Si, Y., Wang, D., Guo, B.: Automatic recognition of arrhythmia based on principal component analysis network and linear support vector machine. Comput. Biol. Med. 101, 22–32 (2018)

    Article  Google Scholar 

  31. Lobabi-Mirghavami, H., Abdali-Mohammadi, F., Faith, A.: A novel grammar-based approach to atrial fibrillation arrhythmia detection for pervasive healthcare environments,". J. Comput. Secur. 2, 155–163 (2016)

    Google Scholar 

  32. de Albuquerque, V.H.C., Nunes, T.M., Pereira, D.R., Luz, EJdS, Menotti, D., Papa, J.P., et al.: Robust automated cardiac arrhythmia detection in ECG beat signals. Neural Comput. Appl. 29, 679–693 (2018)

    Article  Google Scholar 

  33. Li, T., Zhou, M.: ECG classification using wavelet packet entropy and random forests. Entropy 18, 285 (2016)

    Article  Google Scholar 

  34. Martis, R.J., Acharya, U.R., Mandana, K.M., Ray, A.K., Chakraborty, C.: Cardiac decision making using higher order spectra. Biomed. Signal Process. Control 8, 193–203 (2013)

    Article  Google Scholar 

Download references

Acknowledgements

This paper is supported by the National Natural Science Foundation of China, China (Grant Number: 61671185].

Author information

Authors and Affiliations

Authors

Corresponding author

Correspondence to Longwen Wu.

Additional information

Publisher's Note

Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.

Rights and permissions

Reprints and permissions

About this article

Check for updates. Verify currency and authenticity via CrossMark

Cite this article

Abdalla, F.Y.O., Wu, L., Ullah, H. et al. Deep convolutional neural network application to classify the ECG arrhythmia. SIViP 14, 1431–1439 (2020). https://doi.org/10.1007/s11760-020-01688-2

Download citation

  • Received:

  • Revised:

  • Accepted:

  • Published:

  • Issue Date:

  • DOI: https://doi.org/10.1007/s11760-020-01688-2

Keywords

Navigation