Spherical Sliced-Wasserstein - Archive ouverte HAL
Communication Dans Un Congrès Année : 2023

Spherical Sliced-Wasserstein

Résumé

Many variants of the Wasserstein distance have been introduced to reduce its original computational burden. In particular the Sliced-Wasserstein distance (SW), which leverages one-dimensional projections for which a closed-form solution of the Wasserstein distance is available, has received a lot of interest. Yet, it is restricted to data living in Euclidean spaces, while the Wasserstein distance has been studied and used recently on manifolds. We focus more specifically on the sphere, for which we define a novel SW discrepancy, which we call spherical Sliced-Wasserstein, making a first step towards defining SW discrepancies on manifolds. Our construction is notably based on closed-form solutions of the Wasserstein distance on the circle, together with a new spherical Radon transform. Along with efficient algorithms and the corresponding implementations, we illustrate its properties in several machine learning use cases where spherical representations of data are at stake: sampling on the sphere, density estimation on real eath data or hyperspherical auto-encoders.
Fichier non déposé

Dates et versions

hal-04336947 , version 1 (12-12-2023)

Identifiants

  • HAL Id : hal-04336947 , version 1

Citer

Clément Bonet, Paul Berg, Nicolas Courty, François Septier, Lucas Drumetz, et al.. Spherical Sliced-Wasserstein. International Conference on Learning Representations (ICLR), May 2023, Kigali, Rwanda. ⟨hal-04336947⟩
54 Consultations
0 Téléchargements

Partager

More