Machine learning-guided screening of urine cultures for urinary tract infection diagnosis according to AMCLI guidelines

Machine learning-guided screening of urine cultures for urinary tract infection diagnosis according to AMCLI guidelines. Download PDF Abstract Urinary tract infections (UTIs) are among the most prevalent bacterial infections, and their diagnosis relies on the accurate interpretation of urine cultures. Manual evaluation is time-consuming, operator-dependent, and susceptible to diagnostic variability.

In this study, we propose a machine learning–guided pipeline for the automatic classification and segmentation of urine culture images, aligned with the most recent diagnostic guidelines. A dataset of 115 original chromogenic agar plate images, each corresponding to an unique patient, was acquired under standardized conditions and annotated by clinical microbiologists into three diagnostic categories: negative, positive (monomicrobial), and polymicrobial/contaminated. Data augmentation was later applied after dataset partitioning to increase image variability for model development.

Five convolutional neural networks (ResNet-18, -50, -152; VGG-11, -19) were trained and evaluated for image-level classification. VGG-19 achieved the best performance, with an accuracy of 99.49% on the test set. For colony-level detection and segmentation, we employed a modern YOLOv11 instance segmentation model, enabling simultaneous localization, classification, and mask generation for individual bacterial colonies.

Medical Disclaimer: This article is for informational purposes only and does not constitute medical advice. Always consult with qualified healthcare professionals for medical decisions and treatment options.