An Auxiliary Technique for Coronary Heart Disease Prediction by Analyzing ECG Based on ResNet and Bi-LSTM

Authors: Yang Zhang, Jian He

Abstract:

Heart disease is one of the leading causes of death in the world, and coronary heart disease (CHD) is one of the major heart diseases. Electrocardiogram (ECG) is widely used in the detection of heart diseases, but the traditional manual method for CHD prediction by analyzing ECG requires lots of professional knowledge for doctors. This paper presents sliding window and continuous wavelet transform (CWT) to transform ECG signals into images, and then ResNet and Bi-LSTM are introduced to build the ECG feature extraction network (namely ECGNet). At last, an auxiliary system for CHD prediction was developed based on modified ResNet18 and Bi-LSTM, and the public ECG dataset of CHD from MIMIC-3 was used to train and test the system. The experimental results show that the accuracy of the method is 83%, and the F1-score is 83%. Compared with the available methods for CHD prediction based on ECG, such as kNN, decision tree, VGGNet, etc., this method not only improves the prediction accuracy but also could avoid the degradation phenomenon of the deep learning network.

Keywords: Bi-LSTM, CHD, coronary heart disease, ECG, electrocardiogram, ResNet, sliding window.

Digital Object Identifier (DOI): doi.org/10.5281/zenodo.10974322

References:

[1] Benjamin, E. J. et al. (2019). Heart Disease and Stroke Statistics—2019 Update: A Report from the American Heart Association. American Heart Association, 139, 56–528.
[2] Kannel, W. B., Castelli, W. P., Gordon, T. & McNamara, P. M. (1971). Serum cholesterol, lipoproteins, and the risk of coronary heart disease. The Framingham study. Ann Intern Med, 74(1), 1-12.
[3] Irie, F., Iso, H., Sairenchi, T., Fukasawa, N., Yamagishi, K., Ikehara, S., Kanashiki, M. (2006). The relationships of proteinuria, serum creatinine, glomerular filtration rate with cardiovascular disease mortality in Japanese general population. Kidney Int., 69(7), 1264-71.
[4] Burchfiel, C. M., Tracy, R. E., Chyou, P. & Strong, J. P. (1997). Cardiovascular Risk Factors and Hyalinization of Renal Arterioles at Autopsy. Arteriosclerosis, Thrombosis, and Vascular Biology, 17(4), 760–768.
[5] Madani, A., Arnaout, R., Mofrad, M., Arnaout, R. (2018) Fast and accurate view classification of echocardiograms using deep learning. npj Digital Medicine, 1, 6.
[6] Shi Z X, Gu W L. Exploration of TCM syndrome differentiation of coronary heart disease and coronary arteriography (J). Chinese Journal of Integrated Traditional & Western Medicine, 2007, 27(1):76.
[7] Yan Z, Jiang S, Jiao N, et al. The clinical diagnosis effect analysis of electrocardiogram (ECG) and ultrasonic cardiogram (UCG) for coronary atherosclerotic heart disease (CHD) (J). China Modern Doctor, 2015.
[8] Jin, Z., Sun, Y., Cheng, A. C. (2009) Predicting cardiovascular disease from real-time electrocardiographic monitoring: An adaptive machine learning approach on a cell phone. Conf Proc IEEE Eng Med Biol Soc., 6889-92.
[9] Wang Z, Ning X, Du G, et al. Nonlinear Dynamical Characteristics of ECG Signals of CHD Patients (J). Journal of Naijing University (Natural Sciences), 2001.
[10] Cross D S, Mccarty C A, Hytopoulos E, et al. Coronary risk assessment among intermediate risk patients using a clinical and biomarker based algorithm developed and validated in two population cohorts (J). Current Medical Research & Opinion, 2012, 28(11):1819.
[11] Meghan, E, Olesnevich, et al. Serum ferritin levels associated with increased risk for developing CHD in a low-income urban population. (J). Public Health Nutrition, 2012.
[12] Maryam, Tayefi, Mohammad, et al. hs-CRP is strongly associated with coronary heart disease (CHD): A data mining approach using decision tree algorithm (J). Computer Methods & Programs in Biomedicine, 2017.
[13] Karaolis M A, Moutiris J A, Hadjipanayi D, et al. Assessment of the Risk Factors of Coronary Heart Events Based on Data Mining With Decision Trees(J). IEEE Trans Inf Technol Biomed, 2010, 14(3):559-566.
[14] Rajeswari K, Vaithiyanathan D V, Amirtharaj D P. A Novel Risk Level Classification of Ischemic Heart Disease using Artificial Neural Network Technique - An Indian Case Study(J). 2011.
[15] Han X, Liang G. Echocardiographic Features of Patients with Coronary Heart Disease and Angina Pectoris under Deep Learning Algorithms(J). Hindawi Limited, 2021.
[16] Li Yong, He Zihang, Wang Heng, Li Bohan, Li Fengnan, Gao Ying, Ye Xiang. Craftnet: a deep learning ensemble to diagnose cardiovascular diseases. Biomed Signal Process Control 2020;62:102091.
[17] A. E. Awodeyi, S. R. Alty, and M. Ghavami, “Median based method for baseline wander removal in photoplethysmogram signals,” in 2014 IEEE International Conference on Bioinformatics and Bioengineering. IEEE, 2014, pp. 311–314.
[18] Y.-H. Byeon, S.-B. Pan, and K.-C. Kwak, “Intelligent deep models based on scalograms of electrocardiogram signals for biometrics,” Sensors, vol. 19, no. 4, p. 935, 2019.
[19] Wang, Changhong, Low-Power Fall Detector Using Triaxial Accelerometry and Barometric Pressure Sensing (C). IEEE Transactions on Industrial Informatics (2016):1-1. DOI:10.1109/TII.2016.2587761.
[20] Gjoreski, Hristijan, RAReFall — Real-time activity recognition and fall detection system (C). IEEE International Conference on Pervasive Computing & Communications Workshops IEEE, 2014. 7395664, pp.139-145 DOI:10.1109/PerComW.2014.6815182.
[21] Koniusz P, Cherian A, Porikli F. Tensor Representations via Kernel Linearization for Action Recognition from 3D Skeletons (Extended Version) (C). 14th European Conference, ECCV 2016. Vol.9908, pp.37-53 DOI:10.1007/978-3-319-46493-0_3.
[22] He K, Zhang X, Ren S, et al. Deep Residual Learning for Image Recognition(C)// 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR). IEEE, 2016.
[23] A. E. Johnson, T. J. Pollard, L. Shen, L.-W. H Lehman, M. Feng, M. Ghassemi et al., “Mimic-iii, a freely accessible critical care database,” Scientifific data, vol. 3, no. 1, pp. 1–9, 2016.
[24] B. Moody, G. Moody, M. Villarroel, G. Clifford, I. Silva, “Mimic-iii waveform database matched subset (version 1.0),” 2020. (Online). Available: https://physionet.org/content/mimic3wdb-matched/1.0/
[25] Simonyan K, Zisserman A.Very Deep Convolutional Networks for Large-Scale Image Recognition(J). Computer ence, 2014.