Search papers, labs, and topics across Lattice.
This paper introduces a prototype-based weakly-supervised deep learning framework for interpretable prostate cancer grading from histopathology images. The network learns prototypical features associated with each cancer grade through pre-training at the patch level, and is then fine-tuned using a prototype-aware loss function. An attention-based dynamic pruning mechanism is incorporated to handle inter-sample heterogeneity by selectively emphasizing relevant prototypes.
Prototype-based deep learning offers a more trustworthy approach to prostate cancer grading by mirroring a pathologist's workflow of comparing suspicious regions with clinically validated examples.
Prostate cancer being one of the frequently diagnosed malignancy in men, the rising demand for biopsies places a severe workload on pathologists. The grading procedure is tedious and subjective, motivating the development of automated systems. Although deep learning has made inroads in terms of performance, its limited interpretability poses challenges for widespread adoption in high-stake applications like medicine. Existing interpretability techniques for prostate cancer classifiers provide a coarse explanation but do not reveal why the highlighted regions matter. In this scenario, we propose a novel prototype-based weakly-supervised framework for an interpretable grading of prostate cancer from histopathology images. These networks can prove to be more trustworthy since their explicit reasoning procedure mirrors the workflow of a pathologist in comparing suspicious regions with clinically validated examples. The network is initially pre-trained at patch-level to learn robust prototypical features associated with each grade. In order to adapt it to a weakly-supervised setup for prostate cancer grading, the network is fine-tuned with a new prototype-aware loss function. Finally, a new attention-based dynamic pruning mechanism is introduced to handle inter-sample heterogeneity, while selectively emphasizing relevant prototypes for optimal performance. Extensive validation on the benchmark PANDA and SICAP datasets confirms that the framework can serve as a reliable assistive tool for pathologists in their routine diagnostic workflows.
