StainExpert: A Unified Multi-Expert Diffusion Framework for Multi-Target Pathological Stain Translation

Zeyu Liu, Yufang He, Tianyi Zhang, Chenbin Ma, Fan Song, Huijie Wu, Ruxin Cai, Haoran Guo, Bo Wen, Haonan Zhang, Peng Zhang, Dachun Zhao, Guanglei Zhang*
Beihang University, National Singapore University, Shanxi University, Peking Union Medical College Hospital
(IEEE Transactions on Medical Imaging)

*Indicates corresponding author

Abstract

Pathological examination is critical for disease diagnosis, with various staining techniques providing detailed visualization of tissue structures. While hematoxylin and eosin (H&E) staining remains predominant, the special stains, immunohistochemistry (IHC) and multiplex immunofluorescence (mpIF) provide valuable complementary information for comprehensive diagnosis. However, current multi-stain workflows are time-consuming, labor-intensive, and tissue-exhaustive, requiring either re-staining or additional tissue sectioning. To address these challenges, we propose StainExpert, a unified multimodal diffusion framework for source-to-multi-target pathological stain translation. Unlike existing approaches that require separate models for each staining pair, StainExpert establishes the first multi-expert system where specialized networks collaboratively learn staining principles while maintaining domain-specific expertise. Through multi-expert and multi-objective optimization, it enables efficient translation from a single source to multiple targets. Additionally, our multimodal diffusion architecture integrates textual guidance with visual features, achieving superior accuracy and pathology-informed translation. Leveraging parameter-efficient design and model distillation, StainExpert matches GAN-level efficiency while delivering superior generation quality. We validate StainExpert across three datasets spanning H&E, special stains, IHC, and mpIF modalities. Extensive evaluation demonstrates that StainExpert generates high-quality virtual stains that preserve critical pathological features for accurate diagnosis. Beyond robust cross-domain generalization, StainExpert offers a transformative platform for efficient multi-target stain translation, advancing toward streamlined, tissue-conserving, and resource-efficient diagnostic workflows in computational pathology.

Video Presentation (is coming)

BibTeX


  @ARTICLE{11284894,
  author={Liu, Zeyu and He, Yufang and Zhang, Tianyi and Ma, Chenbin and Song, Fan and Wu, Huijie and Cai, Ruxin and Guo, Haoran and Zhang, Haonan and Wen, Bo and Zhang, Peng and Zhao, Dachun and Zhang, Guanglei},
  journal={IEEE Transactions on Medical Imaging},
  title={StainExpert: A Unified Multi-Expert Diffusion Framework for Multi-Target Pathological Stain Translation},
  year={2025},
  volume={},
  number={},
  pages={1-1},
  keywords={Histopathology; multiple stain translation; virtual staining; diffusion model; mixture of experts (MoE)},
  doi={10.1109/TMI.2025.3641562}}