Efficient Transferable Optimal Transport via Min-Sliced Transport Plans

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Liu, Xinran, Akbari, Elaheh, Martin, Rocio Diaz, NaderiAlizadeh, Navid, Kolouri, Soheil
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866910255979429888
author Liu, Xinran
Akbari, Elaheh
Martin, Rocio Diaz
NaderiAlizadeh, Navid
Kolouri, Soheil
author_facet Liu, Xinran
Akbari, Elaheh
Martin, Rocio Diaz
NaderiAlizadeh, Navid
Kolouri, Soheil
contents Optimal Transport (OT) offers a powerful framework for finding correspondences between distributions and addressing matching and alignment problems in various areas of computer vision, including shape analysis, image generation, and multimodal tasks. The computation cost of OT, however, hinders its scalability. Slice-based transport plans have recently shown promise for reducing the computational cost by leveraging the closed-form solutions of 1D OT problems. These methods optimize a one-dimensional projection (slice) to obtain a conditional transport plan that minimizes the transport cost in the ambient space. While efficient, these methods leave open the question of whether learned optimal slicers can transfer to new distribution pairs under distributional shift. Understanding this transferability is crucial in settings with evolving data or repeated OT computations across closely related distributions. In this paper, we study the min-Sliced Transport Plan (min-STP) framework and investigate the transferability of optimized slicers: can a slicer trained on one distribution pair yield effective transport plans for new, unseen pairs? Theoretically, we show that optimized slicers remain close under slight perturbations of the data distributions, enabling efficient transfer across related tasks. To further improve scalability, we introduce a minibatch formulation of min-STP and provide statistical guarantees on its accuracy. Empirically, we demonstrate that the transferable min-STP achieves strong one-shot matching performance and facilitates amortized training for point cloud alignment and flow-based generative modeling.
format Preprint
id arxiv_https___arxiv_org_abs_2511_19741
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Efficient Transferable Optimal Transport via Min-Sliced Transport Plans
Liu, Xinran
Akbari, Elaheh
Martin, Rocio Diaz
NaderiAlizadeh, Navid
Kolouri, Soheil
Computer Vision and Pattern Recognition
Optimal Transport (OT) offers a powerful framework for finding correspondences between distributions and addressing matching and alignment problems in various areas of computer vision, including shape analysis, image generation, and multimodal tasks. The computation cost of OT, however, hinders its scalability. Slice-based transport plans have recently shown promise for reducing the computational cost by leveraging the closed-form solutions of 1D OT problems. These methods optimize a one-dimensional projection (slice) to obtain a conditional transport plan that minimizes the transport cost in the ambient space. While efficient, these methods leave open the question of whether learned optimal slicers can transfer to new distribution pairs under distributional shift. Understanding this transferability is crucial in settings with evolving data or repeated OT computations across closely related distributions. In this paper, we study the min-Sliced Transport Plan (min-STP) framework and investigate the transferability of optimized slicers: can a slicer trained on one distribution pair yield effective transport plans for new, unseen pairs? Theoretically, we show that optimized slicers remain close under slight perturbations of the data distributions, enabling efficient transfer across related tasks. To further improve scalability, we introduce a minibatch formulation of min-STP and provide statistical guarantees on its accuracy. Empirically, we demonstrate that the transferable min-STP achieves strong one-shot matching performance and facilitates amortized training for point cloud alignment and flow-based generative modeling.
title Efficient Transferable Optimal Transport via Min-Sliced Transport Plans
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2511.19741