Stain-transforming cycle-consistent generative adversarial networks for improved segmentation of renal histopathology
Keywords: Deep learning, generative adversarial networks, medical imaging, stain transformation
TL;DR: Adapted cycleGAN architecture great for stain conversion improving segmentation performance
Abstract: The performance of deep learning applications in digital histopathology can deteriorate significantly due to staining variations across centers. We employ cycle-consistent generative adversarial networks (cycleGANs) for unpaired image-to-image translation, facilitating between-center stain transformation. We find that modifications to the original cycleGAN architecture make it more suitable for stain transformation, creating artificially stained images of high quality. Specifically, changing the generator model to a smaller U-net-like architecture, adding an identity loss term, increasing the batch size and the learning all led to improved training stability and performance. Furthermore, we propose a method for dealing with tiling artifacts when applying the network on whole slide images (WSIs). We apply our stain transformation method on two datasets of PAS-stained (Periodic Acid-Schiff) renal tissue sections from different centers. We show that stain transformation is beneficial to the performance of cross-center segmentation, raising the Dice coefficient from 0.36 to 0.85 and from 0.45 to 0.73 on the two datasets.
Code Of Conduct: I have read and accept the code of conduct.
Remove If Rejected: (optional) Remove submission if paper is rejected.