# UNIStainNet: Foundation-Model-Guided Virtual Staining of H&E to IHC

arXiv: https://arxiv.org/abs/2603.12716 · HTML: https://mlsanity.com/paper/2603.12716/

## Key facts

- Cited by 2 papers in the map (data through 14 Oct 2026).
- Builds on 7 of its 23 references that are in the map.
- First posted on arXiv 13 Mar 2026.

## Builds on (most cited first, 7 of 7)

- [The Unreasonable Effectiveness of Deep Features as a Perceptual Metric](https://mlsanity.com/paper/1801.03924/) (2018)
- [Semantic Image Synthesis with Spatially-Adaptive Normalization](https://mlsanity.com/paper/1903.07291/) (2019)
- [Demystifying MMD GANs](https://mlsanity.com/paper/1801.01401/) (2018)
- [Contrastive Learning for Unpaired Image-to-Image Translation](https://mlsanity.com/paper/2007.15651/) (2020)
- [Pathological Semantics-Preserving Learning for H&E-to-IHC Virtual Staining](https://mlsanity.com/paper/2407.03655/) (2024)
- [From Pixels to Pathology: Restoration Diffusion for Diagnostic-Consistent Virtual IHC](https://mlsanity.com/paper/2508.02528/) (2025)
- [Topology-aware Pathological Consistency Matching for Weakly-Paired IHC Virtual Staining](https://mlsanity.com/paper/2601.02806/) (2026)

## Built on by (most cited first, 2 of 2)

- [SheafStain: Sheaf-Theoretic Schr"odinger Bridge for Spatially and Biologically Coherent Virtual Staining](https://mlsanity.com/paper/2606.11846/) (2026)
- [Semi-Supervised Virtual Staining via Morphology Preservation and Histopathological Realism Constraints](https://mlsanity.com/paper/2609.00984/) (2026)

## Abstract (arXiv)

Virtual immunohistochemistry (IHC) staining from hematoxylin and eosin (H&E) images can accelerate diagnostics by providing preliminary molecular insight directly from routine sections, reducing the need for repeat sectioning when tissue is limited. Existing methods improve realism through contrastive objectives, prototype matching, or domain alignment, yet the generator itself receives no direct guidance from pathology foundation models. We present UNIStainNet, a SPADE-UNet conditioned on dense spatial tokens from a frozen pathology foundation model (UNI), providing tissue-level semantic guidance for stain translation. A misalignment-aware loss suite preserves stain quantification accuracy, and learned stain embeddings enable a single model to serve multiple IHC markers simultaneously. On MIST, UNIStainNet achieves state-of-the-art distributional metrics on all four stains (HER2, Ki67, ER, PR) from a single unified model, where prior methods typically train separate per-stain models. On BCI, it also achieves the best distributional metrics. A tissue-type stratified failure analysis reveals that remaining errors are systematic, concentrating in non-tumor tissue. Code is available at https://github.com/facevoid/UNIStainNet.