HOUDARD, Antoine; LECLAIRE, Arthur; PAPADAKIS, Nicolas; RABIN, Julien

Metadatos

Mostrar el registro completo del ítem

Licencia de uso del documento

HOUDARD, Antoine
Institut de Mathématiques de Bordeaux [IMB]

LECLAIRE, Arthur
Institut de Mathématiques de Bordeaux [IMB]

PAPADAKIS, Nicolas
Institut de Mathématiques de Bordeaux [IMB]

Idioma

Article de revue

Este ítem está publicado en

Transactions on Machine Learning Research Journal. 2023-07-14

[Amherst Massachusetts]: OpenReview.net, 2022

Resumen en inglés

The use of optimal transport costs for learning generative models has become popular with Wasserstein Generative Adversarial Networks (WGANs). Training a WGAN requires the computation of the differentiation of the optimal transport cost with respect to the parameters of the generative model. In this work, we provide sufficient conditions for the existence of a gradient formula in two different frameworks: the case of semi-discrete optimal transport (i.e. with a discrete target distribution) and the case of regularized optimal transport (i.e. with an entropic penalty). Both cases are based on the dual formulation of the transport cost, and the gradient formula involves a solution of the dual problem. The learning problem is addressed with an alternate algorithm, whose behavior is examined for the problem of MNIST digits generation. In particular, we analyze the impact of entropic regularization both on visual results and convergence speed.< Leer menos

URI

https://oskar-bordeaux.fr/handle/20.500.12278/191093

Proyecto ANR

Models, Inference and Synthesis for Texture In Color - ANR-19-CE40-0005
Generalized Optimal Transport Models for Image processing - ANR-16-CE33-0010

Orígen

Importado de Hal

Centros de investigación

Institut de Mathématiques de Bordeaux (IMB) - UMR 5251