Pathological Semantics-Preserving Learning for H&E-to-IHC Virtual Staining
Fuqiang Chen, Ranran Zhang, Boyun Zheng, Yiwen Sun, Jiahui He, Wenjian Qin
TL;DR
This work tackles H&E-to-IHC virtual staining by preserving molecular-level semantics and mitigating spatial misalignment between input and label. It introduces two strategies: PALS, which uses a Focal Optical Density map to quantify protein expression in the DAB channel and enforces MLPA-based alignment, and PCLS, which employs cross-image prototypical consistency to stabilize semantic interaction across misaligned pairs. On the BCI and MIST datasets, PSPStain outperforms state-of-the-art methods in pathological relevance metrics (lower $\text{mIOD}$ and $\text{IOD}$, higher $\text{Pearson-}R$) while maintaining image quality, with ablation confirming the value of the FOD and MLPA components. The approach enables clinically meaningful IHC reconstruction without additional annotations and can extend to other DAB-stained IHC applications.
Abstract
Conventional hematoxylin-eosin (H&E) staining is limited to revealing cell morphology and distribution, whereas immunohistochemical (IHC) staining provides precise and specific visualization of protein activation at the molecular level. Virtual staining technology has emerged as a solution for highly efficient IHC examination, which directly transforms H&E-stained images to IHC-stained images. However, virtual staining is challenged by the insufficient mining of pathological semantics and the spatial misalignment of pathological semantics. To address these issues, we propose the Pathological Semantics-Preserving Learning method for Virtual Staining (PSPStain), which directly incorporates the molecular-level semantic information and enhances semantics interaction despite any spatial inconsistency. Specifically, PSPStain comprises two novel learning strategies: 1) Protein-Aware Learning Strategy (PALS) with Focal Optical Density (FOD) map maintains the coherence of protein expression level, which represents molecular-level semantic information; 2) Prototype-Consistent Learning Strategy (PCLS), which enhances cross-image semantic interaction by prototypical consistency learning. We evaluate PSPStain on two public datasets using five metrics: three clinically relevant metrics and two for image quality. Extensive experiments indicate that PSPStain outperforms current state-of-the-art H&E-to-IHC virtual staining methods and demonstrates a high pathological correlation between the staging of real and virtual stains.
