Heart disease, globally recognized as a leading cause of death, has seen its impact magnified by the emergence of COVID-19. The heightened demand for early detection and diagnosis of heart disease has forced the development of innovative, intelligent systems. This research offers a novel approach by leveraging extended short-term memory networks (LSTM) and including COVID-19 as a significant parameter in cardiac arrest analysis. A comparative study is conducted between LSTM and other prevalent techniques, such as support vector machines (SVM), linear regression (LR), and artificial neural networks (ANN), focusing on accuracy and other prognostic criteria for heart disease. We aim to develop an intelligent system powered by LSTM to predict heart disease, thereby assisting healthcare professionals in making well-informed decisions about heart disease management, stroke prevention, and patient monitoring. Additionally, hyperparameter tuning has been performed to optimize the LSTM model's performance in cardiac arrest prediction. The results underscore that LSTM, especially when trained with COVID-19 as an input parameter, surpasses other established techniques in prediction accuracy. The proposed model underwent experimental testing, showcasing its proficiency in predicting cardiovascular disease.
Citation: Kuna Dhananjay Rao, Mudunuru Satya Dev Kumar, Paidi Pavani, Darapureddy Akshitha, Kagitha Nagamaleswara Rao, Hafiz Tayyab Rauf, Mohamed Sharaf. Cardiovascular disease prediction using hyperparameters-tuned LSTM considering COVID-19 with experimental validation[J]. AIMS Bioengineering, 2023, 10(3): 265-282. doi: 10.3934/bioeng.2023017
Heart disease, globally recognized as a leading cause of death, has seen its impact magnified by the emergence of COVID-19. The heightened demand for early detection and diagnosis of heart disease has forced the development of innovative, intelligent systems. This research offers a novel approach by leveraging extended short-term memory networks (LSTM) and including COVID-19 as a significant parameter in cardiac arrest analysis. A comparative study is conducted between LSTM and other prevalent techniques, such as support vector machines (SVM), linear regression (LR), and artificial neural networks (ANN), focusing on accuracy and other prognostic criteria for heart disease. We aim to develop an intelligent system powered by LSTM to predict heart disease, thereby assisting healthcare professionals in making well-informed decisions about heart disease management, stroke prevention, and patient monitoring. Additionally, hyperparameter tuning has been performed to optimize the LSTM model's performance in cardiac arrest prediction. The results underscore that LSTM, especially when trained with COVID-19 as an input parameter, surpasses other established techniques in prediction accuracy. The proposed model underwent experimental testing, showcasing its proficiency in predicting cardiovascular disease.
[1] | Patel J, TejalUpadhyay D, Patel S (2015) Heart disease prediction using machine learning and data mining technique. Heart Dis 7: 129-137. |
[2] | Zang J, You P (2023) An industrial IoT-enabled smart healthcare system using big data mining and machine learning. Wirel Netw 29: 909-918. https://doi.org/10.1007/s11276-022-03129-z |
[3] | Ebrahimi Z, Loni M, Daneshtalab M, et al. (2020) A review on deep learning methods for ECG arrhythmia classification. Expert Syst Appl X 7: 100033. https://doi.org/10.1016/j.eswax.2020.100033 |
[4] | Fujiwara K, Abe E, Kamata K, et al. (2018) Heart rate variability-based driver drowsiness detection and its validation with EEG. IEEE T Biomed Eng 66: 1769-1778. https://doi.org/10.1109/TBME.2018.2879346 |
[5] | Pandey H, Prabha S (2020) Smart health monitoring system using IOT and machine learning techniques. 2020 sixth international conference on bio signals, images, and instrumentation (ICBSII) 2020: 1-4. https://doi.org/10.1109/ICBSII49132.2020.9167660 |
[6] | Ali NS, Alyasseri ZAA, Abdulmohson A (2018) Real-time heart pulse monitoring technique using wireless sensor network and mobile application. Int J Electr Comput Eng 8: 5118. https://doi.org/10.11591/ijece.v8i6.pp5118-5126 |
[7] | Guo D, Zhang G, Peng H, et al. (2021) Segmentation and measurements of carotid intima-media thickness in ultrasound images using the improved convolutional neural network and support vector machine. J Med Imag Health In 11: 15-24. https://doi.org/10.1166/jmihi.2021.3264 |
[8] | Filist S, Al-Kasasbeh RT, Shatalova O, et al. (2022) Developing neural network model for predicting cardiac and cardiovascular health using bioelectrical signal processing. Comput method Biomec Biomed Eng 25: 908-921. https://doi.org/10.1080/10255842.2021.1986486 |
[9] | Attia ZI, Kapa S, Lopez-Jimenez F, et al. (2019) Screening for cardiac contractile dysfunction using an artificial intelligence–enabled electrocardiogram. Nat Med 25: 70-74. https://doi.org/10.1038/s41591-018-0240-2 |
[10] | Rana JS, Tabada GH, Solomon MD, et al. (2016) Accuracy of the atherosclerotic cardiovascular risk equation in a large contemporary, multiethnic population. J Am Coll Cardiol 67: 2118-2130. https://doi.org/10.1016/j.jacc.2016.02.055 |
[11] | Stehli J, Fuchs TA, Bull S, et al. (2014) Accuracy of coronary CT angiography using a submillisievert fraction of radiation exposure: comparison with invasive coronary angiography. J Am Coll Cardiol 64: 772-780. https://doi.org/10.1016/j.jacc.2014.04.079 |
[12] | O'Donoghue M L, Braunwald E, White H D, et al. (2014) Effect of darapladib on major coronary events after an acute coronary syndrome: the SOLID-TIMI 52 randomized clinical trial. Jama 312: 1006-1015. https://doi.org/10.1001/jama.2014.11061 |
[13] | Saeb S, Lonini L, Jayaraman A, et al. (2017) The need to approximate the use-case in clinical machine learning. Gigascience 6: gix019. https://doi.org/10.1093/gigascience/gix019 |
[14] | Cherkassky V (1997) The nature of statistical learning theory~. IEEE T Neural Network 8: 1564-1564. https://doi.org/10.1109/TNN.1997.641482 |
[15] | Tang H, Gao S, Wang L, et al. (2021) A novel intelligent fault diagnosis method for rolling bearings based on wasserstein generative adversarial network and convolutional neural network under unbalanced dataset. Sensors 21: 6754. https://doi.org/10.3390/s21206754 |
[16] | Wang YY, Wang ST, Sima XT, et al. (2023) Expanded feature space-based gradient boosting ensemble learning for risk prediction of type 2 diabetes complications. Appl Soft Comput 144: 110451. https://doi.org/10.1016/j.asoc.2023.110451 |
[17] | Kee OT, Harun H, Mustafa N, et al. (2023) Cardiovascular complications in a diabetes prediction model using machine learning: a systematic review. Cardiovasc Diabetol 22: 13. https://doi.org/10.1186/s12933-023-01741-7 |
[18] | Wang S, Yin Y, Wang D, et al. (2021) An interpretable deep neural network for colorectal polyp diagnosis under colonoscopy. Knowl-based Syst 234: 107568. https://doi.org/10.1016/j.knosys.2021.107568 |
[19] | Tobler DL, Pruzansky AJ, Naderi S, et al. (2022) Long-term cardiovascular effects of COVID-19: emerging data relevant to the cardiovascular clinician. Curr Atheroscler Rep 24: 563-570. https://doi.org/10.1007/s11883-022-01032-8 |
[20] | Xie Y, Xu E, Bowe B, et al. (2022) Long-term cardiovascular outcomes of COVID-19. Nat Med 28: 583-590. https://doi.org/10.1038/s41591-022-01689-3 |
[21] | Dale CE, Takhar R, Carragher R, et al. (2023) The impact of the COVID-19 pandemic on cardiovascular disease prevention and management. Nat Med 29: 219-225. https://doi.org/10.1038/s41591-022-02158-7 |
[22] | Maurya JP, Manoria M, Joshi S (2022) Cardiac Arrhythmia Classification Using Cascaded Deep Learning Approach (LSTM & RNN). International Conference on Machine Learning, Image Processing, Network Security and Data Sciences. Cham: Springer Nature Switzerland 3-13. https://doi.org/10.1007/978-3-031-24352-3_1 |
[23] | Pandya S, Gadekallu TR, Reddy PK, et al. (2022) InfusedHeart: A novel knowledge-infused learning framework for diagnosis of cardiovascular events. IEEE Trans Comput Soc Sy 2022: 1-10. https://doi.org/10.1109/TCSS.2022.3151643 |
[24] | Greff K, Srivastava RK, Koutník J, et al. (2016) LSTM: a search space odyssey. IEEE T Neur Net Lear 28: 2222-2232. https://doi.org/10.1109/TNNLS.2016.2582924 |
[25] | Bergstra J, Bengio Y (2012) Random search for hyper-parameter optimization. J Mach Learn Res 13: 281-305. |
[26] | Li L, Jamieson K, DeSalvo G, et al. (2017) Hyperband: A novel bandit-based approach to hyperparameter optimization. J Mach Learn Res 18: 6765-6816. |
[27] | Rao KD, Kumar MSD, Akshitha D, et al. (2022) Machine Learning Based Cardiovascular Disease Prediction. 2022 International Conference on Computer, Power and Communications (ICCPC). India: Chennai: 118-122. |