entropic optimal transport assignments

Designs, implements, and analyzes entropy-regularized optimal transport solvers (e.g., Sinkhorn iterations) that compute transport plans, couplings, or assignment matrices under entropic regularization. Builds and tunes these assignment computations for structured tasks (e.g., patch-to-prototype assignments, OT-driven prototype discovery), integrates them into end-to-end pipelines (including distillation) and evaluates convergence, stability, and computational trade-offs.

entropicoptimaltransportassignments

Recent Skill Trend

Momentum and market value over time
Trending
Score
No comparison yet
-0.07
Oct 01, 2026Oct 01, 2026
Career
Value
No comparison yet
$202K/year
Oct 01, 2026Oct 01, 2026

Recommended Survey Paper

Quick overview of the field
View more

Must-Read Papers

Most classic and influential ideas
View more

Decreasing Entropic Regularization Averaged Gradient for Semi-Discrete Optimal Transport

Oct 31, 2025
FG
Ferdinand Genans
🏛️ Sorbonne Université | Université Gustave Eiffel | Wolfgang Pauli Institute

To address the trade-off between bias and convergence rate induced by entropy regularization in semi-discrete optimal transport (OT), this paper proposes the Dynamic Regularization Annealing Gradient (DRAG) algorithm. DRAG operates within a stochastic gradient descent framework, adaptively decaying the entropy regularization strength while incorporating gradient averaging to progressively eliminate regularization bias without sacrificing computational efficiency. Theoretically, DRAG guarantees $O(1/t)$ convergence rates for both the OT cost and dual potential estimates, and $O(1/sqrt{t})$ convergence for the OT map; moreover, as the regularization coefficient vanishes, the solution converges to the unbiased primal OT solution. Empirical evaluations demonstrate DRAG’s superior accuracy, faster convergence, and enhanced robustness compared to existing methods.

Accelerating convergence for semi-discrete optimal transport with theoretical guaranteesDeveloping adaptive regularization that decreases during optimization processMitigating bias introduced by entropic regularization in optimal transport problems

This paper addresses the entropy-regularized optimal transport (EOT) problem by proposing a novel optimization-based framework that dispenses with traditional strong structural assumptions—such as strict convexity or smoothness—required by existing algorithms. Methodologically, it (1) formulates either a semi-dual problem or a nonconvex constrained optimization over a subset of joint distributions, and (2) integrates mirror descent with momentum acceleration to achieve provably accelerated non-asymptotic convergence—attaining a rate of $O(1/k^2)$ under minimal assumptions, matching the acceleration performance of classical Euclidean methods. Theoretically, this work provides the first acceleration guarantee for EOT without requiring strong convexity or smoothness. Furthermore, the framework extends naturally to generalized transport problems, including dynamical Schrödinger bridges, demonstrating both broad applicability and empirical effectiveness.

Develop novel methods for entropic-regularised optimal transportPropose momentum method with accelerated convergence guaranteesProvide non-asymptotic convergence rates under minimal assumptions

For large-scale optimal transport (OT) problems, the entropy-regularized Sinkhorn algorithm suffers from numerical instability and slow convergence when the regularization parameter is small. To address this, we propose the Sparse Newton–Proximal Sinkhorn (SNPS) algorithm—a novel coupling of sparse Newton optimization with proximal Sinkhorn iterations. This is the first method to deeply integrate sparse Newton methods into the Sinkhorn framework while preserving global convergence guarantees. Theoretically, we establish rigorous convergence properties; empirically, SNPS achieves near-exact OT solutions on multiple standard benchmarks, with significantly accelerated per-iteration performance. It substantially reduces computational complexity and markedly alleviates sensitivity to the regularization parameter. Compared to state-of-the-art OT solvers, SNPS delivers superior overall performance—achieving high accuracy, high efficiency, and strong robustness simultaneously.

Address numerical instability in Sinkhorn algorithmEnhance efficiency in optimal transportEnsure global convergence for large-scale problems

Imitation-Regularized Optimal Transport on Networks: Provable Robustness and Application to Logistics Planning

Feb 28, 2024
KO
Koshi Oishi
🏛️ Toyota Central R&D Labs, Inc. | Graduate School of Informatics Kyoto University | Toyota Motor Corporation

Networked transportation systems exhibit insufficient robustness under sudden disruptions (e.g., natural disasters). Method: This paper proposes Imitation-regularized Optimal Transport (I-OT), the first framework integrating imitation learning into graph-structured optimal transport. It mathematically embeds human domain knowledge to enhance model interpretability and practicality. Theoretically, I-OT builds upon entropy-regularized optimal transport and convex optimization analysis to rigorously establish transmission stability under node/edge failures and accelerated convergence. Technically, it combines graph neural network-based modeling with simulation-driven validation. Results: Evaluated on automotive parts logistics simulation, I-OT significantly improves path scheduling robustness. Moreover, it establishes an interpretable theoretical linkage between the learned transport policy and real-world logistics resilience—bridging algorithmic design with operational reliability.

