Generalization of prostate cancer classification for multiple sites using deep learning

Research output: Chapter in Book/Report/Conference proceedingPaper in conference proceeding


title = "Generalization of prostate cancer classification for multiple sites using deep learning",
abstract = "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.",
keywords = "Autoencoder, Convolutional neural network, Digital stain separation, Gleason grade, Prostate cancer",
author = "Ida Arvidsson and Overgaard, {Niels Christian} and Marginean, {Felicia Elena} and Agnieszka Krzyzanowska and Anders Bjartell and Kalle Astrom and Anders Heyden",
year = "2018",
month = may,
day = "23",
doi = "10.1109/ISBI.2018.8363552",
language = "English",
volume = "2018-April",
pages = "191--194",
booktitle = "2018 IEEE 15th International Symposium on Biomedical Imaging, ISBI 2018",
publisher = "IEEE Computer Society",
address = "United States",
note = "15th IEEE International Symposium on Biomedical Imaging, ISBI 2018 ; Conference date: 04-04-2018 Through 07-04-2018",