Timezone: »
Poster
Statistical Optimal Transport posed as Learning Kernel Embedding
Saketha Nath Jagarlapudi · Pratik Kumar Jawanpuria
The objective in statistical Optimal Transport (OT) is to consistently estimate the optimal transport plan/map solely using samples from the given source and target marginal distributions. This work takes the novel approach of posing statistical OT as that of learning the transport plan's kernel mean embedding from sample based estimates of marginal embeddings. The proposed estimator controls overfitting by employing maximum mean discrepancy based regularization, which is complementary to $\phi$-divergence (entropy) based regularization popularly employed in existing estimators. A key result is that, under very mild conditions, $\epsilon$-optimal recovery of the transport plan as well as the Barycentric-projection based transport map is possible with a sample complexity that is completely dimension-free. Moreover, the implicit smoothing in the kernel mean embeddings enables out-of-sample estimation. An appropriate representer theorem is proved leading to a kernelized convex formulation for the estimator, which can then be potentially used to perform OT even in non-standard domains. Empirical results illustrate the efficacy of the proposed approach.
Author Information
Saketha Nath Jagarlapudi (IIT Hyderabad)
Pratik Kumar Jawanpuria (Microsoft)
More from the Same Authors
-
2022 : Rieoptax: Riemannian Optimization in JAX »
Saiteja Utpala · Andi Han · Pratik Kumar Jawanpuria · Bamdev Mishra -
2021 Poster: On Riemannian Optimization over Positive Definite Matrices with the Bures-Wasserstein Geometry »
Andi Han · Bamdev Mishra · Pratik Kumar Jawanpuria · Junbin Gao -
2020 : Poster Session 1 (gather.town) »
Laurent Condat · Tiffany Vlaar · Ohad Shamir · Mohammadi Zaki · Zhize Li · Guan-Horng Liu · Samuel Horváth · Mher Safaryan · Yoni Choukroun · Kumar Shridhar · Nabil Kahale · Jikai Jin · Pratik Kumar Jawanpuria · Gaurav Kumar Yadav · Kazuki Koyama · Junyoung Kim · Xiao Li · Saugata Purkayastha · Adil Salim · Dighanchal Banerjee · Peter Richtarik · Lakshman Mahto · Tian Ye · Bamdev Mishra · Huikang Liu · Jiajie Zhu -
2018 Poster: A Dual Framework for Low-rank Tensor Completion »
Madhav Nimishakavi · Pratik Kumar Jawanpuria · Bamdev Mishra -
2015 Poster: Efficient Output Kernel Learning for Multiple Tasks »
Pratik Kumar Jawanpuria · Maksim Lapin · Matthias Hein · Bernt Schiele