Shedding a PAC-Bayesian Light on Adaptive Sliced-Wasserstein Distances
Not on the map. Why?
Abstract (arXiv)
The Sliced-Wasserstein distance (SW) is a computationally efficient and theoretically grounded alternative to the Wasserstein distance. Yet, the literature on its statistical properties -- or, more accurately, its generalization properties -- with respect to the distribution of slices, beyond the uniform measure, is scarce. To bring new contributions to this line of research, we leverage the PAC-Bayesian theory and a central observation that SW may be interpreted as an average risk, the quantity PAC-Bayesian bounds have been designed to characterize. We provide three types of results: i) PAC-Bayesian generalization bounds that hold on what we refer as adaptive Sliced-Wasserstein distances, i.e. SW defined with respect to arbitrary distributions of slices (among which data-dependent distributions), ii) a principled procedure to learn the distribution of slices that yields maximally discriminative SW, by optimizing our theoretical bounds, and iii) empirical illustrations of our theoretical findings.
Key facts. Cited by 7 papers in the map (data through 7 Oct 2026). Builds on 41 of its 74 references that are in the map. First posted on arXiv 7 Jun 2022.
Builds on
41 of 74 references.
References · 41
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 2017Fashion-MNIST: a Novel Image Dataset for Benchmarking Machine Learning Algorithmscited by 4,283
- 2013Sinkhorn Distances: Lightspeed Computation of Optimal Transportation Distancescited by 2,277
- 2015Weight Uncertainty in Neural Networkscited by 554
- 2017Computing Nonvacuous Generalization Bounds for Deep (Stochastic) Neural Networks with Many More Parameters than Training Datacited by 527
- 2015Optimal Transport for Domain Adaptationcited by 457
- 2021User-friendly introduction to PAC-Bayes boundscited by 234
- 2019Generalized Sliced Wasserstein Distancescited by 205
- 2015Learning with a Wasserstein Losscited by 194
- 2018Hyperspherical Variational Auto-Encoderscited by 188
- 2019Max-Sliced Wasserstein Distance and its use for GANscited by 118
- 2018Generative Modeling using the Sliced Wasserstein Distancecited by 108
29 more
- 2019Subspace Robust Wasserstein Distancescited by 89
- 2019Statistical bounds for entropic optimal transport: sample complexity and the central limit theoremcited by 85
- 2020Statistical and Topological Properties of Sliced Probability Divergencescited by 83
- 2017Sliced Wasserstein Kernel for Persistence Diagramscited by 77
- 2018Sliced-Wasserstein Flows: Nonparametric Generative Modeling via Optimal Transport and Diffusionscited by 77
- 2020Distributional Sliced-Wasserstein and Applications to Generative Modelingcited by 69
- 2015Sliced Wasserstein Kernels for Probability Distributionscited by 64
- 2017Sliced Wasserstein Distance for Learning Gaussian Mixture Modelscited by 59
- 2016Simpler PAC-Bayesian Bounds for Hostile Datacited by 45
- 2017von Mises-Fisher Mixture Model-based Deep learning: Application to Face Verificationcited by 45
- 2019Asymptotic Guarantees for Learning Generative Models with the Sliced-Wasserstein Distancecited by 43
- 2020PAC-Bayes unleashed: generalisation bounds with unbounded lossescited by 43
- 2021von Mises-Fisher Loss: An Exploration of Embedding Geometries for Supervised Learningcited by 42
- 2017From optimal transport to generative modeling: the VEGAN cookbookcited by 41
- 2018Convergence and Concentration of Empirical Measures under Wasserstein Distance in Unbounded Functional Spacescited by 41
- 2017Wasserstein Dictionary Learning: Optimal Transport-based unsupervised non-linear dictionary learningcited by 34
- 2020On Projection Robust Optimal Transport: Sample Complexity and Model Misspecificationcited by 34
- 2021Fast Approximation of the Sliced-Wasserstein Distance Using Concentration of Random Projectionscited by 33
- 2015A Smoothed Dual Approach for Variational Wasserstein Problemscited by 29
- 2018Von Mises-Fisher Loss for Training Sequence to Sequence Models with Continuous Outputscited by 26
- 2019Minimax Confidence Intervals for the Sliced Wasserstein Distancecited by 24
- 2019Approximate Bayesian Computation with the Sliced-Wasserstein Distancecited by 22
- 2013Concentration in unbounded metric spaces and algorithmic stabilitycited by 21
- 2022On PAC-Bayesian reconstruction guarantees for VAEscited by 20
- 2021Differentially Private Sliced Wasserstein Distancecited by 19
- 2015A Sparse Multi-Scale Algorithm for Dense Optimal Transportcited by 16
- 2021Learning Stochastic Majority Votes by Minimizing a PAC-Bayes Generalization Boundcited by 16
- 2021Reliable Estimation of KL Divergence using a Discriminator in Reproducing Kernel Hilbert Spacecited by 7
- 2020Generalized Sliced Distances for Probability Distributionscited by 2
Built on by
Citing papers per year: 2022: 1; 2023: 3; 2024: 1; 2025: 1; 2026: 1.
7 papers.
- 2023Energy-Based Sliced Wasserstein Distancecited by 39
- 2022Hyperbolic Sliced-Wasserstein via Geodesic and Horospherical Projectionscited by 16
- 2023PAC-Bayesian Generalization Bounds for Adversarial Generative Modelscited by 13
- 2023Wasserstein PAC-Bayes Learning: Exploiting Optimisation Guarantees to Explain Generalisationcited by 3
- 2024Towards Better Spherical Sliced-Wasserstein Distance Learning with Data-Adaptive Discriminative Projection Directioncited by 1
- 2025Minimax-Optimal Two-Sample Test with Sliced Wassersteincited by 1
- 2026Highly Data Parallelizable Estimation of the Sliced-Wasserstein Distance Using Cumulative Distribution Functionscited by 0
- 2026Highly Data Parallelizable Estimation of the Sliced-Wasserstein Distance Using Cumulative Distribution Functions29 Jun 2026
- 2025Minimax-Optimal Two-Sample Test with Sliced Wasserstein31 Oct 2025
- 2024Towards Better Spherical Sliced-Wasserstein Distance Learning with Data-Adaptive Discriminative Projection Direction26 Dec 2024
- 2023Energy-Based Sliced Wasserstein Distance26 Apr 2023
- 2023Wasserstein PAC-Bayes Learning: Exploiting Optimisation Guarantees to Explain Generalisation14 Apr 2023
- 2023PAC-Bayesian Generalization Bounds for Adversarial Generative Models17 Feb 2023
- 2022Hyperbolic Sliced-Wasserstein via Geodesic and Horospherical Projections18 Nov 2022