Frontiers in Cardiovascular Medicine (Sep 2022)

Using a machine learning model to predict the development of acute kidney injury in patients with heart failure

  • Wen Tao Liu,
  • Xiao Qi Liu,
  • Ting Ting Jiang,
  • Meng Ying Wang,
  • Yang Huang,
  • Yu Lin Huang,
  • Feng Yong Jin,
  • Qing Zhao,
  • Qin Yi Wu,
  • Bi Cheng Liu,
  • Xiong Zhong Ruan,
  • Kun Ling Ma

DOI
https://doi.org/10.3389/fcvm.2022.911987
Journal volume & issue
Vol. 9

Abstract

Read online

BackgroundHeart failure (HF) is a life-threatening complication of cardiovascular disease. HF patients are more likely to progress to acute kidney injury (AKI) with a poor prognosis. However, it is difficult for doctors to distinguish which patients will develop AKI accurately. This study aimed to construct a machine learning (ML) model to predict AKI occurrence in HF patients.Materials and methodsThe data of HF patients from the Medical Information Mart for Intensive Care-IV (MIMIC-IV) database was retrospectively analyzed. A ML model was established to predict AKI development using decision tree, random forest (RF), support vector machine (SVM), K-nearest neighbor (KNN), and logistic regression (LR) algorithms. Thirty-nine demographic, clinical, and treatment features were used for model establishment. Accuracy, sensitivity, specificity, and the area under the receiver operating characteristic curve (AUROC) were used to evaluate the performance of the ML algorithms.ResultsA total of 2,678 HF patients were engaged in this study, of whom 919 developed AKI. Among 5 ML algorithms, the RF algorithm exhibited the highest performance with the AUROC of 0.96. In addition, the Gini index showed that the sequential organ function assessment (SOFA) score, partial pressure of oxygen (PaO2), and estimated glomerular filtration rate (eGFR) were highly relevant to AKI development. Finally, to facilitate clinical application, a simple model was constructed using the 10 features screened by the Gini index. The RF algorithm also exhibited the highest performance with the AUROC of 0.95.ConclusionUsing the ML model could accurately predict the development of AKI in HF patients.

Keywords