Clinically significant prostate cancer detection with deep learning in a multi-center magnetic resonance imaging study.
Authors
Affiliations (10)
Affiliations (10)
- Unidad Mixta de Imagen Biomédica e Inteligencia Artificial FISABIO-CIPF, Fundación para el Fomento de la Investigación Sanitario y Biomédica de la Comunidad Valenciana, 46020, Valencia, Spain. [email protected].
- Unidad Mixta de Imagen Biomédica e Inteligencia Artificial FISABIO-CIPF, Fundación para el Fomento de la Investigación Sanitario y Biomédica de la Comunidad Valenciana, 46020, Valencia, Spain.
- Urology, Hospital Arnau de Vilanova, 46015, Valencia, Spain.
- Computational Biomedicine Laboratory, Principe Felipe Research Center (CIPF), 46012, Valencia, Spain.
- Electronics and Automation Department, Universidad Autónoma de Manizales, Manizales, 170001, Caldas, Colombia.
- Faculty of Engineering and Sciences, Universidad Adolfo Ibáñez, 7941169, Santiago, Chile.
- Department of Systems and Informatics, Universidad de Caldas, Manizales, 170001, Caldas, Colombia.
- GobLab School of Government, Universidad Adolfo Ibáñez, Santiago, Chile.
- BDSLab, Instituto Universitario de Tecnologías de la Información y Comunicaciones, Universitat Politècnica de Valencia, Valencia, 46022, Spain.
- Unidad Mixta de Imagen Biomédica e Inteligencia Artificial FISABIO-CIPF, Fundación para el Fomento de la Investigación Sanitario y Biomédica de la Comunidad Valenciana, 46020, Valencia, Spain. [email protected].
Abstract
Accurate early detection of clinically significant prostate cancer is crucial for improving patient outcomes. However, traditional diagnostic methods such as Digital Rectal Exam and Prostate-Specific Antigen (PSA) tests often lack the sensitivity and specificity needed for effective diagnosis. This study presents an AI-based approach for csPCa classification using MRI data, incorporating both the PI-CAI Challenge dataset and a newly compiled, diverse BIMCV Prostate dataset comprising over 9000 MRI sessions from 16 healthcare centers in the Valencian Region. The methodology includes a robust preprocessing pipeline, featuring prostate segmentation with a custom-trained nnUNet model, and utilizes a 3D variant of EfficientNet-B7. To ensure robustness, we employed a transfer learning strategy where five models pretrained on PI-CAI were fine-tuned on the BIMCV dataset and aggregated using a stacked meta-learner. This ensemble approach yielded a Receiver Operating Characteristic Area Under the Curve of 0.816 on the independent hold-out set, significantly outperforming a non-pretrained baseline (AUC 0.71). Furthermore, we demonstrated that synthesizing missing ADC maps using a mono-exponential model serves as an effective data augmentation strategy, preventing data loss without introducing domain shift. Interpretability techniques such as occlusion sensitivity and guided backpropagation were employed to provide insights into the model's decision-making process, enhancing transparency. This research highlights the potential of AI-enhanced MRI techniques in advancing csPCa detection and diagnosis.