RestainNet: A self-supervised digital re-stainer for stain normalization
Bingchao Zhao,Chu Han,Xipeng Pan,Jiatai Lin,Zongjian Yi,Changhong Liang,Xin Chen,Bingbing Li,Weihao Qiu,Danyi Li,Li Liang,Ying Wang,Zaiyi Liu
DOI: https://doi.org/10.1016/j.compeleceng.2022.108304
2022-10-01
Abstract:Color inconsistency is an inevitable challenge in computational pathology, which harms the pathological image analysis methods, especially the learning-based models. A series of approaches have been proposed for stain normalization. However, most of them are lack of flexibility in practice. In this paper, we formulated stain normalization as a digital re-staining process and proposed a self-supervised learning model, which is called RestainNet. Our network is regarded as a digital re-stainer which learns how to re-stain an unstained (grayscale) image. Two digital stains, Hematoxylin (H) and Eosin (E), were extracted from the original image by Beer–Lambert’s Law. We proposed a staining loss to maintain the correctness of stain intensity during the re-staining process. Our RestainNet outperforms existing approaches and achieves outstanding performance with regard to color correctness and structure preservation. We further conducted experiments on the segmentation and classification tasks and the proposed RestainNet achieved outstanding performance compared with SOTA methods.
engineering, electrical & electronic,computer science, interdisciplinary applications, hardware & architecture