Machine learning-based prognostic modeling of patients with acute heart failure receiving furosemide in intensive care units

Abstract

Purpose This study developed machine learning models to predict in-hospital mortality, initiation of acute renal replacement therapy, and mechanical ventilation in patients with acute heart failure receiving furosemide in intensive care units. Method An extensive database comprising static and dynamic features obtained from a Japanese hospital chain was used to construct and train the machine learning models. Results The results revealed that the proposed machine learning models predict in-hospital mortality, initiation of acute renal replacement therapy, and mechanical ventilation with good accuracy. However, the optimal models vary depending on the predicted outcomes. The linear support vector machine classification models exhibited the highest in-hospital mortality and mechanical ventilation prediction accuracy, with the area under the receiver operating characteristic curve of 0.73 and 0.73, respectively, whereas the multi-layer neural network exhibited the highest accuracy for acute renal replacement therapy initiation prediction with an area under the receiver operating characteristic curve of 0.70. Conclusions In conclusion, this study demonstrated that machine learning models could help predict the clinical outcomes of patients with acute heart failure receiving furosemide. However, the optimal models may differ depending on the outcome of interest.