Cardiovascular diseases are regarded as the most common reason for worldwide deaths. As per World Health Organization, nearly $ 17.9 $ million people die of heart-related diseases each year. The high shares of cardiovascular-related diseases in total worldwide deaths motivated researchers to focus on ways to reduce the numbers. In this regard, several works focused on the development of machine learning techniques/algorithms for early detection, diagnosis, and subsequent treatment of cardiovascular-related diseases. These works focused on a variety of issues such as finding important features to effectively predict the occurrence of heart-related diseases to calculate the survival probability. This research contributes to the body of literature by selecting a standard well defined, and well-curated dataset as well as a set of standard benchmark algorithms to independently verify their performance based on a set of different performance evaluation metrics. From our experimental evaluation, it was observed that decision tree is the best performing algorithm in comparison to logistic regression, support vector machines, and artificial neural networks. Decision trees achieved $ 14 $% better accuracy than the average performance of the remaining techniques. In contrast to other studies, this research observed that artificial neural networks are not as competitive as the decision tree or support vector machine.
Citation: Abdulwahab Ali Almazroi. Survival prediction among heart patients using machine learning techniques[J]. Mathematical Biosciences and Engineering, 2022, 19(1): 134-145. doi: 10.3934/mbe.2022007
Cardiovascular diseases are regarded as the most common reason for worldwide deaths. As per World Health Organization, nearly $ 17.9 $ million people die of heart-related diseases each year. The high shares of cardiovascular-related diseases in total worldwide deaths motivated researchers to focus on ways to reduce the numbers. In this regard, several works focused on the development of machine learning techniques/algorithms for early detection, diagnosis, and subsequent treatment of cardiovascular-related diseases. These works focused on a variety of issues such as finding important features to effectively predict the occurrence of heart-related diseases to calculate the survival probability. This research contributes to the body of literature by selecting a standard well defined, and well-curated dataset as well as a set of standard benchmark algorithms to independently verify their performance based on a set of different performance evaluation metrics. From our experimental evaluation, it was observed that decision tree is the best performing algorithm in comparison to logistic regression, support vector machines, and artificial neural networks. Decision trees achieved $ 14 $% better accuracy than the average performance of the remaining techniques. In contrast to other studies, this research observed that artificial neural networks are not as competitive as the decision tree or support vector machine.
| [1] | Cardiovascular Diseases, 2021. Available from: https://www.who.int/health-topics/cardiovascular-diseases. |
| [2] | D. Chicco, G. Jurman, Machine learning can predict survival of patients with heart failure from serum creatinine and ejection fraction alone, BMC Med. Inf. Decis. Making, 20 (2020), 1–16. doi: 10.1186/s12911-020-1023-5. |
| [3] | P. Ghosh, S. Azam, M. Jonkman, A. Karim, F. J. M. Shamrat, E. Ignatious, et al., Efficient prediction of cardiovascular disease using machine learning algorithms with relief and LASSO feature selection techniques, IEEE Access, 9 (2021), 19304–19326. doi: 10.1109/ACCESS.2021.3053759. |
| [4] | Y. Chen, X. Qin, L. Zhang, B. Yi, A novel method of heart failure prediction based on DPCNN XGBoost model, Comput. Mater. Con., 65 (2020), 495–510. doi: 10.32604/cmc.2020.011278. |
| [5] | I. Ahmad, S. U. Rehman, I. U. Khan, A. Ali, H. Hussain, S. Jan, et al., A hybrid approach for automatic aorta segmentation in abdominal 3D CT scan images, J. Med. Imaging Health Inf., 11 (2021), 712–719. doi: 10.1166/jmihi.2021.3364. |
| [6] | G. Litjens, T. Kooi, B. E. Bejnordi, A. A. A. Setio, F. Ciompi, M. Ghafoorian, et al., A survey on deep learning in medical image analysis, Med. Image Anal., 42 (2017), 60–88. doi: 10.1016/j.media.2017.07.005. |
| [7] | I. Ahmad, M. A. Alqarni, A. A. Almazroi, A. Tariq, Experimental evaluation of clickbait detection using machine learning models, Intell. Autom. Soft Comput., 26 (2020), 1335–1344. doi: 10.32604/iasc.2020.013861. |
| [8] | B. S. Freeman, G. Taylor, B. Gharabaghi, J. Thé, Forecasting air quality time series using deep learning, J. Air Waste Manage. Assoc., 68 (2018), 866–886. doi: 10.1080/10962247.2018.1459956. |
| [9] | M. Munawar, I. Noreen, Duplicate frame video forgery detection using siamese-based RNN, Intell. Autom. Soft Comput., 29 (2021), 927–937. doi: 10.32604/iasc.2021.018854. |
| [10] | I. Ahmad, G. Ahmed, S. A. A. Shah, E. Ahmad, A decade of big data literature: analysis of trends in light of bibliometrics, J. Supercomput., 76 (2020), 3555–3571. doi: 10.1007/s11227-018-2714-x. |
| [11] | O. B. Sezer, M.U. Gudelek, A. M. Ozbayoglu, Financial time series forecasting with deep learning: a systematic literature review: 2005–2019, Appl. Soft Comput., 90 (2020), 106181. doi: 10.1016/j.asoc.2020.106181. |
| [12] | I. Ahmad, M. Yousaf, S. Yousaf, M. O. Ahmad, Fake news detection using machine learning ensemble methods, Complexity, 20 (2020). doi: 10.1155/2020/8885861. |
| [13] | T. Ahmad, A. Munir, S. H. Bhatti, M. Aftab, M. A. Raza, Survival analysis of heart failure patients: a case study, PloS One, 12 (2017), e0181001. doi: 10.1371/journal.pone.0181001. |
| [14] | Y. Khourdifi, M. Bahaj, Heart disease prediction and classification using machine learning algorithms optimized by particle swarm optimization and ant colony optimization, Int. J. Intell. Eng. Syst., 12 (2019), 242–252. doi: 10.22266/ijies2019.0228.24. |
| [15] | D. Shah, S. Patel, S. K. Bharti, Heart disease prediction using machine learning techniques, SN Comput. Sci., 1 (2020), 1–6. doi: 10.1007/s42979-020-00365-y. |
| [16] | M. Diwakar, A. Tripathi, K. Joshi, M. Memoria, P. Singh, N. Kumar, Latest trends on heart disease prediction using machine learning and image fusion, Mater. Today Proc., 37 (2021), 3213–3218. doi: 10.1016/j.matpr.2020.09.078. |
| [17] | S. Mohan, C. Thirumalai, G. Srivastava, Effective heart disease prediction using hybrid machine learning techniques, IEEE Access, 7 (2019), 81542–81554. doi: 10.1109/ACCESS.2019.2923707. |
| [18] | M. Porum, E. Iadanza, S. Massaro, L. Pecchia, A convolutional neural network approach to detect congestive heart failure, Biomed. Signal Process. Control, 55 (2020), 101597. doi: 10.1016/j.bspc.2019.101597. |
| [19] | C. B. Monti, M. Codari, M. van Assen, C. N. De Cecco, R. Vliegenthart, Machine learning and deep neural networks applications in computed tomography for coronary artery disease and myocardial perfusion, J. Thorac. Imaging, 35 (2020), S58–S65. doi: 10.1097/RTI.0000000000000490. |
| [20] | Z. Zhang, Y. Qiu, X. Yang, M. Zhang, Enhanced character-level deep convolutional neural networks for cardiovascular disease prediction, BMC Med. Inf. Decis. Making, 20 (2020), 1–10. doi: 10.1186/s12911-020-1118-z. |
| [21] | K. Rahimi, D. Bennett, N. Conrad, T. M. Williams, J. Basu, J. Dwight, et al., Risk prediction in patients with heart failure: a systematic review and analysis, JACC Heart Failure, 2 (2014), 440–446. doi: 10.1016/j.jchf.2014.04.008. |
| [22] | E. Tripoliti, T. Papadopoulos, G. Karanasiou, K. Naka, D. Fotiadis, Heart failure: diagnosis, severity estimation and prediction of adverse events through machine learning techniques, Comput. Struct. Biotechnol. J., 12 (2017), 26–47. doi: 10.1016/j.csbj.2016.11.001. |
Figures(6) / Tables(2)
Abdulwahab Ali Almazroi. Survival prediction among heart patients using machine learning techniques[J]. Mathematical Biosciences and Engineering, 2022, 19(1): 134-145. doi: 10.3934/mbe.2022007
DownLoad: