A transfer learning approach for classification of clinical significant prostate cancers from mpMRI scans

Quan Chen; Xiang Xu; Shiliang Hu; Xiao Li; Qing Zou; Yunpeng Li

doi:10.1117/12.2279021

16 March 2017 A transfer learning approach for classification of clinical significant prostate cancers from mpMRI scans

Quan Chen, Xiang Xu, Shiliang Hu, Xiao Li, Qing Zou, Yunpeng Li

Proceedings Volume 10134, Medical Imaging 2017: Computer-Aided Diagnosis; 101344F (2017) https://doi.org/10.1117/12.2279021
Event: SPIE Medical Imaging, 2017, Orlando, Florida, United States

Abstract

Deep learning has shown a great potential in computer aided diagnosis. However, in many applications, large dataset is not available. This makes the training of a sophisticated deep learning neural network (DNN) difficult. In this study, we demonstrated that with transfer learning, we can quickly retrain start-of-the-art DNN models with limited data provided by the prostateX challenge. The training data consists of 330 lesions, only 78 were clinical significant. Efforts were made to balance the data during training. We used ImageNet pre-trained inceptionV3 and Vgg-16 model and obtained AUC of 0.81 and 0.83 respectively on the prostateX test data, good for a 4th place finish. We noticed that models trained for different prostate zone has different sensitivity. Applying scaling factors before merging the result improves the AUC for the final result.

Citation Download Citation

Quan Chen, Xiang Xu, Shiliang Hu, Xiao Li, Qing Zou, and Yunpeng Li "A transfer learning approach for classification of clinical significant prostate cancers from mpMRI scans", Proc. SPIE 10134, Medical Imaging 2017: Computer-Aided Diagnosis, 101344F (16 March 2017); https://doi.org/10.1117/12.2279021

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE

Includes PDF, HTML & Video, when available