Machine Learning-Based Prediction Model for Patients with Recurrent Staphylococcus aureus Bacteremia

Abstract

Abstract Objectives Staphylococcus aureus bacteremia (SAB) remains a significant contributor to both community-acquired and healthcare-associated bloodstream infections. SAB exhibits a high recurrence rate and mortality rate, leading to numerous clinical treatment challenges. Particularly, since the outbreak of COVID-19, there has been a gradual increase in SAB patients, with a growing proportion of (Methicillin-resistant Staphylococcus aureus)MRSA infections. Therefore, we have constructed and validated a pediction model for recurrent Staphylococcus aureus bacteremia using machine learning. This model aids physicians in promptly assessing the condition and intervening proactively.Methods The patients data is sourced from the Medical Information Mart for Intensive Care IV (MIMIC-IV) database version 2.2. The patients were divided into training and testing datasets using a 7:3 random sampling ratio. The process of feature selection employed two methods: Recursive Feature Elimination (RFE) and Least Absolute Shrinkage and Selection Operator (LASSO). Prediction models were built using Extreme Gradient Boosting (XGBoost),Random Forest (RF),Logistic Regression (LR),Support Vector Machine (SVM),and Artificial Neural Network (ANN). Model validation encompassed Receiver Operating Characteristic (ROC) analysis and Decision Curve Analysis (DCA). We utilized SHAP (SHapley Additive exPlanations) values to demonstrate the significance of each feature.Results After screening, MRSA, PTT, RBC, RDW, Neutrophils_abs, Sodium, Calcium, Vancomycin concentration, MCHC, MCV, and Prognostic Nutritional Index(PNI) were selected as features for constructing the model. Through combined evaluation using ROC and DCA analyses, XGBoost demonstrated the best predictive performance, achieving an AUC value of 0.76 (95% CI: 0.66–0.85). Building a website based on the Xgboost model.The SHAP plot depicted the importance of each feature within the model.Conclusions The adoption of XGBoost for model development holds widespread acceptance in the medical domain. The prediction model for recurrent Staphylococcus aureus bacteremia readmission, developed by our team, aids physicians in timely diagnosis and treatment of patients.