Efficient Self-Supervised Grading of Prostate Cancer Pathology

TL;DR

This paper introduces a novel self-supervised learning framework for efficient and accurate ISUP grading of prostate cancer from whole slide images, addressing challenges like gigapixel size, stain variability, and limited annotations.

Contribution

It proposes a task-specific self-supervised model fine-tuned with ordinal regression, improving patch-level feature learning and grading accuracy on large multicenter datasets.

Findings

Outperforms state-of-the-art methods on PANDA and SICAP datasets.

Effectively learns stain-agnostic Gleason pattern features.

Achieves high grading accuracy with slide-level labels.

Abstract

Prostate cancer grading using the ISUP system (International Society of Urological Pathology) for treatment decisions is highly subjective and requires considerable expertise. Despite advances in computer-aided diagnosis systems, few have handled efficient ISUP grading on Whole Slide Images (WSIs) of prostate biopsies based only on slide-level labels. Some of the general challenges include managing gigapixel WSIs, obtaining patch-level annotations, and dealing with stain variability across centers. One of the main task-specific challenges faced by deep learning in ISUP grading, is the learning of patch-level features of Gleason patterns (GPs) based only on their slide labels. In this scenario, an efficient framework for ISUP grading is developed. The proposed TSOR is based on a novel Task-specific Self-supervised learning (SSL) model, which is fine-tuned using Ordinal Regression. Since the diversity of training samples plays a crucial role in SSL, a patch-level dataset is created to be relatively balanced w.r.t. the Gleason grades (GGs). This balanced dataset is used for pre-training, so that the model can effectively learn stain-agnostic features of the GP for better generalization. In medical image grading, it is desirable that misclassifications be as close as possible to the actual grade. From this perspective, the model is then fine-tuned for the task of ISUP grading using an ordinal regression-based approach. Experimental results on the most extensive multicenter prostate biopsies dataset (PANDA challenge), as well as the SICAP dataset, demonstrate the effectiveness of this novel framework compared to state-of-the-art methods.