Generalization of prostate cancer classification for multiple sites using deep learning

Deep learning has the potential to drastically increase the accuracy and efficiency of prostate cancer diagnosis, which would be of uttermost use. Today the diagnosis is determined manually from H&E stained specimens using a light microscope. In this paper several different approaches based on c...

Full description

Saved in:

Bibliographic Details
Main Authors:	Arvidsson, Ida, Overgaard, Niels Christian, Marginean, Felicia-Elena, Krzyzanowska, Agnieszka, Bjartell, Anders, Astrom, Kalle, Heyden, Anders
Format:	Conference Proceeding
Language:	eng
Subjects:	autoencoder Convolution Convolutional neural network digital stain separation Gleason grade Image color analysis Machine learning Prostate cancer Testing Training
Online Access:	Request full text
Tags:	Add Tag No Tags, Be the first to tag this record!

Description
Summary:	Deep learning has the potential to drastically increase the accuracy and efficiency of prostate cancer diagnosis, which would be of uttermost use. Today the diagnosis is determined manually from H&E stained specimens using a light microscope. In this paper several different approaches based on convolutional neural networks for prostate cancer classification are presented and compared, using three different datasets with different origins. The issue that algorithms trained on a certain site might not generalize to other sites, due to for example inevitable stain variations, is highlighted. Two different techniques to overcome this complication are compared; by training the networks using color augmentation and by using digital stain separation. Furthermore, the potential of using an autoencoder to get a more efficient downsampling is investigated, which turned out to be the method giving the best generalization. We achieve accuracies of 95% for classification of benign versus malignant tissue and 81% for Gleason grading for data from the same site as the training data. The corresponding accuracies for images from other sites are in average 88% and 52% respectively.
ISSN:	1945-8452