Abstract
With age, our blood vessels are prone to aging, which induces cardiovascular disease. As an important basis for diagnosing heart disease and evaluating heart function, the electrocardiogram (ECG) records cardiac physiological electrical activity. Abnormalities in cardiac physiological activity are directly reflected in the ECG. Thus, ECG research is conducive to heart disease diagnosis. Considering the complexity of arrhythmia detection, we present an improved convolutional neural network (CNN) model for accurate classification. Compared with the traditional machine learning methods, CNN requires no additional feature extraction steps due to the automatic feature processing layers. In this paper, an improved CNN is proposed to automatically classify the heartbeat of arrhythmia. Firstly, all the heartbeats are divided from the original signals. After segmentation, the ECG heartbeats can be inputted into the first convolutional layers. In the proposed structure, kernels with different sizes are used in each convolution layer, which takes full advantage of the features in different scales. Then a max-pooling layer followed. The outputs of the last pooling layer are merged and as the input to fully-connected layers. Our experiment is in accordance with the AAMI inter-patient standard, which included normal beats (N), supraventricular ectopic beats (S), ventricular ectopic beats (V), fusion beats (F), and unknown beats (Q). For verification, the MIT arrhythmia database is introduced to confirm the accuracy of the proposed method, then, comparative experiments are conducted. The experiment demonstrates that our proposed method has high performance for arrhythmia detection, the accuracy is 99.06%. When properly trained, the proposed improved CNN model can be employed as a tool to automatically detect different kinds of arrhythmia from ECG.
Similar content being viewed by others
References
United Nations. Department of economic and social affairs population division. World population aging 2015. New York, 2015.
Tekeste, T. et al., Ultra-low power QRS detection and ECG compression architecture for IoT healthcare devices. IEEE Transactions on Circuits and Systems I: Regular Papers 66(2):669–679, 2019.
Beach, C. et al., An ultra low power Personalizable wrist worn ECG monitor integrated with IoT infrastructure. IEEE Access 6:44010–44021, 2018.
Martis, R. J. et al., Computer aided diagnosis of atrial arrhythmia using dimensionality reduction methods on transform domain representation. Biomedical Signal Processing and Control 13(1):295–305, 2014.
Martis, R. J. et al., Application of higher order statistics for atrial arrhythmia classification. Biomedical Signal Processing and Control 8(6):888–900, 2013.
De Chazal, P., O'Dwyer, M., and Reilly, R. B., Automatic classification of heartbeats using ECG morphology and heartbeat interval features. IEEE Transactions on Biomedical Engineering 51(7):1196–1206, 2004.
de Lannoy, G., Francois, D., Delbeke, J., and Verleysen, M., Weighted conditional random fields for supervised interpatient heartbeat classification. IEEE Trans Biomed Eng 59(1):241–247, 2012.
Li, T. and M. Zhou, ECG classification usingwavelet packet entropy and random forests. Entropy, 18(8), (2016).
Teijeiro, T., Felix, P., Presedo, J., and Castro, D., Heartbeat classification using abstract features from the Abductive interpretation of the ECG. IEEE J Biomed Health Inform 22(2):409–420, 2018.
Luo, K., et al., Patient-specific deep architectural model for ECG classification. Journal of Healthcare Engineering, 2017. (2017).
Tripathy, R. K., Deb, S., and Dandapat, S., Analysis of physiological signals using state space correlation entropy. Healthcare Technology Letters 4(1):30–33, 2017.
Rostaghi, M., and Azami, H., Dispersion entropy: A measure for time-series analysis. IEEE Signal Processing Letters 23(5):610–614, 2016.
Faziludeen, S., and Sankaran, P., ECG beat classification using evidential K -nearest Neighbours. Procedia Computer Science 89:499–505, 2016.
Raj, S., Ray, K. C., and Shankar, O., Cardiac arrhythmia beat classification using DOST and PSO tuned SVM. Computer Methods and Programs in Biomedicine 136:163–177, 2016.
Clifford, G.D., et al. AF classification from a short single lead ECG recording: The PhysioNet/computing in cardiology challenge 2017. in Computing in Cardiology, (2017).
Sodmann, P., et al., A convolutional neural network for ECG annotation as the basis for classification of cardiac rhythms. Physiological Measurement. 39(10), (2018).
Mar, T. et al., Optimization of ECG classification by means of feature selection. IEEE Transactions on Biomedical Engineering 58(8):2168–2177, 2011.
Shyu, L. Y., Wu, Y. H., and Hu, W., Using wavelet transform and fuzzy neural network for VPC detection from the Holter ECG. IEEE Transactions on Biomedical Engineering 51(7):1269–1273, 2004.
Belkheiri, M., Z. Douidi, and A. Belkheiri, ECG beats extraction and classification using radial basis function neural networks, in Lecture Notes in Electrical Engineering. p. 127–136, (2013).
Chen, Y., and Yang, H., Self-organized neural network for the quality control of 12-lead ECG signals. Physiological Measurement 33(9):1399–1418, 2012.
Polanía, L. F., and Plaza, R. I., Compressed sensing ECG using restricted Boltzmann machines. Biomedical Signal Processing and Control 45:237–245, 2018.
Acharya, U. R. et al., A deep convolutional neural network model to classify heartbeats. Computers in Biology and Medicine 89:389–396, 2017.
Kiranyaz, S., Ince, T., and Gabbouj, M., Real-time patient-specific ECG classification by 1-D convolutional neural networks. IEEE Trans Biomed Eng 63(3):664–675, 2016.
Acharya, U. R. et al., Automated detection of arrhythmias using different intervals of tachycardia ECG segments with convolutional neural network. Information Sciences 405:81–90, 2017.
Tan, J. H. et al., Application of stacked convolutional and long short-term memory network for accurate identification of CAD ECG signals. Comput Biol Med 94:19–26, 2018.
Andersen, R. S., Peimankar, A., and Puthusserypady, S., A deep learning approach for real-time detection of atrial fibrillation. Expert Systems with Applications 115:465–473, 2019.
Limam, M. and F. Precioso. Atrial fibrillation detection and ECG classification based on convolutional recurrent neural network. in Computing in Cardiology, (2017).
ANSI/AAMI, Testing and reporting performance results of cardiac rhythm and ST segment measurement algorithms, Association for the Advancement of Medical Instrumentation (AAMI), 2008, American National Standards Institute, Inc. (ANSI), 2008 ANSI/AAMI/ISO EC57, 1998-(R).
Goldberger, A. L., Amaral, L. A., Glass, L., Hausdorff, J. M., Ivanov, P. C., Mark, R. G., Mietus, J. E., Moody, G. B., Peng, C. K., and Stanley, H. E., PhysioBank, PhysioToolkit, and PhysioNet: Components of a new research resource for complex physiologic signals. Circulation 101(23):E215–E220, 2000.
Pan, J., and Tompkins, W. J., A real-time QRS detection algorithm. IEEE transactions on biomedical engineering 3:230–236, 1985.
Huang, H. F., Hu, G. S., and Zhu, L., Sparse representation-based heartbeat classification using independent component analysis. Journal of Medical Systems 36(3):1235–1247, 2012.
Elhaj, F. A., Salim, N., Harris, A. R., Swee, T. T., and Ahmed, T., Arrhythmia recognition and classification using combined linear and nonlinear features of ECG signals. Computer Methods and Programs in Biomedicine 127:52–63, 2016.
Chen, S. et al., Heartbeat classification using projected and dynamic features of ECG signal. Biomedical Signal Processing and Control 31:165–173, 2017.
Acknowledgments
Our research is supported by the National Key R&D Program of China (No. 2018YFB1307005), and the major project from Shanghai Municipal Commission of Health and Family Planning (No. 2018ZHYL0226).
Author information
Authors and Affiliations
Corresponding author
Ethics declarations
Conflict of Interest
All authors declare that there is no conflict of interest in this work.
Ethical Approval
This article does not contain any studies with human participants or animals performed by any of the authors.
Informed Consent
As the article does not involve Human participants, thus, there is no such informed consent.
Additional information
Publisher’s Note
Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.
This article is part of the Topical Collection on Mobile & Wireless Health
Rights and permissions
About this article
Cite this article
Wang, H., Shi, H., Chen, X. et al. An Improved Convolutional Neural Network Based Approach for Automated Heartbeat Classification. J Med Syst 44, 35 (2020). https://doi.org/10.1007/s10916-019-1511-2
Received:
Accepted:
Published:
DOI: https://doi.org/10.1007/s10916-019-1511-2