MIE: Magnification-integrated ensemble method for improving glomeruli segmentation in medical imaging

Abstract

Background and objective: Glomeruli are crucial for blood filtration, waste removal, and regulation of essential substances in the body. Traditional methods for detecting glomeruli rely on human interpretation, which can lead to variability. AI techniques have improved this process; however, most studies have used images with fixed magnification. This study proposes a novel magnification-integrated ensemble method to enhance glomerular segmentation accuracy.

Methods: Whole-slide images (WSIs) from 12 patients were used for training, two for validation, and one for testing. Patch and mask images were extracted at 256 × 256 size × x2, x3, and x4 magnification levels. Data augmentation techniques, such as RandomResize, RandomCrop, and RandomFlip, were used. The segmentation model underwent 80,000 iterations with a stochastic gradient descent (SGD).

Results: Performance varied with changes in magnification. The models trained on high-magnification images showed significant drops when tested at lower magnifications, and vice versa. The proposed method improved segmentation accuracy across different magnifications, achieving 87.72 mIoU and 93.04 Dice score with the U-Net model.

Conclusions: The magnification-integrated ensemble method significantly enhanced glomeruli segmentation accuracy across varying magnifications, thereby addressing the limitations of fixed magnification models. This approach improves the robustness and reliability of AI-driven diagnostic tools, potentially benefiting various medical imaging applications by ensuring consistent performance despite changes in image magnification.