Comparison of different augmentation techniques for improved generalization performance for gleason grading

Arvidsson, Ida; Overgaard, Niels Christian; Astrom, Kalle; Heyden, Anders (2019). Comparison of different augmentation techniques for improved generalization performance for gleason grading 2019 IEEE 16th International Symposium on Biomedical Imaging (ISBI 2019), 923 - 927. 16th IEEE International Symposium on Biomedical Imaging, ISBI 2019. Venice, Italy: IEEE Computer Society
Download:
DOI:
Conference Proceeding/Paper | Published | English
Authors:
Arvidsson, Ida ; Overgaard, Niels Christian ; Astrom, Kalle ; Heyden, Anders
Department:
Mathematics (Faculty of Engineering)
eSSENCE: The e-Science Collaboration
ELLIIT: the Linköping-Lund initiative on IT and mobile communication
Abstract:

The fact that deep learning based algorithms used for digital pathology tend to overfit to the site of the training data is well-known. Since an algorithm that does not generalize is not very useful, we have in this work studied how different data augmentation techniques can reduce this problem but also how data from different sites can be normalized to each other. For both of these approaches we have used cycle generative adversarial networks (GAN); either to generate more examples to train on or to transform images from one site to another. Furthermore, we have investigated to what extent standard augmentation techniques improve the generalization performance. We performed experiments on four datasets with slides from prostate biopsies, stained with HE, detailed annotated with Gleason grades. We obtained results similar to previous studies, with accuracies of 77% for Gleason grading for images from the same site as the training data and 59% for images from other sites. However, we also found out that the use of traditional augmentation techniques gave better performance compared to when using cycle GANs, either to augment the training data or to normalize the test data.

Keywords:
Augmentation ; Convolutional neural network ; Cycle generative adversarial network ; Deep learning ; Digital pathology ; Gleason grading ; Prostate cancer
ISBN:
9781538636411
LUP-ID:
32f41e4b-d2e7-4f3c-a45a-9e2aed57cb5d | Link: https://lup.lub.lu.se/record/32f41e4b-d2e7-4f3c-a45a-9e2aed57cb5d | Statistics

Cite this