Enhancing robustness of transport networks against unforeseen disastersIncorporating prior knowledge into optimal transport for better interpretabilityValidating method effectiveness via logistics simulation with real data

Score-based Generative Neural Networks for Large-Scale Optimal Transport

Oct 07, 2021
MD
Max Daniels
🏛️ Northeastern University | Brandeis University

To address the high computational cost and curse-of-dimensionality challenges in sampling optimal transport (OT) couplings for large-scale, high-dimensional data, this paper proposes an efficient learning framework based on score-based generative models. Specifically, conditioned on source samples, it iteratively generates target samples following the Sinkhorn-regularized OT coupling via Langevin dynamics. Crucially, it jointly parameterizes the score function and Sinkhorn potential functions—enabling, for the first time, end-to-end co-learning of score-based generation and OT coupling. We theoretically establish the convergence of gradient descent on the network parameters under mild assumptions. Experiments demonstrate that our method significantly improves both accuracy and speed of coupling estimation across diverse large-scale OT tasks, while maintaining scalability and practical applicability.

Learning Sinkhorn coupling via score-based networksSampling optimal transport coupling between distributionsSolving high-dimensional transport without linear programming

Latest Papers

What's happening recently
View more

This study addresses the challenge of non-unique identification of model parameters arising from data insufficiency. To overcome this, it proposes a multi-marginal entropy optimal transport framework that reformulates the partial identification problem as an optimal transport problem over path space, and employs the Sinkhorn iterative algorithm to efficiently solve the resulting infinite-dimensional optimization problem. Theoretically, the paper establishes the convergence of the proposed algorithm and derives the consistency rate of the estimator. Empirically, it demonstrates computationally efficient implementation with precise bound estimation, successfully applying the approach to demand model analysis. Ultimately, this work provides a unified theoretical and computational paradigm for addressing partial identification problems in settings characterized by limited data availability.

entropy regularizationinfinite-dimensional optimizationoptimal transport

This work addresses the lack of a unified theoretical foundation for parameter identifiability, estimation consistency, and algorithmic convergence in feature-parameterized inverse optimal transport (IOT). It proposes an analytical framework based on Sinkhorn linearization and spectral surrogates, which—through the introduction of spectral sandwich inequalities, restricted Hessian analysis, and ℓ₁ regularization—establishes, for the first time, four core theoretical guarantees for IOT: global parameter identifiability, exact support recovery, a Lipschitz continuous inverse mapping, and monotonic convergence of gradient descent. The framework combines geometric transparency with spectral precision and further quantifies the Hölder stability of the projection map under model misspecification. Numerical experiments corroborate the theoretical findings.

feature-parameterized costinverse optimal transportmodel misspecification

This work addresses the computational challenges of large-scale entropy-regularized optimal transport, which is notoriously difficult to solve efficiently due to its high complexity. Existing GPU-based methods struggle to balance convergence speed and parallel efficiency. To overcome this limitation, we propose a high-performance GPU solver that synergistically combines the strengths of quasi-Newton methods and the Sinkhorn algorithm. Our approach introduces several key innovations: amortized sign analysis, asynchronous Sinkhorn iterations, and fused gradient kernels, complemented by sparse-plus-low-rank approximations, asynchronous computation, and optimized memory access patterns. While preserving theoretical convergence guarantees, the proposed method consistently outperforms state-of-the-art GPU solvers across multiple benchmark tasks, achieving substantial acceleration.

Computational EfficiencyEntropic RegularizationGPU Acceleration

This work addresses the rigidity of classical optimal transport in scenarios where mass conservation does not hold and its poor sample complexity in high dimensions. The authors investigate the finite-sample theory of entropy-regularized unbalanced optimal transport, introducing a translation-invariant dual formulation and analyzing its geometric structure. They establish, for the first time, high-probability convergence bounds for empirical optimal couplings at the coupling level. Their analysis demonstrates that entropy regularization not only substantially mitigates the curse of dimensionality and reduces the required sample size but also enhances estimation stability, all while preserving compatibility with efficient solvers such as Sinkhorn algorithms.

entropic regularizationhigh-dimensional statisticsoptimal coupling

Hot Scholars

AM

Anthea Monod

Associate Professor, Department of Mathematics, Imperial College London
Applied Algebraic GeometryTopological Data AnalysisMathematical Biology
SB

Samir Bhatt

Professor of Machine Learning and Public Health University of Copenhagen
Public HealthGeneticsInfectious DiseasesMachine Learning
GS

Geoffrey Schiebinger

University of British Columbia
mathematical biologyoptimal transportstatisticssingle cell analysis
AM

Alessandro Micheli

Research Associate in Machine Learning, Imperial College London
Bayesian StatisticsDeep LearningGenerative AIUncertainty Quantification
LB

Linus Bleistein

PhD candidate, Inria Paris, UEVE
Time SeriesTrustworthy MLOptimal TransportHealthcare