Online Sinkhorn: Optimal Transport distances from sample streams
–Neural Information Processing Systems
Optimal Transport (OT) distances are now routinely used as loss functions in ML tasks. This paper introduces a new online estimator of entropy-regularized OT distances between two such arbitrary distributions. It uses streams of samples from both distributions to iteratively enrich a non-parametric representation of the transportation plan. Compared to the classic Sinkhorn algorithm, our method leverages new samples at each iteration, which enables a consistent estimation of the true regularized OT distance. We provide a theoretical analysis of the convergence of the online Sinkhorn algorithm, showing a nearly-1/n asymptotic sample complexity for the iterate sequence.
Neural Information Processing Systems
Oct-9-2024, 13:36:34 GMT
- Technology: