Statistical, Robustness, and Computational Guarantees for Sliced Wasserstein Distances
Sloan Nietert, Ziv Goldfeld, Ritwik Sadhu, Kengo Kato
Abstract
Sliced Wasserstein distances preserve properties of classic Wasserstein distances while being more scalable for computation and estimation in high dimensions. The goal of this work is to quantify this scalability from three key aspects: (i) empirical convergence rates; (ii) robustness to data contamination; and (iii) efficient computational methods. For empirical convergence, we derive fast rates with explicit dependence of constants on dimension, subject to log-concavity of the population distributions. For robustness, we characterize minimax optimal, dimension-free robust estimation risks, and show an equivalence between robust sliced 1-Wasserstein estimation and robust mean estimation. This enables lifting statistical and algorithmic guarantees available for the latter to the sliced 1-Wasserstein setting. Moving on to computational aspects, we analyze the Monte Carlo estimator for the average-sliced distance, demonstrating that larger dimension can result in faster convergence of the numerical integration error. For the max-sliced distance, we focus on a subgradient-based local optimization algorithm that is frequently used in practice, albeit without formal guarantees, and establish an computational complexity bound for it. Our theory is validated by numerical experiments, which altogether provide a comprehensive quantitative account of the scalability question.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers21
- Energy-Based Sliced Wasserstein DistanceKhai Nguyen, Nhat HoNeurIPS 2023 · 51 citations
- Max-Sliced Mutual InformationDor Tsur, Ziv Goldfeld, Kristjan H. GreenewaldNeurIPS 2023 · 20 citations
- Sliced Wasserstein with Random-Path Projecting DirectionsKhai Nguyen, Shujian Zhang, Tam Le, Nhat HoICML 2024 · 17 citations
- Deep Structural Knowledge Exploitation and Synergy for Estimating Node Importance Value on Heterogeneous Information NetworksYankai Chen, Yixiang Fang, Qiongyan Wang, Xin Cao et al.AAAI 2024 · 17 citations
- Markovian Sliced Wasserstein Distances: Beyond Independent ProjectionsKhai Nguyen, Tongzheng Ren, Nhat HoNeurIPS 2023 · 13 citations
Builds on9
- Robust Optimal Transport with Applications in Generative Modeling and Domain AdaptationYogesh Balaji, Rama Chellappa, Soheil FeiziNeurIPS 2020 · 141 citations
- Statistical and Topological Properties of Sliced Probability DivergencesKimia Nadjahi, Alain Durmus, Lénaïc Chizat, Soheil Kolouri et al.NeurIPS 2020 · 115 citations
- Projection Robust Wasserstein Distance and Riemannian OptimizationTianyi Lin, Chenyou Fan, Nhat Ho, Marco Cuturi et al.NeurIPS 2020 · 84 citations
- Robust and Heavy-Tailed Mean Estimation Made Simple, via Regret MinimizationSamuel B. Hopkins, Jerry Li, Fred ZhangNeurIPS 2020 · 74 citations
- Outlier-Robust Optimal TransportDebarghya Mukherjee, Aritra Guha, Justin M. Solomon, Yuekai Sun et al.ICML 2021 · 57 citations
Related papers
- Fast Approximation of the Sliced-Wasserstein Distance Using Concentration of Random ProjectionsKimia Nadjahi, Alain Durmus, Pierre E. Jacob, Roland Badeau et al.NeurIPS 2021 · 54 citations
- Smooth p-Wasserstein Distance: Structure, Empirical Approximation, and Statistical ApplicationsSloan Nietert, Ziv Goldfeld, Kengo KatoICML 2021 · 39 citations
- Shedding a PAC-Bayesian Light on Adaptive Sliced-Wasserstein DistancesRuben Ohana, Kimia Nadjahi, Alain Rakotomamonjy, Liva RalaivolaICML 2023 · 7 citations
- Streaming Sliced Optimal TransportKhai NguyenICML 2026
- Distributional Sliced-Wasserstein and Applications to Generative ModelingKhai Nguyen, Nhat Ho, Tung Pham, Hung BuiICLR 2021 · 111 citations
