Learning optimal transport between two empirical distributions with normalizing flows - Signal et Communications Accéder directement au contenu
Communication Dans Un Congrès Année : 2022

Learning optimal transport between two empirical distributions with normalizing flows

Résumé

Optimal transport (OT) provides effective tools for comparing and mapping probability measures. We propose to leverage the flexibility of neural networks to learn an approximate optimal transport map. More precisely, we present a new and original method to address the problem of transporting a finite set of samples associated with a first underlying unknown distribution towards another finite set of samples drawn from another unknown distribution. We show that a particular instance of invertible neural networks, namely the normalizing flows, can be used to approximate the solution of this OT problem between a pair of empirical distributions. To this aim, we propose to relax the Monge formulation of OT by replacing the equality constraint on the pushforward measure by the minimization of the corresponding Wasserstein distance. The push-forward operator to be retrieved is then restricted to be a normalizing flow which is trained by optimizing the resulting cost function. This approach allows the transport map to be discretized as a composition of functions. Each of these functions is associated to one sub-flow of the network, whose output provides intermediate steps of the transport between the original and target measures. This discretization yields also a set of intermediate barycenters between the two measures of interest. Experiments conducted on toy examples as well as a challenging task of unsupervised translation demonstrate the interest of the proposed method. Finally, some experiments show that the proposed approach leads to a good approximation of the true OT.
Fichier principal
Vignette du fichier
Coeurdoux_ECMLPKDD_2022.pdf (5.45 Mo) Télécharger le fichier
Origine : Fichiers produits par l'(les) auteur(s)

Dates et versions

hal-03713840 , version 1 (05-07-2022)

Identifiants

Citer

Florentin Coeurdoux, Nicolas Dobigeon, Pierre Chainais. Learning optimal transport between two empirical distributions with normalizing flows. European Conference on Machine Learning and Principles and Practice of Knowledge Discovery in Databases (ECML-PKDD 2022), Sep 2022, Grenoble, France. ⟨10.48550/arXiv.2207.01246⟩. ⟨hal-03713840⟩
162 Consultations
124 Téléchargements

Altmetric

Partager

Gmail Facebook X LinkedIn More