ProsDectNet: Bridging the Gap in Prostate Cancer Detection via Transrectal B-mode Ultrasound Imaging

Sulaiman Vesal,Indrani Bhattacharya,Hassan Jahanandish,Xinran Li,Zachary Kornberg,Steve Ran Zhou,Elijah Richard Sommer,Moon Hyung Choi,Richard E. Fan,Geoffrey A. Sonn,Mirabela Rusu
2023-12-09
Abstract:Interpreting traditional B-mode ultrasound images can be challenging due to image artifacts (e.g., shadowing, speckle), leading to low sensitivity and limited diagnostic accuracy. While Magnetic Resonance Imaging (MRI) has been proposed as a solution, it is expensive and not widely available. Furthermore, most biopsies are guided by Transrectal Ultrasound (TRUS) alone and can miss up to 52% cancers, highlighting the need for improved targeting. To address this issue, we propose ProsDectNet, a multi-task deep learning approach that localizes prostate cancer on B-mode ultrasound. Our model is pre-trained using radiologist-labeled data and fine-tuned using biopsy-confirmed labels. ProsDectNet includes a lesion detection and patch classification head, with uncertainty minimization using entropy to improve model performance and reduce false positive predictions. We trained and validated ProsDectNet using a cohort of 289 patients who underwent MRI-TRUS fusion targeted biopsy. We then tested our approach on a group of 41 patients and found that ProsDectNet outperformed the average expert clinician in detecting prostate cancer on B-mode ultrasound images, achieving a patient-level ROC-AUC of 82%, a sensitivity of 74%, and a specificity of 67%. Our results demonstrate that ProsDectNet has the potential to be used as a computer-aided diagnosis system to improve targeted biopsy and treatment planning.
Image and Video Processing,Computer Vision and Pattern Recognition
What problem does this paper attempt to address?
This paper aims to solve several key problems in prostate cancer detection: 1. **Interpretation challenges of traditional B - mode ultrasound images**: Due to the presence of image artifacts (such as shadows, speckle noise), the sensitivity of B - mode ultrasound images in prostate cancer detection is low, and the diagnostic accuracy is limited. 2. **Limitations of MRI**: Although magnetic resonance imaging (MRI) has been proposed as a solution, its high cost and limited availability limit its wide application. 3. **Problems with transrectal ultrasound (TRUS) - guided biopsy**: Most prostate biopsies rely solely on TRUS, and this biopsy method may miss up to 52% of cancers, highlighting the need for improved targeting techniques. To solve these problems, the authors proposed **ProsDectNet**, a multi - task deep - learning method capable of localizing prostate cancer on B - mode ultrasound images. Specifically, ProsDectNet improves detection performance in the following ways: - **Pre - training and fine - tuning**: The model is first pre - trained using data annotated by radiologists and then fine - tuned using pathologically confirmed labels. - **Lesion detection and classification heads**: The model contains a lesion detection head and a classification head to determine whether each area contains cancer, thereby reducing false - positive predictions. - **Uncertainty minimization**: Calculate the entropy map to evaluate the uncertainty of the prediction, and improve the confidence and accuracy of the model by minimizing the entropy. The paper demonstrates the superior performance of this model in detecting prostate cancer by training and validating ProsDectNet on a dataset of 289 patients and evaluating it on a test set of 41 patients, achieving a patient - level ROC - AUC value of 82%, a sensitivity of 74%, and a specificity of 67%. These results suggest that ProsDectNet has the potential to be used as a computer - aided diagnosis system to improve targeted biopsy and treatment planning.