TL;DR
StainNet is a fast, robust stain normalization network that uses distillation learning to achieve efficient and artifact-free color style transfer in biomedical images, outperforming existing deep learning methods in speed.
Contribution
This paper introduces StainNet, a lightweight stain normalization network utilizing distillation learning for efficient and accurate style transformation in biomedical images.
Findings
StainNet achieves comparable performance to deep learning methods.
StainNet is over 40 times faster than StainGAN.
It can normalize large whole slide images in 40 seconds.
Abstract
Stain normalization often refers to transferring the color distribution of the source image to that of the target image and has been widely used in biomedical image analysis. The conventional stain normalization is regarded as constructing a pixel-by-pixel color mapping model, which only depends on one reference image, and can not accurately achieve the style transformation between image datasets. In principle, this style transformation can be well solved by the deep learning-based methods due to its complicated network structure, whereas, its complicated structure results in the low computational efficiency and artifacts in the style transformation, which has restricted the practical application. Here, we use distillation learning to reduce the complexity of deep learning methods and a fast and robust network called StainNet to learn the color mapping between the source image and…
Peer Reviews
No public reviews on file for this paper yet. If you reviewed it on a platform where reviews are public (OpenReview, ICLR, NeurIPS, ICML), you can paste yours below so the community can read it here.
Code & Models
Videos
No videos yet. Explain this paper in a talk, walkthrough, or lecture? Add one.
