Automated detection of multi-class urinary sediment particles: An accurate deep learning approach

Abstract

Urine microscopy is an essential diagnostic tool for kidney and urinary tract diseases, with automated analysis of urinary sediment particles improving diagnostic efficiency. However, some urinary sediment particles remain challenging to identify due to individual variations, blurred boundaries, and unbalanced samples. This research aims to mitigate the adverse effects of urine sediment particles while improving multi-class detection performance. We proposed an innovative model based on improved YOLOX for detecting urine sediment particles (YUS-Net). The combination of urine sediment data augmentation and overall pre-trained weights enhances model optimization potential. Furthermore, we incorporate the attention module into the critical feature transfer path and employ a novel loss function, Varifocal loss, to facilitate the extraction of discriminative features, which assists in the identification of densely distributed small objects. Based on the USE dataset, YUS-Net achieves the mean Average Precision (mAP) of 96.07%, 99.35% average precision, and 96.77% average recall, with a latency of 26.13 ms per image. The specific metrics for each category are as follows: cast: 99.66% AP; cryst: 100% AP; epith: 92.31% AP; epithn: 100% AP; eryth: 92.31% AP; leuko: 99.90% AP; mycete: 99.96% AP. With a practical network structure, YUS-Net achieved efficient, accurate, end-to-end urinary sediment particle detection. The model takes native high-resolution images as input without additional steps. Finally, a data augmentation strategy appropriate for the urinary microscopic image domain is established, which provides a novel approach for applying other methods in urine microscopic images.