Skip to yearly menu bar Skip to main content


Mapping Estimation for Discrete Optimal Transport

Michaël Perrot · Nicolas Courty · Rémi Flamary · Amaury Habrard

Area 5+6+7+8 #137

Keywords: [ Multi-task and Transfer Learning ] [ (Other) Machine Learning Topics ] [ (Other) Regression ] [ (Application) Computer Vision ]

Abstract: We are interested in the computation of the transport map of an Optimal Transport problem. Most of the computational approaches of Optimal Transport use the Kantorovich relaxation of the problem to learn a probabilistic coupling $\mgamma$ but do not address the problem of learning the underlying transport map $\funcT$ linked to the original Monge problem. Consequently, it lowers the potential usage of such methods in contexts where out-of-samples computations are mandatory. In this paper we propose a new way to jointly learn the coupling and an approximation of the transport map. We use a jointly convex formulation which can be efficiently optimized. Additionally, jointly learning the coupling and the transport map allows to smooth the result of the Optimal Transport and generalize it to out-of-samples examples. Empirically, we show the interest and the relevance of our method in two tasks: domain adaptation and image editing.

Live content is unavailable. Log in and register to view live content