Development and validation of an interpretable machine learning-based model for predicting carbapenem-resistant

Yan Gao1, Guangxin Gu2,3, Ruiwen Wang2,3

  • 1Department of Disease Prevention and Control, General Hospital of Northern Theater Command, Shenyang, China.

Abstract

Insights

Machine learning models can now predict carbapenem-resistant Acinetobacter baumannii (CRAB) infections in postoperative ICU patients. Key predictors include ventilation duration and central line use, enabling targeted prevention and antimicrobial stewardship.

Area of Science:

  • Infectious Diseases
  • Critical Care Medicine
  • Machine Learning in Healthcare

Background:

  • Carbapenem-resistant Acinetobacter baumannii (CRAB) poses a significant threat in intensive care units (ICUs), especially for postoperative patients.
  • High rates of CRAB infections are linked to prolonged ICU stays, invasive procedures, and increased antimicrobial use.
  • Currently, limited tools exist for early identification of high-risk postoperative ICU patients susceptible to CRAB.

Purpose of the Study:

  • To develop and validate machine learning models for early prediction of CRAB infection risk in postoperative ICU patients.
  • To identify key clinical and laboratory predictors of CRAB infection in this vulnerable population.
  • To create an interpretable and clinically applicable risk assessment framework.

Main Methods:

  • A retrospective cohort study of 2,195 postoperative ICU patients was conducted.
  • Eight machine learning models were developed using demographic, treatment, and laboratory data.
  • Boruta algorithm was used for feature selection, and SHAP analysis for model interpretability.

Main Results:

  • CRAB infection developed in 31.6% of patients, associated with worse outcomes.
  • All eight models demonstrated good predictive performance (AUC > 0.83) in an independent test set.
  • Gradient Boosting achieved the highest AUC (0.867), with duration of mechanical ventilation, central venous catheterization, ICU length of stay, and carbapenem exposure identified as key predictors by SHAP analysis.

Conclusions:

  • An interpretable machine learning framework for CRAB infection risk assessment in postoperative ICU patients was developed.
  • The findings support targeted prevention strategies and optimized antimicrobial stewardship.
  • Feature reduction techniques demonstrated the feasibility of clinical application.

Related Concept Videos