Sliced-Wasserstein Estimation with Spherical Harmonics as Control Variates

Archive ouverte

Leluc, Rémi | Dieuleveut, Aymeric | Portier, François | Segers, Johan | Zhuman, Aigerim

Edité par CCSD -

The Sliced-Wasserstein (SW) distance between probability measures is defined as the average of the Wasserstein distances resulting for the associated one-dimensional projections. As a consequence, the SW distance can be written as an integral with respect to the uniform measure on the sphere and the Monte Carlo framework can be employed for calculating the SW distance. Spherical harmonics are polynomials on the sphere that form an orthonormal basis of the set of square-integrable functions on the sphere. Putting these two facts together, a new Monte Carlo method, hereby referred to as Spherical Harmonics Control Variates (SHCV), is proposed for approximating the SW distance using spherical harmonics as control variates. The resulting approach is shown to have good theoretical properties, e.g., a no-error property for Gaussian measures under a certain form of linear dependency between the variables. Moreover, an improved rate of convergence, compared to Monte Carlo, is established for general measures. The convergence analysis relies on the Lipschitz property associated to the SW integrand. Several numerical experiments demonstrate the superior performance of SHCV against state-of-the-art methods for SW distance computation.

Suggestions

Du même auteur

A Quadrature Rule combining Control Variates and Adaptive Importance Sampling

Archive ouverte | Leluc, Rémi | CCSD

International audience. Driven by several successful applications such as in stochastic gradient descent or in Bayesian computation, control variates have become a major tool for Monte Carlo integration. However, st...

SGD with Coordinate Sampling: Theory and Practice

Archive ouverte | Leluc, Rémi | CCSD

Journal of Machine Learning Research 2022. International audience. While classical forms of stochastic gradient descent algorithm treat the different coordinates in the same way, a framework allowing for adaptive (n...

On the weak convergence of the empirical conditional copula under a simplifying assumption

Archive ouverte | Portier, François | CCSD

Chargement des enrichissements...