Comparative analysis of machine learning models for predicting hospital- and community-associated urinary tract infections using demographic, hospital, and socioeconomic predictors.

BACKGROUND

Differentiating between community-associated (CA-) and healthcare-associated (HA-) urinary tract infection (UTI) is crucial for understanding their epidemiology, identifying risk factors, and developing appropriate treatment strategies.

AIM

To build, validate, and compare machine learning models: Decision Tree, Neural Network, Logistic Regression, Random Forest, and Extreme Gradient Boosting to differentiate between the incidences of HA-UTI and CA-UTI; additionally, to identify key predictors of UTI using demographic, hospital, and socioeconomic variables.

METHODS

Patient demographics, hospital, and socioeconomic data from 2019 to 2023 were analysed.

FINDINGS

The Decision Tree model demonstrated the highest sensitivity, particularly in handling the highly imbalanced data of HAI, with a sensitivity of 87%. Logistic Regression achieved the best overall accuracy, at 95.9% for distinguishing HA-UTI from UTI-free and 93.2% for HA-UTI vs CA-UTI. Random Forest performed best in cross-validation, reaching 99.1% for HA-UTI vs UTI free and 96.2% for HA-UTI vs CA-UTI. Neural Network showed the highest specificity, at 93.4%, for HA-UTI vs CA-UTI. The area-under-the-curve values further supported these findings, ranging from 71.9% for Neural Network to 94% for Random Forest, reflecting the robustness of these models across different annual datasets. Among patient demographics, hospital, and socioeconomic variables, all models consistently identified the Nurse Units (e.g. Inpatient Units and Mental Health Units) as the most significant predictors of UTI. In addition to Nurse Units, Logistic Regression and Decision Tree identified location (e.g. various clinics and medical centres) as a key predictor.

CONCLUSION

The machine learning models demonstrated comparable overall accuracy, but differed in sensitivity and specificity across the two classification tasks-HA-UTI vs CA-UTI and HA-UTI vs UTI-free. Nurse Units consistently emerge as the most significant predictors across all years. The importance of all predictors varies from year to year.