Towards more reliable prostate cancer detection: Incorporating clinical data and uncertainty in MRI deep learning.
Authors
Affiliations (6)
Affiliations (6)
- LaTIM, UMR1101, INSERM, University of Brest, Brest, France. Electronic address: [email protected].
- LaTIM, UMR1101, INSERM, University of Brest, Brest, France; IMT Mines Alès, Alès, France.
- LaTIM, UMR1101, INSERM, University of Brest, Brest, France.
- LaTIM, UMR1101, INSERM, University of Brest, Brest, France; Toulouse Oncopole University Cancer Institute, Toulouse, France.
- IETR - UMR CNRS 6164, CentraleSupélec, Cesson-Sévigné, France.
- Toulouse Oncopole University Cancer Institute, Toulouse, France.
Abstract
Prostate cancer (PCa) is one of the most common cancers among men, and artificial intelligence (AI) is emerging as a promising tool to enhance its diagnosis. This work proposes a classification approach for PCa cases using deep learning techniques. We conducted a comparison between unimodal models based either on biparametric magnetic resonance imaging (bpMRI) or clinical data (such as prostate-specific antigen levels, prostate volume, and age). We also introduced a bimodal model that simultaneously integrates imaging and clinical data to address the limitations of unimodal approaches. Furthermore, we propose a framework that not only detects the presence of PCa but also evaluates the uncertainty associated with the predictions. This approach makes it possible to identify highly confident predictions and distinguish them from those characterized by uncertainty, thereby enhancing the reliability and applicability of automated medical decisions in clinical practice. The results show that the bimodal model significantly improves performance, with an area under the curve (AUC) reaching 0.82±0.03, a sensitivity of 0.73±0.04, while maintaining high specificity. Uncertainty analysis revealed that the bimodal model produces more confident predictions, with an uncertainty accuracy of 0.85, surpassing the imaging-only model (which is 0.71). This increase in reliability is crucial in a clinical context, where precise and dependable diagnostic decisions are essential for patient care. The integration of clinical data with imaging data in a bimodal model not only improves diagnostic performance but also strengthens the reliability of predictions, making this approach particularly suitable for clinical use.