Representation disentanglement for multi-task learning with application to fetal ultrasound
File(s)1908.07885.pdf (1.02 MB)
Accepted version
OA Location
Author(s)
Meng, Qingjie
Pawlowski, Nick
Rueckert, Daniel
Kainz, Bernhard
Type
Conference Paper
Abstract
One of the biggest challenges for deep learning algorithms in medical image analysis is the indiscriminate mixing of image properties, e.g. artifacts and anatomy. These entangled image properties lead to a semantically redundant feature encoding for the relevant task and thus lead to poor generalization of deep learning algorithms. In this paper we propose a novel representation disentanglement method to extract semantically meaningful and generalizable features for different tasks within a multi-task learning framework. Deep neural networks are utilized to ensure that the encoded features are maximally informative with respect to relevant tasks, while an adversarial regularization encourages these features to be disentangled and minimally informative about irrelevant tasks. We aim to use the disentangled representations to generalize the applicability of deep neural networks. We demonstrate the advantages of the proposed method on synthetic data as well as fetal ultrasound images. Our experiments illustrate that our method is capable of learning disentangled internal representations. It outperforms baseline methods in multiple tasks, especially on images with new properties, e.g. previously unseen artifacts in fetal ultrasound.
Date Issued
2019-10-01
Date Acceptance
2019-08-20
Citation
PIPPI 2019, SUSI 2019: Smart Ultrasound Imaging and Perinatal, Preterm and Paediatric Image Analysis, 2019, 11798 LNCS, pp.47-55
ISBN
9783030328740
ISSN
0302-9743
Publisher
Springer International Publishing
Start Page
47
End Page
55
Journal / Book Title
PIPPI 2019, SUSI 2019: Smart Ultrasound Imaging and Perinatal, Preterm and Paediatric Image Analysis
Volume
11798 LNCS
Copyright Statement
© Springer Nature Switzerland AG 2019. The final publication is available at Springer via https://link.springer.com/chapter/10.1007%2F978-3-030-32875-7_6
Source
Smart Ultrasound Imaging and Perinatal, Preterm and Paediatric Image Analysis
Subjects
Artificial Intelligence & Image Processing
Publication Status
Published
Start Date
2019-10-13
Coverage Spatial
Shenzhen, China
Date Publish Online
2019-10-08