Abstract
Electrocardiogram (ECG) is widely used in computer-aided systems for arrhythmia detection because it provides essential information for the heart functionalities. The cardiologist uses it to diagnose and detect the abnormalities of the heart. Hence, automating the process of ECG heartbeat classification plays a vital role in the clinical diagnosis. In this paper, a two-stage hierarchical method is proposed using deep Convolution Neural Networks (CNN) to determine the category of the heartbeats in the first stage, and then classify the classes belonging to that category in the second stage. This work is based on 16 different classes from the public MIT-BIH arrhythmia dataset. But the MIT-BIH dataset is unbalanced, which degrades the classification accuracy of the deep learning models. This problem is solved by using an adaptive synthetic sampling technique to generate synthetic heartbeats to restore the balance of the dataset.
In this study, an overall accuracy of 97.30% and an average accuracy of 91.32% are obtained, which surpasses several ECG classification methods.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
References
World Health Organization. Cardiovascular diseases (CVDs) (2017). http://www.who.int/mediacentre/factsheets/fs317/en/
American Heart Association Arrhythmia (2017). https://www.heart.org/en/health-topics/consumer-healthcare/what-is-cardiovascular-disease
Artis, S.G., Mark, R.G., Moody, G.B.: Detection of atrial fibrillation using artificial neural networks. In: Proceedings of the Computers in Cardiology, Venice, Italy, 23–26 September 1991, pp. 173–176. IEEE, Piscataway (1991)
Kastor, J.A.: Arrhythmias, 2nd edn. W.B. Saunders, London (1994)
Moody, G.B., Mark, R.G.: The impact of the MIT-BIH arrhythmia database. IEEE Eng. Med. Biol. Mag. 20(3), 45–50 (2001)
El-Saadawy, H., Tantawi, M., Shedeed, H.A., Tolba, M.F.: Electrocardiogram (ECG) classification based on dynamic beats segmentation. In: Proceedings of the 10th International Conference on Informatics and Systems - INFOS’16 (2016). https://doi.org/10.1145/2908446.2908452
Perez, R.R., Marques, A., Mohammadi, F.: The application of supervised learning through feed-forward neural networks for ECG signal classification. In: Proceedings of the IEEE Canadian Conference on Electrical and Computer Engineering (CCECE), Vancouver, BC, Canada, 15–18 May 2016, pp. 1–4. IEEE, Piscataway (2016)
Zebardast, B., Ghaffari, A., Masdari, M.: A new generalized regression artificial neural networks approach for diagnosing heart disease. Int. J. Innov. Appl. Stud. 4, 679 (2013)
Alqudah, A.M., Albadarneh, A., Abu-Qasmieh, I., Alquran, H.: Developing of robust and high accurate ECG beat classification by combining gaussian mixtures and wavelets features. Australas. Phys. Eng. Sci. Med. 42(1), 149–157 (2019)
Li, D., Zhang, J., Zhang, Q., Wei, X.: Classification of ECG signals based on 1D convolution neural network. In: 2017 IEEE 19th International Conference on e-Health Networking, Applications and Services (Healthcom) (2017). https://doi.org/10.1109/healthcom.2017.8210784
Shaker, A.M., Tantawi, M., Shedeed, H.A., Tolba M.F.: Heartbeat classification using 1D convolutional neural networks. In: Hassanien, A., Shaalan, K., Tolba, M. (eds) Proceedings of the International Conference on Advanced Intelligent Systems and Informatics 2019. AISI 2019. Advances in Intelligent Systems and Computing, vol. 1058. Springer, Cham (2020)
Shaker, A.M., Tantawi, M., Shedeed, H.A., Tolba, M.F.: Generalization of convolutional neural networks for ECG classification using generative adversarial networks. IEEE Access 8, 35592–35605 (2020)
Yildirim, Ö.: A novel wavelet sequence based on deep bidirectional LSTM network model for ECG signal classification. Comput. Biol. Med. 96, 189–202 (2018). https://doi.org/10.1016/j.compbiomed.2018.03.016
Shaker, A.M., Tantawi, M., Shedeed, H.A., Tolba M.F.: Combination of convolutional and recurrent neural networks for heartbeat classification. In: Hassanien, AE., Azar, A., Gaber, T., Oliva, D., Tolba, F. (eds) Proceedings of the International Conference on Artificial Intelligence and Computer Vision (AICV2020). AICV 2020. Advances in Intelligent Systems and Computing, vol. 1153. Springer, Cham (2020)
Voulodimos, A., Doulamis, N., Doulamis, A., Protopapadakis, E.: Deep learning for computer vision: a brief review. Comput. Intell. Neurosci. 2018, 1–13 (2018). https://doi.org/10.1155/2018/7068349
Min, S., Lee, B., Yoon, S.: Deep learning in bioinformatics. Brief. Bioinf. 18, 851–869 (2017)
Bakator, M., Radosav, D.: Deep learning and medical diagnosis: a review of literature. Multimodal Technol. Interact. 2, 47 (2018). https://doi.org/10.3390/mti2030047
Martis, R.J., Acharya, U.R., Mandana, K., Ray, A.K., Chakraborty, C.: Application of principal component analysis to ECG signals for automated diagnosis of cardiac health. Expert Syst. Appl. 39, 11792–11800 (2012)
Yazdanian, H., Nomani, A., Yazdchi, M.R.: Autonomous detection of heartbeats and categorizing them by using support vector machines. IEEE (2013)
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)
Yu, S.N., Chou, K.T.: Integration of independent component analysis and neural networks for ECG beat classification. Expert Syst. Appl. 34, 2841–2846 (2008)
Martis, R.J., Acharya, U.R., Min, L.C.: ECG beat classification using PCA, LDA, ICA and discrete wavelet transform. Biomed. Sign. Process Contr. 8, 437–448 (2013)
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. Meth. Progr. Biomed. 127, 52–63 (2016)
El-Saadawy, H., Tantawi, M., Shedeed, H.A., Tolba, M.F.: Hybrid hierarchical method for electrocardiogram heartbeat classification. IET Sig. Process. 12(4), 506–513 (2018). https://doi.org/10.1049/iet-spr.2017.0108
Acharya, U.R., Oh, S.L., Hagiwara, Y., Tan, J.H., Adam, M., Gertych, A., San, T.R.: A deep convolutional neural network model to classify heartbeats. Comput. Biol. Med. (2017). https://doi.org/10.1016/j.compbiomed.2017.08.022
He, H., Bai, Y., Garcia, E.A., Li, S.: ADASYN: adaptive synthetic sampling approach for imbalanced learning, In: IEEE International Joint Conference on Neural Networks (IEEE World Congress on Computational Intelligence), pp. 1322–1328 (2008)
Simonyan, K., Zisserman, A.: Very deep convolutional networks for large-scale image recognition. In: ICLR (2015)
MIT-BIH Arrhythmias Database. http://www.physionet.org/physiobank/database/mitdb/. Accessed 3 Apr 2020
Kingma, D.P., Jimmy, B.: Adam: a method for stochastic optimization. CoRR, abs/1412.6980 (2014)
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2021 The Editor(s) (if applicable) and The Author(s), under exclusive license to Springer Nature Switzerland AG
About this paper
Cite this paper
Shaker, A.M., Tantawi, M., Shedeed, H.A., Tolba, M.F. (2021). Deep Convolutional Neural Networks for ECG Heartbeat Classification Using Two-Stage Hierarchical Method. In: Hassanien, A.E., Slowik, A., Snášel, V., El-Deeb, H., Tolba, F.M. (eds) Proceedings of the International Conference on Advanced Intelligent Systems and Informatics 2020. AISI 2020. Advances in Intelligent Systems and Computing, vol 1261. Springer, Cham. https://doi.org/10.1007/978-3-030-58669-0_12
Download citation
DOI: https://doi.org/10.1007/978-3-030-58669-0_12
Published:
Publisher Name: Springer, Cham
Print ISBN: 978-3-030-58668-3
Online ISBN: 978-3-030-58669-0
eBook Packages: Intelligent Technologies and RoboticsIntelligent Technologies and Robotics (R0)