Document Type

Article

Publication Date

2024

DOI

10.35882/jeeemi.v6i2.359

Publication Title

Journal of Electronics, Electromedical Engineering, and Medical Informatics

Volume

6

Issue

2

Pages

92-106

Abstract

This research examines the efficacy of ensemble Machine Learning (ML) models, mainly focusing on Deep Neural Networks (DNNs), in predicting the need for cardiovascular surgery, a critical aspect of clinical decision-making. It addresses key challenges such as class imbalance, which is pivotal in healthcare settings. The research involved a comprehensive comparison and evaluation of the performance of previously published ML methods against a new Deep Learning (DL) model. This comparison utilized a dataset encompassing 50,000 patient records from a large hospital between 2015-2022. The study proposes enhancing the efficacy of these models through feature selection and hyperparameter optimization, employing techniques like grid search. A novel aspect of this research was the comparison of a newly developed DNN model with existing ensemble models based on similar cardiovascular datasets. The results indicated the DNN model's superior predictive accuracy, demonstrating an Area Under the Curve (AUC) of 74%, alongside notable precision (68%) and recall (72%) for the minority class, which indicates patients requiring surgery. The model further achieved a 70% F1-Score and a balanced accuracy rate of 72%, significantly outperforming the existing ensemble models in every key performance metric. The study underscores the transformative potential of DNNs in predictive modeling for cardiovascular care and highlights the importance of integrating advanced ML techniques into clinical workflows. Future research should delve into the practical application and integration of these models.

Rights

© 2024 Arman Ghavidel, Pilar Pazos, Rolando Del Aguila Suarez, Alireza Atashi.

This work is licensed under a Creative Commons Attribution-ShareAlike 4.0 International (CC BY-SA 4.0) License.

Data Availability

Article states: "Due to the sensitive nature of the medical date used in this study and in strict compliance with patient confidentiality and privacy regulations, the dataset cannot be publicly shared."

ORCID

0009-0003-9798-8958 (Ghavidel), 0000-0003-4348-7798 (Pazos)

Original Publication Citation

Ghavidel, A., Pazos, P., Suarez, R. D. A., & Atashi, A. (2024). Predicting the need for cardiovascular surgery: A comparative study of machine learning models. Journal of Electronics, Electromedical Engineering, and Medical Informatics, 6(2), 92-106. https://doi.org/10.35882/jeeemi.v6i2.359

Share

COinS