Sliced-Wasserstein Estimation with Spherical Harmonics as Control Variates

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Leluc, Rémi, Dieuleveut, Aymeric, Portier, François, Segers, Johan, Zhuman, Aigerim
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866909202717343744
author Leluc, Rémi
Dieuleveut, Aymeric
Portier, François
Segers, Johan
Zhuman, Aigerim
author_facet Leluc, Rémi
Dieuleveut, Aymeric
Portier, François
Segers, Johan
Zhuman, Aigerim
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2402_01493
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Sliced-Wasserstein Estimation with Spherical Harmonics as Control Variates
Leluc, Rémi
Dieuleveut, Aymeric
Portier, François
Segers, Johan
Zhuman, Aigerim
Machine Learning
65C05 (Primary) 65D30, 68Txx, 68Wxx (Secondary)
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.
title Sliced-Wasserstein Estimation with Spherical Harmonics as Control Variates
topic Machine Learning
65C05 (Primary) 65D30, 68Txx, 68Wxx (Secondary)
url https://arxiv.org/abs/2402.01493