Prostate cancer classification using radiomics and machine learning on mp-MRI validated using co-registered histology.

Background: Multi-parametric magnetic resonance imaging (mp-MRI) is emerging as a useful tool for prostate cancer (PCa) detection but currently has unaddressed limitations. Computer aided diagnosis (CAD) systems have been developed to address these needs, but many approaches used to generate and validate the models have inherent biases.

Method: All clinically significant PCa on histology was mapped to mp-MRI using a previously validated registration algorithm. Shape and size matched non-PCa regions were selected using a proposed sampling algorithm to eliminate biases towards shape and size. Further analysis was performed to assess biases regarding inter-zonal variability.

Results: A 5-feature Naïve-Bayes classifier produced an area under the receiver operating characteristic curve (AUC) of 0.80 validated using leave-one-patient-out cross-validation. As mean inter-class area mismatch increased, median AUC trended towards positively biasing classifiers to producing higher AUCs. Classifiers were invariant to differences in shape between PCa and non-PCa lesions (AUC: 0.82 vs 0.82). Performance for models trained and tested only in the peripheral zone was found to be lower than in the central gland (AUC: 0.75 vs 0.95).

Conclusion: We developed a radiomics based machine learning system to classify PCa vs non-PCa tissue on mp-MRI validated on accurately co-registered mid-gland histology with a measured target registration error. Potential biases involved in model development were interrogated to provide considerations for future work in this area.