PCa-RadHop: A transparent and lightweight feed-forward method for clinically significant prostate cancer segmentation

Comput Med Imaging Graph. 2024 Sep:116:102408. doi: 10.1016/j.compmedimag.2024.102408. Epub 2024 Jun 10.

Abstract

Prostate Cancer is one of the most frequently occurring cancers in men, with a low survival rate if not early diagnosed. PI-RADS reading has a high false positive rate, thus increasing the diagnostic incurred costs and patient discomfort. Deep learning (DL) models achieve a high segmentation performance, although require a large model size and complexity. Also, DL models lack of feature interpretability and are perceived as "black-boxes" in the medical field. PCa-RadHop pipeline is proposed in this work, aiming to provide a more transparent feature extraction process using a linear model. It adopts the recently introduced Green Learning (GL) paradigm, which offers a small model size and low complexity. PCa-RadHop consists of two stages: Stage-1 extracts data-driven radiomics features from the bi-parametric Magnetic Resonance Imaging (bp-MRI) input and predicts an initial heatmap. To reduce the false positive rate, a subsequent stage-2 is introduced to refine the predictions by including more contextual information and radiomics features from each already detected Region of Interest (ROI). Experiments on the largest publicly available dataset, PI-CAI, show a competitive performance standing of the proposed method among other deep DL models, achieving an area under the curve (AUC) of 0.807 among a cohort of 1,000 patients. Moreover, PCa-RadHop maintains orders of magnitude smaller model size and complexity.

Keywords: Biparametric MRI; Data-driven radiomics; Feed-forward model; Interpretable pipeline; Prostate cancer segmentation.

MeSH terms

  • Algorithms
  • Deep Learning
  • Humans
  • Image Interpretation, Computer-Assisted / methods
  • Magnetic Resonance Imaging* / methods
  • Male
  • Prostatic Neoplasms* / diagnostic imaging