Timezone: »
In this work we propose a batch multimarginal version of the Greenkhornalgorithm for the entropic-regularized optimal transport problem. This framework is general enough to cover, as particular cases, existing Sinkhorn and Greenkhorn algorithms for the bi-marginal setting, and greedy MultiSinkhorn for the general multimarginal case. We provide a comprehensive convergence analysis based on the properties of the iterative Bregman projections method with greedy control.Linear rate of convergence as well as explicit bounds on the iteration complexity are obtained. When specialized to the above mentioned algorithms, our results give new convergence rates or provide key improvements over the state-of-the-art rates. We present numerical experiments showing that the flexibility of the batch can be exploited to improve performance of Sinkhorn algorithm both in bi-marginal and multimarginal settings.
Author Information
Vladimir Kostic (Istituto Italiano di Tecnologia)
Saverio Salzo (Istituto Italiano di Tecnologia)
Massimiliano Pontil ( Istituto Italiano di Tecnologia & University College London)
Related Events (a corresponding poster, oral, or spotlight)
-
2022 Spotlight: Batch Greenkhorn Algorithm for Entropic-Regularized Multimarginal Optimal Transport: Linear Rate of Convergence and Iteration Complexity »
Tue. Jul 19th 03:40 -- 03:45 PM Room Room 307
More from the Same Authors
-
2022 Poster: Bregman Neural Networks »
Jordan Frecon · Gilles Gasso · Massimiliano Pontil · Saverio Salzo -
2022 Spotlight: Bregman Neural Networks »
Jordan Frecon · Gilles Gasso · Massimiliano Pontil · Saverio Salzo -
2021 Poster: Robust Unsupervised Learning via L-statistic Minimization »
Andreas Maurer · Daniela Angela Parletta · Andrea Paudice · Massimiliano Pontil -
2021 Spotlight: Robust Unsupervised Learning via L-statistic Minimization »
Andreas Maurer · Daniela Angela Parletta · Andrea Paudice · Massimiliano Pontil -
2020 Poster: On the Iteration Complexity of Hypergradient Computation »
Riccardo Grazzi · Luca Franceschi · Massimiliano Pontil · Saverio Salzo -
2018 Poster: Bilevel Programming for Hyperparameter Optimization and Meta-Learning »
Luca Franceschi · Paolo Frasconi · Saverio Salzo · Riccardo Grazzi · Massimiliano Pontil -
2018 Oral: Bilevel Programming for Hyperparameter Optimization and Meta-Learning »
Luca Franceschi · Paolo Frasconi · Saverio Salzo · Riccardo Grazzi · Massimiliano Pontil