Properties of Discrete Sliced Wasserstein Losses - CNRS - Centre national de la recherche scientifique
Article Dans Une Revue Mathematics of Computation Année : 2023

Properties of Discrete Sliced Wasserstein Losses

Résumé

The Sliced Wasserstein (SW) distance has become a popular alternative to the Wasserstein distance for comparing probability measures. Widespread applications include image processing, domain adaptation and generative modelling, where it is common to optimise some parameters in order to minimise SW, which serves as a loss function between discrete probability measures (since measures admitting densities are numerically unattainable). All these optimisation problems bear the same sub-problem, which is minimising the Sliced Wasserstein energy. In this paper we study the properties of E : Y −→ SW 2 2 (γ Y , γ Z), i.e. the SW distance between two uniform discrete measures with the same amount of points as a function of the support Y ∈ R n×d of one of the measures. We investigate the regularity and optimisation properties of this energy, as well as its Monte-Carlo approximation E p (estimating the expectation in SW using only p samples) and show convergence results on the critical points of E p to those of E, as well as an almost-sure uniform convergence. Finally, we show that in a certain sense, Stochastic Gradient Descent methods minimising E and E p converge towards (Clarke) critical points of these energies.
Fichier principal
Vignette du fichier
main.pdf (10.36 Mo) Télécharger le fichier
Origine Fichiers produits par l'(les) auteur(s)

Dates et versions

hal-04232766 , version 1 (09-10-2023)
hal-04232766 , version 2 (18-03-2024)
hal-04232766 , version 3 (23-09-2024)

Identifiants

Citer

Eloi Tanguy, Rémi Flamary, Julie Delon. Properties of Discrete Sliced Wasserstein Losses. Mathematics of Computation, 2023, ⟨https://doi.org/10.1090/mcom/3994⟩. ⟨hal-04232766v3⟩
87 Consultations
57 Téléchargements

Altmetric

Partager

More