Early Detection of Congenital Heart Diseases among Infants Using Artificial Neural Network Algorithm
Article Sidebar
Main Article Content
Abstract
Congenital Heart Disease (CHD) detection has continued to witness a consistent increase in research attention. CHD diseases are vast and also span across diverse demographics, without sparing pregnant women, unborn babies, or newly born babies. The aim of this study is to develop a detection model capable of detecting heart disease among infants with high accuracy and to also suggest solutions to manage it. To achieve this, the CHD types were identified to develop a data model that considered infants. The data model was processed through imputation, feature selection, and transformation using the imputation method and Principal Component Analysis (PCA). After the data processing stage, Artificial Neural Network (ANN) algorithm is adopted and trained with the data to generate the model for the detection of the CHD. Comparative analysis was used to evaluate the performance of the models in comparison with other models adopted in other works, considering metrics that defined the success of the detection models. The results showed that the ANN has the best detection outcome with an accuracy of 97.44%. Although the use of Logistic Regression algorithm attained a high level of performance with an accuracy of 95.00%, but it still falls below the proposed ANN algorithm. Another highly performing algorithm is the use of Extreme Gradient Boosting (XGB) which achieved an accuracy of 92.00%.
Downloads
Article Details
This work is licensed under a Creative Commons Attribution-NonCommercial-ShareAlike 4.0 International License.
References
Zhu, X., Wang, X., Zhang, S., & Zhang, Y. (2019). Heart disease prediction using machine learning techniques. Journal Informatics, 98, 103-115. https://doi.org/10.1016/j.jbi.2019.103115
Yang, X., Li, Y., Li, Y., Ren, X., Zhang, X., Hu, D., & Sun, Y. (2017). Predictive models for the risk of coronary artery disease among Chinese people. Scientific Reports, 7(1), 1-9. https://doi.org/10.1038/s41598-017-15138-3
Wilson, P. W., D’Agostino, R. B., Levy, D., Belanger, A. M., Silbershatz, H., & Kannel, W. B. (1998). Prediction of coronary heart disease using risk factor categories. Circulation, 97(18), 1837-1847. https://doi.org/10.1161/01.CIR.97.18.1837
World Health Organization (WHO). (2023). A woman dies every two minutes due to pregnancy or childbirth. Geneva/New York/Washington. Retrieved from https://www.who.int/news/item/23-02-2023-a-woman-dies-every-two-minutes-due-to-pregnancy-or-childbirth--un-agencies
Singh, N., & Jindal, S. (2018). Heart disease prediction system using hybrid technique of data mining algorithms. International Journal of Advance Research, Ideas and Innovations in Technology, 4(2), 15-18. https://www.ijarai.thesai.org
Adil, H. S., & Chaurasia, P. K. (2019). A review on heart disease prediction using machine learning techniques. International Journal of Management, IT & Engineering, 9(4), 208-224.
Juma, P. A., Mohamed, S. F., Matanje-Mwagomba, B. L., et al. (2018). Non-communicable disease prevention policy process in five African countries. BMC Public Health, 18(Suppl 1), 961. https://doi.org/10.1186/s12889-018-5830-6
Shirwalkar, N., & Tak, T. (2018). Human heart disease prediction system using data mining techniques. International Journal of Innovations & Advancement in Computer Science, 7(3), 101-108.
Nnenna, H. N., Odo, H., Ozoemena, P., & Ebere Uzoka Chidi. (2022). Predictive model for the monitoring and detection of heart disease using wavelet-based machine learning technique. American Journal of Applied Sciences and Engineering, 3(5), 1-12. https://doi.org/10.21203/rs.3.rs-97004/v1
Arabasadi, Z., Alizadehsani, R., Roshanzamir, M., Moosaei, H., & Yarifard, A. A. (2017). Computer aided decision making for heart disease detection using hybrid neural network-genetic algorithm. Computer Methods and Programs in Biomedicine, 141, 19-26. https://doi.org/10.1016/j.cmpb.2017.01.007
Kohil, P. (2023). Everything you need to know about heart disease. Available at: https://www.medicalnewstoday.com/articles/237191. Retrieved on September 28, 2023.
Adeniyi, K. M., Oladapo, O. J., Araoye, T. O., Adebayo, T. F., Egoigwe, S. V., & Odo, M. C. (2023). Prediction of heart disease outcomes using machine learning classifier. Indonesian Journal of Electrical Engineering and Computer Science, 30(2), 917-926. https://doi.org/10.11591/ijeecs.v30.i2.33207
Manrique, A., Carrasco, S., & Buil, P. (2019). Machine learning in cardiovascular medicine: Are we there yet? Revista Española de Cardiología (English Edition), 72(1), 81-83. https://doi.org/10.1016/j.rec.2018.07.013
Patel, J., Upadhyay, T., & Patel, S. (2018). Heart disease prediction using machine learning and data mining technique. IJCSC, 7(1), 129-137. https://www.ijcsc.org
Roger, V. L. (2021). Epidemiology of heart failure: A contemporary perspective. Circulation Research, 128(10), 1421-1434. https://doi.org/10.1161/CIRCRESAHA.121.318365
Shah, D., Patel, S., & Bharti, K. S. (2020). Heart disease prediction using machine learning techniques. SN Computer Science, 1, 345. https://doi.org/10.1007/s42979-020-00365-y
Singh, A., & Kumar, R. (2020). Heart disease prediction using machine learning algorithms. In 2020 International Conference on Electrical and Electronics Engineering (ICE3-2020). https://doi.org/10.1109/ice3.2020.9280795
Hossen, M. (2022). Linear regression-based heart disease prediction using machine learning. Journal of Health Informatics and Statistics, 7(1), 55-68. https://doi.org/10.1007/s41899-022-00109-0
Tiwari, A., Chugh, A., & Sharma, A. (2022). Ensemble framework for cardiovascular disease prediction. In 2022 International Conference on Machine Learning and Data Science (MLDS). https://doi.org/10.1109/mlds54165.2022.9882904
Srivenkatesh, M. (2020). Prediction of cardiovascular disease using machine learning algorithms. International Journal of Engineering and Advanced Technology (IJEAT), 9(3), 2249-8958. https://doi.org/10.35940/ijeat.C5864.059320
Chithambaram, T., Kannan, N. L., & Gowsalya, M. (2020). Heart disease detection using machine learning. ResearchSquare. https://doi.org/10.21203/rs.3.rs-97004/v1
Chandrika, L., & Madhavi, K. (2020). A hybrid framework for heart disease prediction using machine learning algorithms. E3S Web of Conferences, 309, 01043. https://doi.org/10.1051/e3sconf/202130901043
Ibrahim, S., Salhab, N., & Falou, A. E. (2023). Heart disease prediction using machine learning. In 2023 1st International Conference on Advanced Innovations in Smart Cities (ICAISC). https://doi.org/10.1109/icaisc56366.2023.10085522
Nandal, N., Goel, L., & Tanwar, R. (2022). Machine learning-based heart attack prediction: Asymptomatic heart attack prediction method and exploratory analysis. F1000Research, 11, 1126. https://doi.org/10.12688/f1000research.29083.1
Pandit, B. (2023). Four most popular data normalization techniques every data scientist should know. Available at https://dataaspirant.com/data-normalization-techniques/. Accessed October 23, 2023.
Al-Garad, M. (2020). Machine learning techniques for heart disease prediction. Journal of Data Science and Analytics, 4(2), 99-108. https://doi.org/10.1007/s10115-020-04896-7
Reyna, M., Kiarashi, Y., Elola, A., Oliveira, J., Renna, F., Gu, A., Perez Alday, E. A., Sadr, N., Mattos, S., Coimbra, M., Sameni, R., Bahrami Rad, A., Koscova, Z., & Clifford, G. (2023). Heart murmur detection from phonocardiogram recordings: The George B. Moody PhysioNet Challenge 2022 (version 1.0.0). PhysioNet. https://doi.org/10.13026/t49p-5v35
PhysioNet. (2019). PhysioBank databases. Retrieved from https://physionet.org Accessed in September 2024