TRUSWorthy: Toward Clinically Applicable Deep Learning for Confident Detection of Prostate Cancer in Micro-Ultrasound

TL;DR

TRUSWorthy is a comprehensive deep learning system designed to improve prostate cancer detection in micro-ultrasound, addressing challenges like tissue heterogeneity, class imbalance, and limited annotations, achieving high accuracy and confidence calibration.

Contribution

The paper introduces TRUSWorthy, an integrated deep learning pipeline combining self-supervised learning, transformers, and ensembling to enhance prostate cancer detection in challenging clinical ultrasound data.

Findings

Outperforms previous methods in accuracy and calibration.

Achieves 79.9% AUROC and 71.5% balanced accuracy.

Up to 91% balanced accuracy on top-confidence predictions.

Abstract

While deep learning methods have shown great promise in improving the effectiveness of prostate cancer (PCa) diagnosis by detecting suspicious lesions from trans-rectal ultrasound (TRUS), they must overcome multiple simultaneous challenges. There is high heterogeneity in tissue appearance, significant class imbalance in favor of benign examples, and scarcity in the number and quality of ground truth annotations available to train models. Failure to address even a single one of these problems can result in unacceptable clinical outcomes.We propose TRUSWorthy, a carefully designed, tuned, and integrated system for reliable PCa detection. Our pipeline integrates self-supervised learning, multiple-instance learning aggregation using transformers, random-undersampled boosting and ensembling: these address label scarcity, weak labels, class imbalance, and overconfidence, respectively. We train and rigorously evaluate our method using a large, multi-center dataset of micro-ultrasound data. Our method outperforms previous state-of-the-art deep learning methods in terms of accuracy and uncertainty calibration, with AUROC and balanced accuracy scores of 79.9% and 71.5%, respectively. On the top 20% of predictions with the highest confidence, we can achieve a balanced accuracy of up to 91%. The success of TRUSWorthy demonstrates the potential of integrated deep learning solutions to meet clinical needs in a highly challenging deployment setting, and is a significant step towards creating a trustworthy system for computer-assisted PCa diagnosis.