StainFuser: Controlling Diffusion for Faster Neural Style Transfer in Multi-Gigapixel Histology Images
Not on the map. Why?
Abstract (arXiv)
Stain normalization algorithms aim to transform the color and intensity characteristics of a source multi-gigapixel histology image to match those of a target image, mitigating inconsistencies in the appearance of stains used to highlight cellular components in the images. We propose a new approach, StainFuser, which treats this problem as a style transfer task using a novel Conditional Latent Diffusion architecture, eliminating the need for handcrafted color components. With this method, we curate SPI-2M the largest stain normalization dataset to date of over 2 million histology images with neural style transfer for high-quality transformations. Trained on this data, StainFuser outperforms current state-of-the-art deep learning and handcrafted methods in terms of the quality of normalized images and in terms of downstream model performance on the CoNIC dataset.
Key facts. Cited by 6 papers in the map (data through 7 Oct 2026). Builds on 29 of its 52 references that are in the map. First posted on arXiv 14 Mar 2024.
Builds on
29 of 52 references.
References · 29
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 2020Denoising Diffusion Probabilistic Modelscited by 20,851
- 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,694
- 2014ImageNet Large Scale Visual Recognition Challengecited by 8,441
- 2021Diffusion Models Beat GANs on Image Synthesiscited by 8,185
- 2015Deep Unsupervised Learning using Nonequilibrium Thermodynamicscited by 7,005
- 2021Emerging Properties in Self-Supervised Vision Transformerscited by 6,893
- 2022Photorealistic Text-to-Image Diffusion Models with Deep Language Understandingcited by 6,322
- 2023Adding Conditional Control to Text-to-Image Diffusion Modelscited by 5,466
- 2022LAION-5B: An open large-scale dataset for training next generation image-text modelscited by 4,131
- 2022DreamBooth: Fine Tuning Text-to-Image Diffusion Models for Subject-Driven Generationcited by 3,286
- 2022RePaint: Inpainting using Denoising Diffusion Probabilistic Modelscited by 1,419
17 more
- 2018ESRGAN: Enhanced Super-Resolution Generative Adversarial Networkscited by 1,250
- 2015A Neural Algorithm of Artistic Stylecited by 480
- 2018HoVer-Net: Simultaneous Segmentation and Classification of Nuclei in Multi-Tissue Histology Imagescited by 387
- 2023Implicit Diffusion Models for Continuous Super-Resolutioncited by 216
- 2019Quantifying the effects of data augmentation and stain color normalization in convolutional neural networks for computational pathologycited by 172
- 2023PathLDM: Text conditioned Latent Diffusion Model for Histopathologycited by 51
- 2021CoNIC: Colon Nuclei Identification and Counting Challenge 2022cited by 44
- 2022Handcrafted Histological Transformer (H2T): Unsupervised Representation of Whole Slide Imagescited by 32
- 2023Domain Generalization in Computational Pathology: Survey and Guidelinescited by 29
- 2020Pix2Pix-based Stain-to-Stain Translation: A Solution for Robust Stain Normalization in Histopathology Images Analysiscited by 23
- 2017The importance of stain normalization in colorectal tissue classification with convolutional networkscited by 20
- 2017Neural Stain-Style Transfer Learning using GAN for Histopathological Imagescited by 18
- 2022Panoptic segmentation with highly imbalanced semantic labelscited by 5
- 2021A QuadTree Image Representation for Computational Pathologycited by 4
- 2023Nuclear Segmentation and Classification: On Color & Compression Generalizationcited by 4
- 2022Mapping the landscape of histomorphological cancer phenotypes using self-supervised learning on unlabeled, unannotated pathology slidescited by 3
- 2020HydraMix-Net: A Deep Multi-task Semi-supervised Learning Approach for Cell Detection and Classificationcited by 2
Built on by
Citing papers per year: 2025: 4; 2026: 2.
6 papers.
- 2025Content Generation Models in Computational Pathology: A Comprehensive Survey on Methods, Applications, and Challengescited by 5
- 2026Cross-Fusion Distance: A Novel Metric for Measuring Fusion and Separability Between Data Groups in Representation Spacecited by 1
- 2025IMPLICITSTAINER: Resolution Agnostic Data-Efficient Virtual Staining Using Neural Implicit Functionscited by 0
- 2025K-Stain: Keypoint-Driven Correspondence for H&E-to-IHC Virtual Stainingcited by 0
- 2025When normalization hallucinates: unseen risks in AI-powered whole slide image processingcited by 0
- 2026Histopathology Image Normalization via Latent Manifold Compactioncited by 0
- 2026Histopathology Image Normalization via Latent Manifold Compaction27 Feb 2026
- 2026Cross-Fusion Distance: A Novel Metric for Measuring Fusion and Separability Between Data Groups in Representation Space29 Jan 2026
- 2025When normalization hallucinates: unseen risks in AI-powered whole slide image processing8 Dec 2025
- 2025K-Stain: Keypoint-Driven Correspondence for H&E-to-IHC Virtual Staining10 Nov 2025
- 2025Content Generation Models in Computational Pathology: A Comprehensive Survey on Methods, Applications, and Challenges16 May 2025
- 2025IMPLICITSTAINER: Resolution Agnostic Data-Efficient Virtual Staining Using Neural Implicit Functions14 May 2025
Similar, not cited
Close in wording, no citation either way
- 2026HistDiT: A Structure-Aware Latent Conditional Diffusion Model for High-Fidelity Virtual Staining in Histopathology
- 2024Style-Extracting Diffusion Models for Semi-Supervised Histopathology Segmentation
- 2025GANs vs. Diffusion Models for virtual staining with the HER2match dataset
- 2025Virtual Multiplex Staining for Histological Images using a Marker-wise Conditioned Diffusion Model
- 2022A Morphology Focused Diffusion Probabilistic Model for Synthesis of Histopathology Images
- 2024Mask-guided cross-image attention for zero-shot in-silico histopathologic image generation with a diffusion model
- 2026A Generative Foundation Model for Multimodal Histopathology
- 2025VLM-based Prompts as the Optimal Assistant for Unpaired Histopathology Virtual Staining