デジタルパソロジーでは、染色手順とスキャナーが異なると、特に異なる研究所間で、スライド全体の画像(WSI)にかなりの色の変化が生じます。これらの色の変化により、トレーニングドメインから外部の病理学データへの深層学習ベースの方法の一般化が不十分になります。テストのパフォーマンスを向上させるために、染色正規化手法を使用して、トレーニングとテストドメイン間の差異を減らします。あるいは、トレーニング中に色の増強を適用して、テスト時に色を正規化する追加の手順を実行せずに、より堅牢なモデルを作成することもできます。多種多様な現実的な組織学の染色色をシミュレートできる新しい色増強技術、HistAuGANを提案します。これにより、トレーニング中に適用したときにニューラルネットワークが染色不変になります。画像から画像への変換のための生成的敵対的ネットワーク(GAN)に基づいて、私たちのモデルは、画像のコンテンツ、つまり形態学的組織構造を染色色属性から解きほぐします。複数のドメインでトレーニングできるため、スライドの準備とイメージングのプロセスで導入された他のドメイン固有のバリエーションだけでなく、さまざまな染色色をカバーすることを学びます。 HistAuGANが、公開されているデータセットCamelyon17の分類タスクで従来の色増強手法よりも優れていることを示し、現在のバッチ効果を軽減できることを示します。
In digital pathology, different staining procedures and scanners cause substantial color variations in whole-slide images (WSIs), especially across different laboratories. These color shifts result in a poor generalization of deep learning-based methods from the training domain to external pathology data. To increase test performance, stain normalization techniques are used to reduce the variance between training and test domain. Alternatively, color augmentation can be applied during training leading to a more robust model without the extra step of color normalization at test time. We propose a novel color augmentation technique, HistAuGAN, that can simulate a wide variety of realistic histology stain colors, thus making neural networks stain-invariant when applied during training. Based on a generative adversarial network (GAN) for image-to-image translation, our model disentangles the content of the image, i.e., the morphological tissue structure, from the stain color attributes. It can be trained on multiple domains and, therefore, learns to cover different stain colors as well as other domain-specific variations introduced in the slide preparation and imaging process. We demonstrate that HistAuGAN outperforms conventional color augmentation techniques on a classification task on the publicly available dataset Camelyon17 and show that it is able to mitigate present batch effects.