中文
相关论文

相关论文: Accelerating Sinkhorn for Entropy-Regularized Opti…

200 篇论文

The current best practice for computing optimal transport (OT) is via entropy regularization and Sinkhorn iterations. This algorithm runs in quadratic time as it requires the full pairwise cost matrix, which is prohibitively expensive for…

机器学习 · 计算机科学 2022-04-06 Johannes Gasteiger , Marten Lienen , Stephan Günnemann

Efficient computation of the optimal transport distance between two distributions serves as an algorithm subroutine that empowers various applications. This paper develops a scalable first-order optimization-based method that computes…

机器学习 · 计算机科学 2024-06-21 Gen Li , Yanxi Chen , Yu Huang , Yuejie Chi , H. Vincent Poor , Yuxin Chen

Multimarginal optimal transport (MOT) is a powerful framework for modeling interactions between multiple distributions, yet its applicability is bottlenecked by a high computational overhead. Entropic regularization provides computational…

机器学习 · 计算机科学 2025-06-03 Dor Tsur , Ziv Goldfeld , Kristjan Greenewald , Haim Permuter

The Sinkhorn algorithm is a widely used method for solving the optimal transport problem, and the Greenkhorn algorithm is one of its variants. While there are modified versions of these two algorithms whose computational complexities are…

最优化与控制 · 数学 2023-10-20 Jianzhou Luo , Dingchuan Yang , Ke Wei

Sinkhorn algorithm has been used pervasively to approximate the solution to optimal transport (OT) and unbalanced optimal transport (UOT) problems. However, its practical application is limited due to the high computational complexity. To…

机器学习 · 统计学 2026-04-07 Mengyu Li , Jun Yu , Tao Li , Cheng Meng

Optimal transport (OT) serves as a natural framework for comparing probability measures, with applications in statistics, machine learning, and applied mathematics. Alas, statistical estimation and exact computation of the OT distances…

统计理论 · 数学 2024-05-14 Tao Wang , Ziv Goldfeld

In this work, we develop a collection of novel methods for the entropic-regularised optimal transport problem, which are inspired by existing mirror descent interpretations of the Sinkhorn algorithm used for solving this problem. These are…

最优化与控制 · 数学 2025-07-17 Vishwak Srinivasan , Qijia Jiang

This paper studies the Partial Optimal Transport (POT) problem between two unbalanced measures with at most $n$ supports and its applications in various AI tasks such as color transfer or domain adaptation. There is hence the need for fast…

机器学习 · 计算机科学 2023-12-25 Anh Duc Nguyen , Tuan Dung Nguyen , Quang Minh Nguyen , Hoang H. Nguyen , Lam M. Nguyen , Kim-Chuan Toh

This work introduces novel computational methods for entropic optimal transport (OT) problems under martingale-type conditions. The considered problems include the discrete martingale optimal transport (MOT) problem. Moreover, as the…

最优化与控制 · 数学 2025-08-26 Xun Tang , Michael Shavlovsky , Holakou Rahmanian , Tesi Xiao , Lexing Ying

We consider the entropic regularization of discretized optimal transport and propose to solve its optimality conditions via a logarithmic Newton iteration. We show a quadratic convergence rate and validate numerically that the method…

最优化与控制 · 数学 2018-02-12 Christoph Brauer , Christian Clason , Dirk Lorenz , Benedikt Wirth

We analyze two algorithms for approximating the general optimal transport (OT) distance between two discrete distributions of size $n$, up to accuracy $\varepsilon$. For the first algorithm, which is based on the celebrated Sinkhorn's…

数据结构与算法 · 计算机科学 2018-06-08 Pavel Dvurechensky , Alexander Gasnikov , Alexey Kroshnin

We develop a fast and reliable method for solving large-scale optimal transport (OT) problems at an unprecedented combination of speed and accuracy. Built on the celebrated Douglas-Rachford splitting technique, our method tackles the…

最优化与控制 · 数学 2021-10-25 Vien V. Mai , Jacob Lindbäck , Mikael Johansson

Entropic optimal transport (EOT) via Sinkhorn iterations is widely used in modern machine learning, yet GPU solvers remain inefficient at scale. Tensorized implementations suffer quadratic HBM traffic from dense $n\times m$ interactions,…

机器学习 · 计算机科学 2026-05-22 Felix X. -F. Ye , Xingjie Li , An Yu , Ming-Ching Chang , Linsong Chu , Davis Wertheimer

Several recent applications of optimal transport (OT) theory to machine learning have relied on regularization, notably entropy and the Sinkhorn algorithm. Because matrix-vector products are pervasive in the Sinkhorn algorithm, several…

机器学习 · 统计学 2021-03-09 Meyer Scetbon , Marco Cuturi , Gabriel Peyré

Adapted optimal transport (AOT) problems are optimal transport problems for distributions of a time series where couplings are constrained to have a temporal causal structure. In this paper, we develop computational tools for solving AOT…

概率论 · 数学 2023-04-26 Stephan Eckstein , Gudmund Pammer

Entropic regularization is quickly emerging as a new standard in optimal transport (OT). It enables to cast the OT computation as a differentiable and unconstrained convex optimization problem, which can be efficiently solved using the…

机器学习 · 统计学 2018-02-21 Mathieu Blondel , Vivien Seguy , Antoine Rolet

This article describes a set of methods for quickly computing the solution to the regularized optimal transport problem. It generalizes and improves upon the widely-used iterative Bregman projections algorithm (or Sinkhorn--Knopp…

数值分析 · 数学 2021-04-02 Alexis Thibault , Lénaïc Chizat , Charles Dossal , Nicolas Papadakis

Inverse optimal transport (OT) refers to the problem of learning the cost function for OT from observed transport plan or its samples. In this paper, we derive an unconstrained convex optimization formulation of the inverse OT problem,…

机器学习 · 计算机科学 2021-07-06 Shaojun Ma , Haodong Sun , Xiaojing Ye , Hongyuan Zha , Haomin Zhou

Optimal transport (OT) distances are finding evermore applications in machine learning and computer vision, but their wide spread use in larger-scale problems is impeded by their high computational cost. In this work we develop a family of…

机器学习 · 统计学 2018-03-06 Brahim Khalil Abid , Robert M. Gower

Regularized optimal transport (OT) is now increasingly used as a loss or as a matching layer in neural networks. Entropy-regularized OT can be computed using the Sinkhorn algorithm but it leads to fully-dense transportation plans, meaning…

机器学习 · 统计学 2023-04-17 Tianlin Liu , Joan Puigcerver , Mathieu Blondel