Spherical Sliced-Wasserstein
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:
density estimation on the sphere, variational inference or hyperspherical
auto-encoders.