A General Pipeline for Glomerulus Whole-Slide Image Segmentation
Quan Huu Cap
TL;DR
Problem: Glomerulus segmentation in whole-slide kidney images is critical for CKD diagnosis but challenging due to glomerular variability and staining differences. Method: A general stitching-based segmentation pipeline extracts overlapping patches, aggregates their predictions over overlaps, and evaluates multiple models on two large public datasets (KPIs and Mice glomeruli). Findings: The approach achieves state-of-the-art performance for both patch-level and WSI-level segmentation, notably with Mask2Former attaining a WSI Dice of 94.64 on KPIs; stitching consistently improves results across models and datasets. Significance: The work provides a practical, reproducible framework with publicly available code and pretrained models that can accelerate CKD-related diagnostics and further research.
Abstract
Whole-slide images (WSI) glomerulus segmentation is essential for accurately diagnosing kidney diseases. In this work, we propose a general and practical pipeline for glomerulus segmentation that effectively enhances both patch-level and WSI-level segmentation tasks. Our approach leverages stitching on overlapping patches, increasing the detection coverage, especially when glomeruli are located near patch image borders. In addition, we conduct comprehensive evaluations from different segmentation models across two large and diverse datasets with over 30K glomerulus annotations. Experimental results demonstrate that models using our pipeline outperform the previous state-of-the-art method, achieving superior results across both datasets and setting a new benchmark for glomerulus segmentation in WSIs. The code and pre-trained models are available at https://github.com/huuquan1994/wsi_glomerulus_seg.
