Share Email Print

Proceedings Paper

Domain-adversarial neural network for improved generalization performance of Gleason grade classification
Author(s): Ida Arvidsson; Niels Christian Overgaard; Agnieszka Krzyzanowska; Felicia-Elena Marginean; Athanasios Simoulis; Anders Bjartell; Kalle Aström; Anders Heyden
Format Member Price Non-Member Price
PDF $17.00 $21.00

Paper Abstract

ABSTRACT When training a deep learning model, the dataset used is of great importance to make sure that the model learns relevant features of the data and that it will be able to generalize to new data. However, it is typically difficult to produce a dataset without some bias toward any specific feature. Deep learning models used in histopathology have a tendency to overfit to the stain appearance of the training data { if the model is trained on data from one lab only, it will usually not be able to generalize to data from other labs. The standard technique to overcome this problem is to use color augmentation of the training data which, artificially, generates more variations for the network to learn. In this work we instead test the use of a so called domain-adversarial neural network, which is designed to prevent the model from being biased towards features that in reality are irrelevant such as the origin of an image. To test the technique, four datasets from different hospitals for Gleason grading of prostate cancer are used. We achieve state of the art results for these particular datasets, and furthermore for two of our three test datasets the approach outperforms the use of color augmentation.

Paper Details

Date Published: 16 March 2020
PDF: 6 pages
Proc. SPIE 11320, Medical Imaging 2020: Digital Pathology, 1132016 (16 March 2020); doi: 10.1117/12.2549011
Show Author Affiliations
Ida Arvidsson, Lund Univ. (Sweden)
Niels Christian Overgaard, Lund Univ. (Sweden)
Agnieszka Krzyzanowska, Lund Univ. (Sweden)
Felicia-Elena Marginean, Lund Univ. (Sweden)
Athanasios Simoulis, Lund Univ. (Sweden)
Anders Bjartell, Lund Univ. (Sweden)
Kalle Aström, Lund Univ. (Sweden)
Anders Heyden, Lund Univ. (Sweden)

Published in SPIE Proceedings Vol. 11320:
Medical Imaging 2020: Digital Pathology
John E. Tomaszewski; Aaron D. Ward, Editor(s)

© SPIE. Terms of Use
Back to Top
Sign in to read the full article
Create a free SPIE account to get access to
premium articles and original research
Forgot your username?