Self-transfer learning via patches: A prostate cancer triage approach based on bi-parametric MRI

TL;DR

This paper introduces a patch-based transfer learning method for prostate cancer triage using bi-parametric MRI, reducing reliance on annotated data and improving classification accuracy over standard approaches.

Contribution

A novel patch-based pre-training strategy that leverages ROI information to enhance CNN performance in prostate cancer classification without extensive annotations.

Findings

Patch-based pre-training outperforms traditional transfer learning methods.

Cross-domain transfer learning improves classification accuracy.

Proposed approach reduces need for annotated data in medical imaging.

Abstract

Prostate cancer (PCa) is the second most common cancer diagnosed among men worldwide. The current PCa diagnostic pathway comes at the cost of substantial overdiagnosis, leading to unnecessary treatment and further testing. Bi-parametric magnetic resonance imaging (bp-MRI) based on apparent diffusion coefficient maps (ADC) and T2-weighted (T2w) sequences has been proposed as a triage test to differentiate between clinically significant (cS) and non-clinically significant (ncS) prostate lesions. However, analysis of the sequences relies on expertise, requires specialized training, and suffers from inter-observer variability. Deep learning (DL) techniques hold promise in tasks such as classification and detection. Nevertheless, they rely on large amounts of annotated data which is not common in the medical field. In order to palliate such issues, existing works rely on transfer learning (TL) and ImageNet pre-training, which has been proven to be sub-optimal for the medical imaging domain. In this paper, we present a patch-based pre-training strategy to distinguish between cS and ncS lesions which exploit the region of interest (ROI) of the patched source domain to efficiently train a classifier in the full-slice target domain which does not require annotations by making use of transfer learning (TL). We provide a comprehensive comparison between several CNNs architectures and different settings which are presented as a baseline. Moreover, we explore cross-domain TL which exploits both MRI modalities and improves single modality results. Finally, we show how our approaches outperform the standard approaches by a considerable margin