Abstract:Background Machine learning (ML) has been used to build high performance prediction model. Patients with congestive heart failure (CHF) are vulnerable to acute kidney injury (AKI) which makes treatment difficult. We aimed to establish an ML-based prediction model for the early identification of AKI in patients with CHF. Methods Patients data were extracted from the Medical Information Mart for Intensive Care III (MIMIC-III) database, and patients with CHF were selected. Comparisons between several common ML classifiers were conducted to select the best prediction model. Recursive feature elimination (RFE) was used to select important prediction features. The model was improved using hyperparameters optimization (HPO). The final model was validated using an external validation set from the eICU Collaborative Research Database. The area under the receiver operating characteristic curve (AUROC), accuracy, calibration curve and decision curve analysis were used to evaluate prediction performance. Additionally, the final model was used to predict renal replacement therapy (RRT) requirement and to assess the short-term prognosis of patients with CHF. Finally, a software program was developed based on the selected features, which could intuitively report the probability of AKI. Results A total of 8,580 patients with CHF were included, among whom 2,364 were diagnosed with AKI. The LightGBM model showed the best prediction performance (AUROC = 0.803) among the 13 ML-based models. After RFE and HPO, the final model was established with 18 features including serum creatinine (SCr), blood urea nitrogen (BUN) and urine output (UO). The prediction performance of LightGBM was better than that of measuring SCr, UO or SCr combined with UO (AUROCs: 0.809, 0.703, 0.560 and 0.714, respectively). Additionally, the final model could accurately predict RRT requirement in patients with (AUROC = 0.954). Moreover, the participants were divided into high- and low-risk groups for AKI, and the 90-day mortality in the high-risk group was significantly higher than that in the low-risk group (log-rank p < 0.001). Finally, external validation using the eICU database comprising 9,749 patients with CHF revealed satisfactory prediction outcomes (AUROC = 0.816). Conclusion A prediction model for AKI in patients with CHF was established based on LightGBM, and the prediction performance of this model was better than that of other models. This model may help in predicting RRT requirement and in identifying the population with poor prognosis among patients with CHF.

Construction and evaluation of a mortality prediction model for patients with acute kidney injury undergoing continuous renal replacement therapy based on machine learning algorithms

Development of an Acute Kidney Injury Risk Prediction Model for Patients Undergoing Extracorporeal Membrane Oxygenation

Using machine learning to predict the risk of short-term and long-term death in acute kidney injury patients after commencing CRRT

Risk prediction models for successful discontinuation in acute kidney injury undergoing continuous renal replacement therapy

Development and validation of prediction model using nursing notes on sentiment scores for prognosis of patients with severe acute kidney injury receiving continuous renal replacement therapy based on computational intelligence algorithms

Machine learning algorithm to predict the in-hospital mortality in critically ill patients with chronic kidney disease

Construction and Validation of Prognostic Models in Critically Ill Patients with Sepsis-associated Acute Kidney Injury: machine learning approaches compare with traditional logistic regression.

Prediction of Mortality Risk after Ischemic Acute Kidney Injury with a Novel Prognostic Model: A Multivariable Prediction Model Development and Validation Study

Machine learning-based prediction of in-hospital mortality for critically ill patients with sepsis-associated acute kidney injury

Machine learning algorithm to predict mortality in critically ill patients with sepsis-associated acute kidney injury

Machine Learning for the Prediction of First-Year Mortality in Incident Hemodialysis Patients

Identification of risk factors and establishment of prediction models for mortality risk in patients with acute kidney injury: A retrospective cohort study

Explainable Machine Learning-Based Risk Prediction Model for In-Hospital Mortality after Continuous Renal Replacement Therapy Initiation

Development and validation of outcome prediction models for acute kidney injury patients undergoing continuous renal replacement therapy

Prognostic Machine Learning Models for First-Year Mortality in Incident Hemodialysis Patients: Development and Validation Study.

Machine Learning for the Prediction of Volume Responsiveness in Patients with Oliguric Acute Kidney Injury in Critical Care

Predicting in-hospital outcomes of patients with acute kidney injury

Machine learning for the prediction of 1-year mortality in patients with sepsis-associated acute kidney injury

The Prediction of In-Hospital Mortality in Chronic Kidney Disease Patients with Coronary Artery Disease Using Machine Learning Models

Machine learning algorithm for predict the in-hospital mortality in critically ill patients with congestive heart failure combined with chronic kidney disease

A Machine Learning-Based Prediction Model for Acute Kidney Injury in Patients With Congestive Heart Failure