mlsanityData to 14 Oct 2026

Scalable Learning from Probability Measures with Mean Measure Quantization

arXiv 2502.04907 ↗First posted 7 Feb 2025stat.ML cs.LGcited by 1

Not on the map. Why?

Abstract (arXiv)

We consider statistical learning problems in which data are observed as a set of probability measures. Optimal transport (OT) is a popular tool to compare and manipulate such objects, but its computational cost becomes prohibitive when the measures have large support. We study a quantization-based approach in which all input measures are approximated by K-point discrete measures sharing a common support. We establish consistency of the resulting quantized measures. We further derive convergence guarantees for several OT-based downstream tasks computed from the quantized measures. Numerical experiments on synthetic and real datasets demonstrate that the proposed approach achieves performance comparable to individual quantization while substantially reducing runtime.

Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 11 of its 34 references that are in the map. First posted on arXiv 7 Feb 2025.

Builds on

11 of 34 references.

References · 11

  1. 2016Kernel Mean Embedding of Distributions: A Review and Beyondcited by 364
  2. 2013Fast Computation of Wasserstein Barycenterscited by 216
  3. 2016Coresets for Scalable Bayesian Logistic Regressioncited by 79
  4. 2023Recent Advances in Optimal Transport for Machine Learningcited by 66
  5. 2020Linear Optimal Transport Embedding: Provable Wasserstein classification for certain rigid transformations and perturbationscited by 37
  6. 2023Scalable Optimal Transport Methods in Machine Learning: A Contemporary Surveycited by 27
  7. 2015Principal Geodesic Analysis for Probability Measures under the Optimal Transport Metriccited by 15
  8. 2021Controlling Wasserstein Distances by Kernel Norms with Application to Compressive Statistical Learningcited by 14
  9. 2018Wasserstein Measure Coresetscited by 11
  10. 2021Improving Approximate Optimal Transport Distances using Quantizationcited by 5
  11. 2024Low dimensional representation of multi-patient flow cytometry datasets using optimal transport for minimal residual disease detection in leukemiacited by 1

Built on by

Cited by year

Citing papers per year: 2025: 1.

1 paper.

Similar, not cited

Close in wording, no citation either way

  1. 2026Sparse Dictionary Learning for Measure-Valued Data via Linearized Optimal Transport: Efficient Algorithms and Sample Complexity
  2. 2025Quantization-based Bounds on the Wasserstein Metric
  3. 2026Barron Optimal Transport I: Generative Modeling
  4. 2021On learning parametric distributions from quantized samples
  5. 2022Exploiting the Partly Scratch-off Lottery Ticket for Quantization-Aware Training
  6. 2022Convergence Rates for Regularized Optimal Transport via Quantization
  7. 2025Dataset Distillation as Pushforward Optimal Quantization
  8. 2024QGen: On the Ability to Generalize in Quantization Aware Training