arXiv · 2210.13630
Budget-Constrained Bounds for Mini-Batch Estimation of Optimal Transport
Abstract
Optimal Transport (OT) is a fundamental tool for comparing probability distributions, but its exact computation remains prohibitive for large datasets. In this work, we introduce novel families of upper and lower bounds for the OT problem constructed by aggregating solutions of mini-batch OT problems. The upper bound family contains traditional mini-batch averaging at one extreme and a tight bound found by optimal coupling of mini-batches at the other. In between these extremes, we propose various methods to construct bounds based on a fixed computational budget. Through various experiments, we explore the trade-off between computational budget and bound tightness and show the usefulness of these bounds in computer vision applications.
Explore related subjects
Keep this discovery
David Alvarez-Melis, Nicolò Fusi, Lester Mackey, Tal Wagner. 2022-10-24. Budget-Constrained Bounds for Mini-Batch Estimation of Optimal Transport. https://arxiv.org/abs/2210.13630
Cite the original work for its findings. Save a collection to share your selection of sources.