Maximum flow and minimum-cost flow in almost-linear time

L Chen, R Kyng, YP Liu, R Peng… - 2022 IEEE 63rd …, 2022 - ieeexplore.ieee.org
We give an algorithm that computes exact maximum flows and minimum-cost flows on
directed graphs with m edges and polynomially bounded integral demands, costs, and …

Spectr: Fast speculative decoding via optimal transport

Z Sun, AT Suresh, JH Ro, A Beirami… - Advances in Neural …, 2024 - proceedings.neurips.cc
Autoregressive sampling from large language models has led to state-of-the-art results in
several natural language tasks. However, autoregressive sampling generates tokens one at …

Ot-filter: An optimal transport filter for learning with noisy labels

C Feng, Y Ren, X Xie - … of the IEEE/CVF Conference on …, 2023 - openaccess.thecvf.com
The success of deep learning is largely attributed to the training over clean data. However,
data is often coupled with noisy labels in practice. Learning with noisy labels is challenging …

MICo: Improved representations via sampling-based state similarity for Markov decision processes

PS Castro, T Kastner… - Advances in Neural …, 2021 - proceedings.neurips.cc
We present a new behavioural distance over the state space of a Markov decision process,
and demonstrate the use of this distance as an effective means of shaping the learnt …

On unbalanced optimal transport: An analysis of sinkhorn algorithm

K Pham, K Le, N Ho, T Pham… - … Conference on Machine …, 2020 - proceedings.mlr.press
We provide a computational complexity analysis for the Sinkhorn algorithm that solves the
entropic regularized Unbalanced Optimal Transport (UOT) problem between two measures …

Accelerated Bregman primal-dual methods applied to optimal transport and Wasserstein Barycenter problems

A Chambolle, JP Contreras - SIAM Journal on Mathematics of Data Science, 2022 - SIAM
This paper discusses the efficiency of Hybrid Primal-Dual (HPD) type algorithms to
approximately solve discrete Optimal Transport (OT) and Wasserstein Barycenter (WB) …

Mirror sinkhorn: Fast online optimization on transport polytopes

M Ballu, Q Berthet - International Conference on Machine …, 2023 - proceedings.mlr.press
Optimal transport is an important tool in machine learning, allowing to capture geometric
properties of the data through a linear program on transport polytopes. We present a single …

Importance sparsification for sinkhorn algorithm

M Li, J Yu, T Li, C Meng - Journal of Machine Learning Research, 2023 - jmlr.org
Sinkhorn algorithm has been used pervasively to approximate the solution to optimal
transport (OT) and unbalanced optimal transport (UOT) problems. However, its practical …

Hilbert curve projection distance for distribution comparison

T Li, C Meng, H Xu, J Yu - IEEE Transactions on Pattern …, 2024 - ieeexplore.ieee.org
Distribution comparison plays a central role in many machine learning tasks like data
classification and generative modeling. In this study, we propose a novel metric, called …

On the efficiency of entropic regularized algorithms for optimal transport

T Lin, N Ho, MI Jordan - Journal of Machine Learning Research, 2022 - jmlr.org
We present several new complexity results for the entropic regularized algorithms that
approximately solve the optimal transport (OT) problem between two discrete probability …