Abstract
We study the use of amortized optimization to predict optimal transport (OT) maps from the input measures, which we call Meta OT. This helps repeatedly solve similar OT problems between different measures by leveraging the knowledge and information present from past problems to rapidly predict and solve new problems. Otherwise, standard methods ignore the knowledge of the past solutions and sub-optimally re-solve each problem from scratch. We instantiate Meta OT models in discrete and continuous settings between grayscale images, spherical data, classification labels, and color palettes and use them to improve the computational time of standard OT solvers. Our source code is available at http://github.com/facebookresearch/meta-ot.
| Original language | English |
|---|---|
| Pages (from-to) | 791-813 |
| Number of pages | 23 |
| Journal | Proceedings of Machine Learning Research |
| Volume | 202 |
| Publication status | Published - 2023 |
| MoE publication type | A4 Conference publication |
| Event | International Conference on Machine Learning - Honolulu, United States Duration: 23 Jul 2023 → 29 Jul 2023 Conference number: 40 |