Sample Complexity Bounds for Estimating Probability Divergences under Invariances
Not on the map. Why?
Abstract (arXiv)
Group-invariant probability distributions appear in many data-generative models in machine learning, such as graphs, point clouds, and images. In practice, one often needs to estimate divergences between such distributions. In this work, we study how the inherent invariances, with respect to any smooth action of a Lie group on a manifold, improve sample complexity when estimating the 1-Wasserstein distance, the Sobolev Integral Probability Metrics (Sobolev IPMs), the Maximum Mean Discrepancy (MMD), and also the complexity of the density estimation problem (in the L2 and L^∞ distance). Our results indicate a two-fold gain: (1) reducing the sample complexity by a multiplicative factor corresponding to the group size (for finite groups) or the normalized volume of the quotient space (for groups of positive dimension); (2) improving the exponent in the convergence rate (for groups of positive dimension). These results are completely new for groups of positive dimension and extend recent bounds for finite group actions.
Key facts. Cited by 11 papers in the map (data through 7 Oct 2026). Builds on 23 of its 56 references that are in the map. First posted on arXiv 6 Nov 2023.
Builds on
23 of 56 references.
References · 23
- 2022Generative Adversarial Networkscited by 5,603
- 2015Optimal Transport for Domain Adaptationcited by 457
- 2019Cormorant: Covariant Molecular Neural Networkscited by 262
- 2013Fast Computation of Wasserstein Barycenterscited by 216
- 2018Statistical Aspects of Wasserstein Distancescited by 202
- 2018Improving GANs Using Optimal Transportcited by 122
- 2019Statistical bounds for entropic optimal transport: sample complexity and the central limit theoremcited by 85
- 2018How Well Generative Adversarial Networks Learn Distributionscited by 78
- 2019Equivariant Flows: sampling configurations for multi-body systems with symmetric energiescited by 57
- 2018Convergence and Concentration of Empirical Measures under Wasserstein Distance in Unbounded Functional Spacescited by 41
- 2017Sobolev GANcited by 29
- 2023The Exact Sample Complexity Gain from Invariances for Kernel Regressioncited by 27
11 more
- 2018Nonparametric Density Estimation under Adversarial Lossescited by 26
- 2021Measuring Generalization with Optimal Transportcited by 24
- 2019Nonparametric Density Estimation & Convergence Rates for GANs under Besov IPM Lossescited by 22
- 2020Scalable Normalizing Flows for Permutation Invariant Densitiescited by 22
- 2021Intrinsic Dimension Estimation Using Wasserstein Distancescited by 20
- 2020Group Equivariant Generative Adversarial Networkscited by 19
- 2022Structure-preserving GANscited by 17
- 2023Sample Complexity of Probability Divergences under Group Symmetrycited by 13
- 2019Estimating Certain Integral Probability Metric (IPM) is as Hard as Estimating under the IPMcited by 11
- 2019(q,p)-Wasserstein GANs: Comparing Ground Metrics for Wasserstein GANscited by 10
- 2020A Review on Modern Computational Optimal Transport Methods with Applications in Biomedical Researchcited by 8
Built on by
Citing papers per year: 2024: 3; 2025: 3; 2026: 5.
11 papers.
- 2024SemlaFlow -- Efficient 3D Molecular Generation with Latent Attention and Equivariant Flow Matchingcited by 40
- 2025Achieving Approximate Symmetry Is Exponentially Easier than Exact Symmetrycited by 6
- 2025Learning with Exact Invariances in Polynomial Timecited by 5
- 2024Robust Generative Learning with Lipschitz-Regularized α-Divergences Allows Minimal Assumptions on Target Distributionscited by 4
- 2024Robustness and Structure Preservation in Flow-Based Generative Models via Wasserstein Path-Space Divergencescited by 3
- 2026Error Analysis of Bayesian Inverse Problems with Generative Priorscited by 2
- 2026Adaptive Symmetry Discovery for Dynamical System Identificationcited by 2
- 2026Efficient Learning and Symmetry Discovery under Exact Invariancescited by 2
- 2025Statistical Learning Guarantees for Group-Invariant Barron Functionscited by 1
- 2026GEPC: Group-Equivariant Posterior Consistency for Out-of-Distribution Detection in Diffusion Modelscited by 1
- 2026Sparse Data Augmentation for Optimization with Provable Guaranteescited by 0
- 2026Sparse Data Augmentation for Optimization with Provable Guarantees8 Sep 2026
- 2026Efficient Learning and Symmetry Discovery under Exact Invariances7 Sep 2026
- 2026Adaptive Symmetry Discovery for Dynamical System Identification8 Aug 2026
- 2026GEPC: Group-Equivariant Posterior Consistency for Out-of-Distribution Detection in Diffusion Models30 Jan 2026
- 2026Error Analysis of Bayesian Inverse Problems with Generative Priors24 Jan 2026
- 2025Achieving Approximate Symmetry Is Exponentially Easier than Exact Symmetry5 Dec 2025
- 2025Statistical Learning Guarantees for Group-Invariant Barron Functions27 Sep 2025
- 2025Learning with Exact Invariances in Polynomial Time27 Feb 2025
- 2024Robustness and Structure Preservation in Flow-Based Generative Models via Wasserstein Path-Space Divergences2 Oct 2024
- 2024SemlaFlow -- Efficient 3D Molecular Generation with Latent Attention and Equivariant Flow Matching11 Jun 2024
- 2024Robust Generative Learning with Lipschitz-Regularized α-Divergences Allows Minimal Assumptions on Target Distributions22 May 2024