Learning to solve inverse problems using Wasserstein loss

Author name not available (Why is that?)

Publication date: 30 October 2017

Abstract: We propose using the Wasserstein loss for training in inverse problems. In particular, we consider a learned primal-dual reconstruction scheme for ill-posed inverse problems using the Wasserstein distance as loss function in the learning. This is motivated by miss-alignments in training data, which when using standard mean squared error loss could severely degrade reconstruction quality. We prove that training with the Wasserstein loss gives a reconstruction operator that correctly compensates for miss-alignments in certain cases, whereas training with the mean squared error gives a smeared reconstruction. Moreover, we demonstrate these effects by training a reconstruction algorithm using both mean squared error and optimal transport loss for a problem in computerized tomography.

Has companion code repository: https://github.com/odlgroup/odl

This page was built for publication: Learning to solve inverse problems using Wasserstein loss

Report a bug (only for logged in users!)Click here to report a bug for this page (MaRDI item Q6293235)