English
Related papers

Related papers: Fully Probabilistic Design for Optimal Transport

200 papers

Optimal transport (OT) has become a widely used tool in the machine learning field to measure the discrepancy between probability distributions. For instance, OT is a popular loss function that quantifies the discrepancy between an…

Machine Learning · Computer Science 2022-12-27 Shintaro Nakamura , Han Bao , Masashi Sugiyama

Optimal transport is the problem of designing a joint distribution for two random variables with fixed marginals. In virtually the entire literature on this topic, the objective is to minimize expected cost. This paper is the first to study…

Econometrics · Economics 2026-02-13 Yinchu Zhu , Ilya O. Ryzhov

This paper studies the estimation of large-scale optimal transport maps (OTM), which is a well-known challenging problem owing to the curse of dimensionality. Existing literature approximates the large-scale OTM by a series of…

Machine Learning · Statistics 2021-06-11 Cheng Meng , Yuan Ke , Jingyi Zhang , Mengrui Zhang , Wenxuan Zhong , Ping Ma

Whilst optimal transport (OT) is increasingly being recognized as a powerful and flexible approach for dealing with fairness issues, current OT fairness methods are confined to the use of discrete OT. In this paper, we leverage recent…

Machine Learning · Computer Science 2021-01-07 Silvia Chiappa , Aldo Pacchiano

Optimal transport has recently been brought forward as a tool for modeling and efficiently solving a variety of flow problems, such as origin-destination problems and multi-commodity flow problems. Although the framework has shown to be…

Optimization and Control · Mathematics 2025-07-29 Anqi Dong , Karl Henrik Johansson , Johan Karlsson

Optimal transport has been one of the most exciting subjects in mathematics, starting from the 18th century. As a powerful tool to transport between two probability measures, optimal transport methods have been reinvigorated nowadays in a…

Machine Learning · Statistics 2021-05-21 Jingyi Zhang , Wenxuan Zhong , Ping Ma

With the widespread application of optimal transport (OT), its calculation becomes essential, and various algorithms have emerged. However, the existing methods either have low efficiency or cannot represent discontinuous maps. A novel…

Computer Vision and Pattern Recognition · Computer Science 2023-11-14 Zezeng Li , Shenghao Li , Lianbao Jin , Na Lei , Zhongxuan Luo

In this paper, we introduce a neural network-based method to address the high-dimensional dynamic unbalanced optimal transport (UOT) problem. Dynamic UOT focuses on the optimal transportation between two densities with unequal total mass,…

Optimization and Control · Mathematics 2024-09-23 Wei Wan , Jiangong Pan , Yuejin Zhang , Chenglong Bao , Zuoqiang Shi

Mini-batch optimal transport (m-OT) has been successfully used in practical applications that involve probability measures with a very high number of supports. The m-OT solves several smaller optimal transport problems and then returns the…

Machine Learning · Statistics 2022-06-08 Khai Nguyen , Dang Nguyen , Quoc Nguyen , Tung Pham , Hung Bui , Dinh Phung , Trung Le , Nhat Ho

Obtaining solutions to Optimal Transportation (OT) problems is typically intractable when the marginal spaces are continuous. Recent research has focused on approximating continuous solutions with discretization methods based on i.i.d.…

Optimization and Control · Mathematics 2021-02-17 Junqi Wang , Pei Wang , Patrick Shafto

The empirical optimal transport (OT) cost between two probability measures from random data is a fundamental quantity in transport based data analysis. In this work, we derive novel guarantees for its convergence rate when the involved…

Statistics Theory · Mathematics 2022-02-22 Shayan Hundrieser , Thomas Staudt , Axel Munk

We propose integrating optimal transport (OT) into operator learning for partial differential equations (PDEs) on complex geometries. Classical geometric learning methods typically represent domains as meshes, graphs, or point clouds. Our…

Machine Learning · Computer Science 2025-07-29 Xinyi Li , Zongyi Li , Nikola Kovachki , Anima Anandkumar

Learning conditional distributions $\pi^*(\cdot|x)$ is a central problem in machine learning, which is typically approached via supervised methods with paired data $(x,y) \sim \pi^*$. However, acquiring paired data samples is often…

We introduce a novel neural network-based algorithm to compute optimal transport (OT) plans for general cost functionals. In contrast to common Euclidean costs, i.e., $\ell^1$ or $\ell^2$, such functionals provide more flexibility and allow…

Machine Learning · Computer Science 2024-05-31 Arip Asadulaev , Alexander Korotin , Vage Egiazarian , Petr Mokrov , Evgeny Burnaev

When deploying a trained machine learning model in the real world, it is inevitable to receive inputs from out-of-distribution (OOD) sources. For instance, in continual learning settings, it is common to encounter OOD samples due to the…

Machine Learning · Computer Science 2024-01-23 Chuanwen Feng , Wenlong Chen , Ao Ke , Yilong Ren , Xike Xie , S. Kevin Zhou

In the standard formulation of the denoising problem, one is given a probabilistic model relating a latent variable $\Theta \in \Omega \subset \mathbb{R}^m \; (m\ge 1)$ and an observation $Z \in \mathbb{R}^d$ according to: $Z \mid \Theta…

Statistics Theory · Mathematics 2024-10-01 Nicolas Garcia Trillos , Bodhisattva Sen

Unbalanced optimal transport (UOT) has recently gained much attention due to its flexible framework for handling un-normalized measures and its robustness properties. In this work, we explore learning (structured) sparse transport plans in…

Machine Learning · Computer Science 2025-02-03 Piyushi Manupriya , Pratik Jawanpuria , Karthik S. Gurumoorthy , SakethaNath Jagarlapudi , Bamdev Mishra

Optimal transport is a powerful framework for the efficient allocation of resources between sources and targets. However, traditional models often struggle to scale effectively in the presence of large and heterogeneous populations. In this…

Artificial Intelligence · Computer Science 2024-11-13 Navpreet Kaur , Juntao Chen , Yingdong Lu

Optimal transport (OT) compares probability distributions by computing a meaningful alignment between their samples. CO-optimal transport (COOT) takes this comparison further by inferring an alignment between features as well. While this…

We investigate the optimal transport problem between probability measures when the underlying cost function is understood to satisfy a least action principle, also known as a Lagrangian cost. These generalizations are useful when connecting…

Machine Learning · Computer Science 2024-06-04 Aram-Alexandre Pooladian , Carles Domingo-Enrich , Ricky T. Q. Chen , Brandon Amos