Normalization of HE-stained histological images using cycle consistent generative adversarial networks

Diagnostic Pathology

Stefan Schmidt¹  Cleo-Aron Weis²  Daniel Rusche²  Marlen Runz³  Jürgen Hesser⁴  Martin R. Weihrauch⁵ 

[1] Center for Mass Spectrometry and Optical Spectroscopy (CeMOS), Mannheim University of Applied Sciences, Mannheim, Germany;Institute of Pathology, University Medical Centre Mannheim, Heidelberg University, Mannheim, Germany;Institute of Pathology, University Medical Centre Mannheim, Heidelberg University, Mannheim, Germany;Mannheim Institute for Intelligent Systems in Medicine (MIISM), Medical Faculty Mannheim, Heidelberg University, Mannheim, Germany;Mannheim Institute for Intelligent Systems in Medicine (MIISM), Medical Faculty Mannheim, Heidelberg University, Mannheim, Germany;Interdisciplinary Center for Scientific Computing (IWR), Heidelberg University, Heidelberg, Germany;Central Institute for Computer Engineering (ZITI), Heidelberg University, Heidelberg, Germany;Smart In Media AG, Köln, Germany;

关键词: Histology stain normalization;    HE-stain;    Digital pathology;    Generative adversarial networks;    Unpaired image-to-image translation;    Style transfer;    Deep learning;   

DOI  :  10.1186/s13000-021-01126-y

来源: Springer

【 摘 要 】

BackgroundHistological images show strong variance (e.g. illumination, color, staining quality) due to differences in image acquisition, tissue processing, staining, etc. This can impede downstream image analysis such as staining intensity evaluation or classification. Methods to reduce these variances are called image normalization techniques.MethodsIn this paper, we investigate the potential of CycleGAN (cycle consistent Generative Adversarial Network) for color normalization in hematoxylin-eosin stained histological images using daily clinical data with consideration of the variability of internal staining protocol variations. The network consists of a generator network GB that learns to map an image X from a source domain A to a target domain B, i.e. GB:XA→XB. In addition, a discriminator network DB is trained to distinguish whether an image from domain B is real or generated. The same process is applied to another generator-discriminator pair (GA,DA), for the inverse mapping GA:XB→XA. Cycle consistency ensures that a generated image is close to its original when being mapped backwards (GA(GB(XA))≈XA and vice versa). We validate the CycleGAN approach on a breast cancer challenge and a follicular thyroid carcinoma data set for various stain variations. We evaluate the quality of the generated images compared to the original images using similarity measures. In addition, we apply stain normalization on pathological lymph node data from our institute and test the gain from normalization on a ResNet classifier pre-trained on the Camelyon16 data set.ResultsQualitative results of the images generated by our network are compared to original color distributions. Our evaluation indicates that by mapping images to a target domain, the similarity training images from that domain improves up to 96%. We also achieve a high cycle consistency for the generator networks by obtaining similarity indices greater than 0.9. When applying the CycleGAN normalization to HE-stain images from our institute the kappa-value of the ResNet-model that is only trained on Camelyon16 data is increased more than 50%.ConclusionsCycleGANs have proven to efficiently normalize HE-stained images. The approach compensates for deviations resulting from image acquisition (e.g. different scanning devices) as well as from tissue staining (e.g. different staining protocols), and thus overcomes the staining variations in images from various institutions.The code is publicly available at https://github.com/m4ln/stainTransfer_CycleGAN_pytorch. The data set supporting the solutions is available at https://doi.org/10.11588/data/8LKEZF.

【 授权许可】

CC BY