Short-term prognostic models for severe acute kidney injury patients receiving prolonged intermittent renal replacement therapy based on machine learning

BACKGROUND: As an effective measurement for severe acute kidney injury (AKI), the prolonged intermittent renal replacement therapy (PIRRT) received attention. Also, machine learning has advanced and been applied to medicine. This study aimed to establish short-term prognosis prediction models for severe AKI patients who received PIRRT by machine learning. METHODS: The hospitalized AKI patients who received PIRRT were assigned to this retrospective case-control study. They were grouped based on survival situation and renal recovery status. To screen the correlation, Pearson’s correlation coefficient, partial ETA square, and chi-square test were applied, eight machine learning models were used for training. RESULTS: Among 493 subjects, the mortality rate was 51.93% and the kidney recovery rate was 30.43% at 30 days post-discharge, respectively. The indices related to survival were Sodium, Total protein, Lactate dehydrogenase (LDH), Phosphorus, Thrombin time, Liver cirrhosis, chronic kidney disease stage, number of vital organ injuries, and AKI stage, while Sodium, Total protein, LDH, Phosphorus, Thrombin time, Diabetes, peripherally inserted central catheter and AKI stage were selected to predict the 30-day renal recovery. Naive Bayes has a good performance in the prediction model for survival, Random Forest has a good performance in 30-day renal recovery prediction model, while for 90-day renal recovery prediction model, it’s K-Nearest Neighbor. CONCLUSIONS: Machine learning can not only screen out indicators influencing prognosis of AKI patients receiving PIRRT, but also establish prediction models to optimize the risk assessment of these people. Moreover, attention should be paid to serum electrolytes to improve prognosis. SUPPLEMENTARY INFORMATION: The online version contains supplementary material available at 10.1186/s12911-023-02231-2.