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

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...

Full description

Saved in:
Bibliographic Details
Main Authors: Ali Bahrami, Morteza Rakhshaninejad, Rouzbeh Ghousi, Alireza Atashi
Format: Article
Language:English
Published: Public Library of Science (PLoS) 2025-01-01
Series:PLoS ONE
Online Access:https://doi.org/10.1371/journal.pone.0311250
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1850144390997082112
author Ali Bahrami
Morteza Rakhshaninejad
Rouzbeh Ghousi
Alireza Atashi
author_facet Ali Bahrami
Morteza Rakhshaninejad
Rouzbeh Ghousi
Alireza Atashi
author_sort Ali Bahrami
collection DOAJ
description 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.
format Article
id doaj-art-dd21df621f104493b27d544c96dedba8
institution OA Journals
issn 1932-6203
language English
publishDate 2025-01-01
publisher Public Library of Science (PLoS)
record_format Article
series PLoS ONE
spelling doaj-art-dd21df621f104493b27d544c96dedba82025-08-20T02:28:23ZengPublic Library of Science (PLoS)PLoS ONE1932-62032025-01-01202e031125010.1371/journal.pone.0311250Enhancing machine learning performance in cardiac surgery ICU: Hyperparameter optimization with metaheuristic algorithm.Ali BahramiMorteza RakhshaninejadRouzbeh GhousiAlireza AtashiThe 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.https://doi.org/10.1371/journal.pone.0311250
spellingShingle Ali Bahrami
Morteza Rakhshaninejad
Rouzbeh Ghousi
Alireza Atashi
Enhancing machine learning performance in cardiac surgery ICU: Hyperparameter optimization with metaheuristic algorithm.
PLoS ONE
title Enhancing machine learning performance in cardiac surgery ICU: Hyperparameter optimization with metaheuristic algorithm.
title_full Enhancing machine learning performance in cardiac surgery ICU: Hyperparameter optimization with metaheuristic algorithm.
title_fullStr Enhancing machine learning performance in cardiac surgery ICU: Hyperparameter optimization with metaheuristic algorithm.
title_full_unstemmed Enhancing machine learning performance in cardiac surgery ICU: Hyperparameter optimization with metaheuristic algorithm.
title_short Enhancing machine learning performance in cardiac surgery ICU: Hyperparameter optimization with metaheuristic algorithm.
title_sort enhancing machine learning performance in cardiac surgery icu hyperparameter optimization with metaheuristic algorithm
url https://doi.org/10.1371/journal.pone.0311250
work_keys_str_mv AT alibahrami enhancingmachinelearningperformanceincardiacsurgeryicuhyperparameteroptimizationwithmetaheuristicalgorithm
AT mortezarakhshaninejad enhancingmachinelearningperformanceincardiacsurgeryicuhyperparameteroptimizationwithmetaheuristicalgorithm
AT rouzbehghousi enhancingmachinelearningperformanceincardiacsurgeryicuhyperparameteroptimizationwithmetaheuristicalgorithm
AT alirezaatashi enhancingmachinelearningperformanceincardiacsurgeryicuhyperparameteroptimizationwithmetaheuristicalgorithm