Unsupervised Domain Adaptation in Brain Lesion Segmentation with Adversarial Networks

• Published in: Information Processing in Medical Imaging Vol. 10265; pp. 597 - 609
• Main Author: Niethammer, Marc
• Format: Book Chapter
• Language: English
• Published: Switzerland Springer International Publishing AG 2017; Springer International Publishing
• Series: Lecture Notes in Computer Science
• Subjects: Convolutional Neural Network; Domain Adaptation; Source Domain; Target Domain; Transfer Learning
• ISBN: 9783319590493; 3319590499
• ISSN: 0302-9743; 1611-3349
• DOI: 10.1007/978-3-319-59050-9_47

Significant advances have been made towards building accurate automatic segmentation systems for a variety of biomedical applications using machine learning. However, the performance of these systems often degrades when they are applied on new data that differ from the training data, for example, due to variations in imaging protocols. Manually annotating new data for each test domain is not a feasible solution. In this work we investigate unsupervised domain adaptation using adversarial neural networks to train a segmentation method which is more robust to differences in the input data, and which does not require any annotations on the test domain. Specifically, we derive domain-invariant features by learning to counter an adversarial network, which attempts to classify the domain of the input data by observing the activations of the segmentation network. Furthermore, we propose a multi-connected domain discriminator for improved adversarial training. Our system is evaluated using two MR databases of subjects with traumatic brain injuries, acquired using different scanners and imaging protocols. Using our unsupervised approach, we obtain segmentation accuracies which are close to the upper bound of supervised domain adaptation.