Unpaired Image-to-Image Translation for Segmentation and Signal Unmixing
This work addresses domain adaptation and signal separation in biomedical imaging, offering a novel solution for researchers in computational pathology, but it is incremental as it builds on CycleGAN with specific modifications.
The authors tackled the problem of unpaired image-to-image translation for biomedical tasks, introducing Ui2i to preserve content integrity, and demonstrated its effectiveness in nuclear segmentation and signal unmixing, achieving improved structural preservation compared to typical methods.
This work introduces Ui2i, a novel model for unpaired image-to-image translation, trained on content-wise unpaired datasets to enable style transfer across domains while preserving content. Building on CycleGAN, Ui2i incorporates key modifications to better disentangle content and style features, and preserve content integrity. Specifically, Ui2i employs U-Net-based generators with skip connections to propagate localized shallow features deep into the generator. Ui2i removes feature-based normalization layers from all modules and replaces them with approximate bidirectional spectral normalization -- a parameter-based alternative that enhances training stability. To further support content preservation, channel and spatial attention mechanisms are integrated into the generators. Training is facilitated through image scale augmentation. Evaluation on two biomedical tasks -- domain adaptation for nuclear segmentation in immunohistochemistry (IHC) images and unmixing of biological structures superimposed in single-channel immunofluorescence (IF) images -- demonstrates Ui2i's ability to preserve content fidelity in settings that demand more accurate structural preservation than typical translation tasks. To the best of our knowledge, Ui2i is the first approach capable of separating superimposed signals in IF images using real, unpaired training data.