Enhancing machine learning performance in cardiac surgery ICU: Hyperparameter optimization with metaheuristic algorithm

Ali Bahrami, Morteza Rakhshaninejad, Rouzbeh Ghousi and Alireza Atashi

PLOS ONE, 2025, vol. 20, issue 2, 1-25

Abstract: The healthcare industry is generating a massive volume of data, promising a potential goldmine of information that can be extracted through machine learning (ML) techniques. The Intensive Care Unit (ICU) stands out as a focal point within hospitals and provides a rich source of data for informative analyses. This study examines the cardiac surgery ICU, where the vital topic of patient ventilation takes center stage. In other words, ventilator-supported breathing is a fundamental need within the ICU, and the limited availability of ventilators in hospitals has become a significant issue. A crucial consideration for healthcare professionals in the ICU is prioritizing patients who require ventilators immediately. To address this issue, we developed a prediction model using four ML and deep learning (DL) models—LDA, CatBoost, Artificial Neural Networks (ANN), and XGBoost—that are combined in an ensemble model. We utilized Simulated Annealing (SA) and Genetic Algorithm (GA) to tune the hyperparameters of the ML models constructing the ensemble. The results showed that our approach enhanced the sensitivity of the tuned ensemble model to 85.84%, which are better than the results of the ensemble model without hyperparameter tuning and those achieved using AutoML model. This significant improvement in model performance underscores the effectiveness of our hybrid approach in prioritizing the need for ventilators among ICU patients.

Date: 2025
References: Add references at CitEc
Citations:

Downloads: (external link)
https://journals.plos.org/plosone/article?id=10.1371/journal.pone.0311250 (text/html)
https://journals.plos.org/plosone/article/file?id= ... 11250&type=printable (application/pdf)

Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.

Export reference: BibTeX RIS (EndNote, ProCite, RefMan) HTML/Text

Persistent link: https://EconPapers.repec.org/RePEc:plo:pone00:0311250

DOI: 10.1371/journal.pone.0311250

Access Statistics for this article

More articles in PLOS ONE from Public Library of Science
Bibliographic data for series maintained by plosone ().