Figure 4 | Scientific Reports

Figure 4

From: A deep learning system for prostate cancer diagnosis and grading in whole slide images of core needle biopsies

Figure 4

Proposed Multi-Task Model for learning features robust to color stain variations in histopathology images. During the training phase, the model learns from three supervisory signals—reconstruction loss between raw and color augmented image, KL divergence loss between raw and color augmented image logits, and segmentation DICE loss. At inference, only the layers required for segmentation are used.

Back to article page