Get our free extension to see links to code for papers anywhere online!

 Add to Chrome

 Add to Firefox

CatalyzeX Code Finder - Browser extension linking code for ML papers across the web! | Product Hunt Embed

Evolution-based Fine-tuning of CNNs for Prostate Cancer Detection

Nov 04, 2019
Khashayar Namdar, Isha Gujrathi, Masoom A. Haider, Farzad Khalvati



Convolutional Neural Networks (CNNs) have been used for automated detection of prostate cancer where Area Under Receiver Operating Characteristic (ROC) curve (AUC) is usually used as the performance metric. Given that AUC is not differentiable, common practice is to train the CNN using a loss functions based on other performance metrics such as cross entropy and monitoring AUC to select the best model. In this work, we propose to fine-tune a trained CNN for prostate cancer detection using a Genetic Algorithm to achieve a higher AUC. Our dataset contained 6-channel Diffusion-Weighted MRI slices of prostate. On a cohort of 2,955 training, 1,417 validation, and 1,334 test slices, we reached test AUC of 0.773; a 9.3% improvement compared to the base CNN model.

* Accepted for the 33rd Conference on Neural Information Processing Systems (NeurIPS 2019), Medical Imaging Meets NEURIPS Workshop 


Share this with someone who'll enjoy it:

   Access Paper Source



Share this with someone who'll enjoy it: