染色の正規化は、多くの場合、ソース画像の色分布をターゲット画像の色分布に転送することを指し、生物医学画像分析で広く使用されています。従来の染色正規化は、1つの参照画像のみに依存し、画像データセット間のスタイル変換を正確に実現できない、ピクセルごとのカラーマッピングモデルを構築するものと見なされます。原則として、このスタイル変換は、ネットワーク構造が複雑であるため、ディープラーニングベースの方法で十分に解決できますが、構造が複雑なため、計算効率が低くなり、スタイル変換のアーティファクトが発生し、実用化が制限されます。ここでは、蒸留学習を使用して深層学習手法の複雑さを軽減し、StainNetと呼ばれる高速で堅牢なネットワークを使用してソース画像とターゲット画像の間のカラーマッピングを学習します。 StainNetは、データセット全体からカラーマッピングの関係を学習し、ピクセルごとにカラー値を調整できます。ピクセルからピクセルへの方法は、ネットワークサイズを制限し、スタイル変換でのアーティファクトを回避します。細胞病理学および組織病理学データセットの結果は、StainNetが深層学習ベースの方法と同等のパフォーマンスを達成できることを示しています。計算結果は、StainNetがStainGANよりも40倍以上高速であり、40秒で100,000x100,000のスライド画像全体を正規化できることを示しています。
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 target image. StainNet can learn the color mapping relationship from a whole dataset and adjust the color value in a pixel-to-pixel manner. The pixel-to-pixel manner restricts the network size and avoids artifacts in the style transformation. The results on the cytopathology and histopathology datasets show that StainNet can achieve comparable performance to the deep learning-based methods. Computation results demonstrate StainNet is more than 40 times faster than StainGAN and can normalize a 100,000x100,000 whole slide image in 40 seconds.