🧬Theory

Optimal Transport

Optimal transport in ML

120 papers(showing top 100)302 total citations
Compare with other topics
Mar '24 — Feb '2696 papers
Also includes: optimal transport, wasserstein, earth mover distance, sinkhorn

Top Papers

#1

Optimal Transport Aggregation for Visual Place Recognition

Sergio Izquierdo, Javier Civera

CVPR 2024arXiv:2311.15937
138
citations
#2

Energy-guided Entropic Neural Optimal Transport

Petr Mokrov, Alexander Korotin, Alexander Kolesov et al.

ICLR 2024arXiv:2304.06094
27
citations
#3

Multi-Level Optimal Transport for Universal Cross-Tokenizer Knowledge Distillation on Language Models

Xiao Cui, Mo Zhu, Yulei Qin et al.

AAAI 2025arXiv:2412.14528
22
citations
#4

Optimal Transport for Time Series Imputation

Hao Wang, zhengnan li, Haoxuan Li et al.

ICLR 2025
17
citations
#5

Integrating Efficient Optimal Transport and Functional Maps For Unsupervised Shape Correspondence Learning

Tung Le, Khai Nguyen, Shanlin Sun et al.

CVPR 2024arXiv:2403.01781
10
citations
#6

Probability-Polarized Optimal Transport for Unsupervised Domain Adaptation

Yan Wang, Chuan-Xian Ren, Yi-Ming Zhai et al.

AAAI 2024
6
citations
#7

Analyzing and Improving Optimal-Transport-based Adversarial Networks

Jaemoo Choi, Jaewoong Choi, Myungjoo Kang

ICLR 2024arXiv:2310.02611
6
citations
#8

Variational Regularized Unbalanced Optimal Transport: Single Network, Least Action

Yuhao Sun, Zhenyi Zhang, Zihan Wang et al.

NeurIPS 2025arXiv:2505.11823
optimal transportschrödinger bridgevariational inferenceunbalanced distributions+4
6
citations
#9

Robust Barycenter Estimation using Semi-Unbalanced Neural Optimal Transport

Milena Gazdieva, Jaemoo Choi, Alexander Kolesov et al.

ICLR 2025
5
citations
#10

Unify ML4TSP: Drawing Methodological Principles for TSP and Beyond from Streamlined Design Space of Learning and Search

Yang Li, Jiale Ma, Wenzheng Pan et al.

ICLR 2025
5
citations
#11

Accelerating 3D Molecule Generation via Jointly Geometric Optimal Transport

Haokai Hong, Wanyu LIN, KC Tan

ICLR 2025arXiv:2405.15252
3d molecule generationgeometric optimal transportflow-matching objectiveequivariant networks+3
5
citations
#12

Expected Sliced Transport Plans

Xinran Liu, Rocio Diaz Martin, Yikun Bai et al.

ICLR 2025arXiv:2410.12176
optimal transport problemsliced optimal transportwasserstein distancestransportation plan construction+4
5
citations
#13

On Partial Optimal Transport: Revising the Infeasibility of Sinkhorn and Efficient Gradient Methods

Anh Duc Nguyen, Tuan Dung Nguyen, Quang Nguyen et al.

AAAI 2024arXiv:2312.13970
partial optimal transportunbalanced measuressinkhorn algorithmgradient descent methods+4
5
citations
#14

Many-Objective Multi-Solution Transport

Ziyue Li, Tian Li, Virginia Smith et al.

ICLR 2025arXiv:2403.04099
5
citations
#15

Mahalanobis Distance-based Multi-view Optimal Transport for Multi-view Crowd Localization

Qi Zhang, Kaiyi Zhang, Antoni Chan et al.

ECCV 2024
5
citations
#16

From Alexnet to Transformers: Measuring the Non-linearity of Deep Neural Networks with Affine Optimal Transport

Quentin Bouniot, Ievgen Redko, Anton Mallasto et al.

CVPR 2025
5
citations
#17

POT: Prototypical Optimal Transport for Weakly Supervised Semantic Segmentation

Jian Wang, Tianhong Dai, Bingfeng Zhang et al.

CVPR 2025
5
citations
#18

Exploiting Geometry for Treatment Effect Estimation via Optimal Transport

Yuguang Yan, Zeqin Yang, Weilin Chen et al.

AAAI 2024
4
citations
#19

Lightspeed Geometric Dataset Distance via Sliced Optimal Transport

Khai Nguyen, Hai Nguyen, Tuan Pham et al.

ICML 2025arXiv:2501.18901
4
citations
#20

Score-Based Diffusion Policy Compatible with Reinforcement Learning via Optimal Transport

Mingyang Sun, Pengxiang Ding, Weinan Zhang et al.

ICML 2025arXiv:2502.12631
4
citations
#21

Hierarchical Refinement: Optimal Transport to Infinity and Beyond

Peter Halmos, Julian Gold, Xinhao Liu et al.

ICML 2025arXiv:2503.03025
3
citations
#22

Linear Spherical Sliced Optimal Transport: A Fast Metric for Comparing Spherical Data

Xinran Liu, Yikun Bai, Rocio Diaz Martin et al.

ICLR 2025
2
citations
#23

Auditing and Enforcing Conditional Fairness via Optimal Transport

Mohsen Ghassemi, Alan Mishler, Niccolo Dalmasso et al.

AAAI 2025arXiv:2410.14029
1
citations
#24

Learning Structured Representations by Embedding Class Hierarchy with Fast Optimal Transport

Siqi Zeng, Sixian Du, Makoto Yamada et al.

ICLR 2025
1
citations
#25

Generalized Variational Inference via Optimal Transport

Jinjin Chi, Zhichao Zhang, Zhiyao Yang et al.

AAAI 2024
1
citations
#26

Differential Private Stochastic Optimization with Heavy-tailed Data: Towards Optimal Rates

Puning Zhao, Jiafei Wu, Zhe Liu et al.

AAAI 2025arXiv:2408.09891
1
citations
#27

Unsupervised Learning for Optimal Transport plan prediction between unbalanced graphs

Sonia Mazelet, Rémi Flamary, Bertrand Thirion

NeurIPS 2025
1
citations
#28

Optimizing Distributional Geometry Alignment with Optimal Transport for Generative Dataset Distillation

Xiao Cui, Yulei Qin, Wengang Zhou et al.

NeurIPS 2025
1
citations
#29

Continual Optimization with Symmetry Teleportation for Multi-Task Learning

Zhipeng Zhou, Ziqiao Meng, Pengcheng Wu et al.

NeurIPS 2025
1
citations
#30

Tightening Causal Bounds via Covariate-Aware Optimal Transport

Sirui Lin, Zijun Gao, Jose Blanchet et al.

ICML 2025arXiv:2502.01164
1
citations
#31

Generalized Sobolev Transport for Probability Measures on a Graph

Tam Le, Truyen Nguyen, Kenji Fukumizu

ICML 2024
optimal transportgraph metric spacesobolev transportorlicz-wasserstein+3
—
not collected
#32

Sparsistency for inverse optimal transport

Francisco Andrade, Gabriel Peyré, Clarice Poon

ICLR 2024
—
not collected
#33

On the Private Estimation of Smooth Transport Maps

Clément Lalanne, Franck Iutzeler, Loubes Jean-Michel et al.

ICML 2025arXiv:2502.01168
—
not collected
#34

Reducing Confounding Bias without Data Splitting for Causal Inference via Optimal Transport

Yuguang Yan, Zongyu Li, Haolin Yang et al.

ICML 2025
—
not collected
#35

Overcoming Spurious Solutions in Semi-Dual Neural Optimal Transport: A Smoothing Approach for Learning the Optimal Transport Plan

Jaemoo Choi, Jaewoong Choi, Dohyun Kwon

ICML 2025arXiv:2502.04583
—
not collected
#36

Wasserstein Wormhole: Scalable Optimal Transport Distance with Transformer

Doron Haviv, Russell Kunes, Thomas Dougherty et al.

ICML 2024arXiv:2404.09411
optimal transportwasserstein metrictransformer autoencoderdistribution embedding+3
—
not collected
#37

Estimating Barycenters of Distributions with Neural Optimal Transport

Alexander Kolesov, Petr Mokrov, Igor Udovichenko et al.

ICML 2024arXiv:2402.03828
wasserstein barycenteroptimal transportneural ot solveradversarial learning+2
—
not collected
#38

TAROT: Targeted Data Selection via Optimal Transport

Lan Feng, Fan Nie, Yuejiang Liu et al.

ICML 2025arXiv:2412.00420
—
not collected
#39

Nonlinear Filtering with Brenier Optimal Transport Maps

Mohammad Al-Jarrah, Niyizhen Jin, Bamdad Hosseini et al.

ICML 2024
nonlinear filteringoptimal transport mapsparticle filtersdegenerate likelihoods+4
—
not collected
#40

Optimal transport-based conformal prediction

Gauthier Thurin, Kimia Nadjahi, Claire Boyer

ICML 2025arXiv:2501.18991
—
not collected
#41

Neural Optimal Transport with General Cost Functionals

Arip Asadulaev, Alexander Korotin, Vage Egiazarian et al.

ICLR 2024arXiv:2205.15403
—
not collected
#42

LCOT: Linear Circular Optimal Transport

ROCIO DIAZ MARTIN, Ivan Medri, Yikun Bai et al.

ICLR 2024
—
not collected
#43

P$^2$OT: Progressive Partial Optimal Transport for Deep Imbalanced Clustering

Chuyu Zhang, Hui Ren, Xuming He

ICLR 2024arXiv:2401.09266
—
not collected
#44

Transport meets Variational Inference: Controlled Monte Carlo Diffusions

Francisco Vargas, Shreyas Padhy, Denis Blessing et al.

ICLR 2024arXiv:2307.01050
—
not collected
#45

Joint Metric Space Embedding by Unbalanced Optimal Transport with Gromov–Wasserstein Marginal Penalization

Florian Beier, Moritz Piening, Robert Beinert et al.

ICML 2025
—
not collected
#46

The Sparse-Plus-Low-Rank Quasi-Newton Method for Entropic-Regularized Optimal Transport

Chenrui Wang, Yixuan Qiu

ICML 2025
—
not collected
#47

Computing Optimal Transport Maps and Wasserstein Barycenters Using Conditional Normalizing Flows

Gabriele Visentin, Patrick Cheridito

ICML 2025arXiv:2505.22364
—
not collected
#48

Robust Multi-Task Learning with Excess Risks

Yifei He, Shiji Zhou, Guojun Zhang et al.

ICML 2024
multi-task learninglabel noise robustnessexcess risk estimationtask balancing methods+3
—
not collected
#49

Universal Neural Optimal Transport

Jonathan Geuter, Gregor Kornhardt, Ingimar Tomasson et al.

ICML 2025arXiv:2212.00133
—
not collected
#50

Residual-Conditioned Optimal Transport: Towards Structure-Preserving Unpaired and Paired Image Restoration

Xiaole Tang, Hu Xin, Xiang Gu et al.

ICML 2024arXiv:2405.02843
optimal transportimage restorationtransport residualfourier residual-guided+4
—
not collected
#51

Revisiting Deep Audio-Text Retrieval Through the Lens of Transportation

Tien Manh Luong, Khai Nguyen, Nhat Ho et al.

ICLR 2024
—
not collected
#52

Unbalancedness in Neural Monge Maps Improves Unpaired Domain Translation

Luca Eyring, Dominik Klein, Théo Uscidda et al.

ICLR 2024
—
not collected
#53

Optimal Transport Barycenter via Nonconvex-Concave Minimax Optimization

Kaheon Kim, Rentian Yao, Changbo Zhu et al.

ICML 2025
—
not collected
#54

Recover and Match: Open-Vocabulary Multi-Label Recognition through Knowledge-Constrained Optimal Transport

Hao Tan, Zichang Tan, Jun Li et al.

CVPR 2025
—
not collected
#55

Optimal Transport for Brain-Image Alignment: Unveiling Redundancy and Synergy in Neural Information Processing

Yang Xiao, Wang Lu, Jie Ji et al.

ICCV 2025
—
not collected
#56

CLOT: Closed Loop Optimal Transport for Unsupervised Action Segmentation

Elena Bueno-Benito, Mariella Dimiccoli

ICCV 2025arXiv:2507.03539
—
not collected
#57

OPTICAL: Leveraging Optimal Transport for Contribution Allocation in Dataset Distillation

Xiao Cui, Yulei Qin, Wengang Zhou et al.

CVPR 2025
—
not collected
#58

Discovering Fine-Grained Visual-Concept Relations by Disentangled Optimal Transport Concept Bottleneck Models

Yan Xie, Zequn Zeng, Hao Zhang et al.

CVPR 2025
—
not collected
#59

LaCoOT: Layer Collapse through Optimal Transport

Victor Quétu, Zhu LIAO, Nour Hezbri et al.

ICCV 2025
—
not collected
#60

Visual Prompting via Partial Optimal Transport

MENGYU ZHENG, Zhiwei Hao, Yehui Tang et al.

ECCV 2024
—
not collected
#61

The Curse of Conditions: Analyzing and Improving Optimal Transport for Conditional Flow-Based Generation

Ho Kei Cheng, Alex Schwing

ICCV 2025
—
not collected
#62

Taming Flow Matching with Unbalanced Optimal Transport into Fast Pansharpening

Zihan Cao, Yu Zhong, Liang-Jian Deng

ICCV 2025
—
not collected
#63

OTMatch: Improving Semi-Supervised Learning with Optimal Transport

Zhiquan Tan, Kaipeng Zheng, Weiran Huang

ICML 2024arXiv:2310.17455
semi-supervised learningoptimal transportself-training techniquesdistribution matching+3
—
not collected
#64

A Computational Framework for Solving Wasserstein Lagrangian Flows

Kirill Neklyudov, Rob Brekelmans, Alexander Tong et al.

ICML 2024
optimal transportwasserstein lagrangian flowsschrödinger bridgeunbalanced optimal transport+4
—
not collected
#65

Position: Optimization in SciML Should Employ the Function Space Geometry

Johannes Müller, Marius Zeinhofer

ICML 2024
function space geometryinfinite-dimensional optimizationscientific machine learningoptimize then discretize+1
—
not collected
#66

Light and Optimal Schrödinger Bridge Matching

Nikita Gushchin, Sergei Kholkin, Evgeny Burnaev et al.

ICML 2024
schrödinger bridgesentropic optimal transportbridge matching proceduresdiffusion models+2
—
not collected
#67

Reducing Balancing Error for Causal Inference via Optimal Transport

Yuguang Yan, Hao Zhou, Zeqin Yang et al.

ICML 2024
causal inferenceconfounding biasoptimal transportdistribution shift+3
—
not collected
#68

GiLOT: Interpreting Generative Language Models via Optimal Transport

Xuhong Li, Jiamin Chen, Yekun Chai et al.

ICML 2024
optimal transportfeature attribution methodslarge language modelsautoregressive language models+4
—
not collected
#69

Submodular framework for structured-sparse optimal transport

Piyushi Manupriya, Pratik Kumar Jawanpuria, Karthik Gurumoorthy et al.

ICML 2024
unbalanced optimal transportstructured sparse transportmaximum mean discrepancyweakly submodular function+4
—
not collected
#70

Generative Conditional Distributions by Neural (Entropic) Optimal Transport

Bao Nguyen, Binh Nguyen, Trung Hieu Nguyen et al.

ICML 2024
conditional distribution learningneural optimal transportentropic optimal transportgenerative modeling+3
—
not collected
#71

Optimal Transport for Structure Learning Under Missing Data

Vy Vo, He Zhao, Trung Le et al.

ICML 2024arXiv:2402.15255
causal discoverymissing data imputationoptimal transportstructure learning+3
—
not collected
#72

Quantum Theory and Application of Contextual Optimal Transport

Nicola Mariella, Albert Akhriev, Francesco Tacchino et al.

ICML 2024arXiv:2402.14991
optimal transportcontextual distribution learningquantum computing formulationdoubly stochastic matrices+4
—
not collected
#73

Parameter Estimation in DAGs from Incomplete Data via Optimal Transport

Vy Vo, Trung Le, Tung-Long Vuong et al.

ICML 2024
parameter estimationdirected graphical modelsincomplete dataoptimal transport+2
—
not collected
#74

Partially Observed Trajectory Inference using Optimal Transport and a Dynamics Prior

Anming Gu, Edward Chien, Kristjan Greenewald

ICLR 2025
—
not collected
#75

Hierarchical Optimization via LLM-Guided Objective Evolution for Mobility-on-Demand Systems

Yi Zhang, Yushen Long, Yun Ni et al.

NeurIPS 2025arXiv:2510.10644
mobility-on-demand systemshierarchical optimizationlarge language model guidancereinforcement learning approaches+4
—
not collected
#76

Stability and Oracle Inequalities for Optimal Transport Maps between General Distributions

Shubo Li, Yizhe Ding, Lingzhou Xue et al.

NeurIPS 2025
—
not collected
#77

On the Relation between Rectified Flows and Optimal Transport

Johannes Hertrich, Antonin Chambolle, Julie Delon

NeurIPS 2025arXiv:2505.19712
rectified flowsflow matchingoptimal transportvelocity fields+2
—
not collected
#78

Efficient Training-Free Online Routing for High-Volume Multi-LLM Serving

Fangzhou Wu, Sandeep Silwal

NeurIPS 2025arXiv:2509.02718
llm routingonline servingtraining-free algorithmapproximate nearest neighbor search+4
—
not collected
#79

Unbalanced Optimal Total Variation Transport: A Theoretical Approach to Spatial Resource Allocation Problems

Nhan-Phu Chung, Jinhui Han, Bohan Li et al.

NeurIPS 2025
unbalanced optimal transporttotal variation penaltiesspatial resource allocationsemi-discrete setting+4
—
not collected
#80

Optimal Flow Transport and its Entropic Regularization: a GPU-friendly Matrix Iterative Algorithm for Flow Balance Satisfaction

Liangliang Shi, Yufeng Li, Kaipeng Zeng et al.

ICLR 2025
—
not collected
#81

MMD-Regularized Unbalanced Optimal Transport

SakethaNath Jagarlapudi, Pratik Jawanpuria, Piyushi Manupriya

ICLR 2025
—
not collected
#82

Stochastic Optimization in Semi-Discrete Optimal Transport: Convergence Analysis and Minimax Rate

Ferdinand Genans, Antoine Godichon-Baggioni, François-Xavier Vialard et al.

NeurIPS 2025
—
not collected
#83

Differentiable Generalized Sliced Wasserstein Plans

Laetitia Chapel, Romain Tavenard, Samuel Vaiter

NeurIPS 2025
—
not collected
#84

Learning Partial Graph Matching via Optimal Partial Transport

Gathika Ratnayaka, James Nichols, Qing Wang

ICLR 2025arXiv:2410.16718
partial graph matchingoptimal partial transportlinear sum assignmentgraph matching architecture+2
—
not collected
#85

Non-exchangeable Conformal Prediction with Optimal Transport: Tackling Distribution Shift with Unlabeled Data

Alvaro Correia, Christos Louizos

NeurIPS 2025
—
not collected
#86

Tree-Sliced Entropy Partial Transport

Viet-Hoang Tran, Thanh Tran, Thanh Chu et al.

NeurIPS 2025
optimal transportpartial transporttree-sliced wassersteinimbalanced data+2
—
not collected
#87

An Efficient Orlicz-Sobolev Approach for Transporting Unbalanced Measures on a Graph

Tam Le, Truyen Nguyen, Hideitsu Hino et al.

NeurIPS 2025arXiv:2502.00739
optimal transportunbalanced measuresgraph metric spacesorlicz-wasserstein distance+3
—
not collected
#88

A Truncated Newton Method for Optimal Transport

Mete Kemertas, Amir-massoud Farahmand, Allan Jepson

ICLR 2025
—
not collected
#89

Cross-Domain Offline Policy Adaptation with Optimal Transport and Dataset Constraint

Jiafei Lyu, Mengbei Yan, Zhongjian Qiao et al.

ICLR 2025
—
not collected
#90

Bootstrap Your Uncertainty: Adaptive Robust Classification Driven by Optimal-Transport

Jiawei Huang, Minming Li, Hu Ding

NeurIPS 2025
—
not collected
#91

Efficient Algorithms for Robust and Partial Semi-Discrete Optimal Transport

Pankaj Agarwal, Sharath Raghvendra, Pouyan Shirzadian et al.

NeurIPS 2025
—
not collected
#92

Estimation of Stochastic Optimal Transport Maps

Sloan Nietert, Ziv Goldfeld

NeurIPS 2025
—
not collected
#93

Towards Prospective Medical Image Reconstruction via Knowledge-Informed Dynamic Optimal Transport

Taoran Zheng, Yan Yang, Xing Li et al.

NeurIPS 2025
—
not collected
#94

Pairwise Optimal Transports for Training All-to-All Flow-Based Condition Transfer Model

Kotaro Ikeda, Masanori Koyama, Jinzhe Zhang et al.

NeurIPS 2025arXiv:2504.03188
optimal transportflow-based modelsconditional distributionscontinuous conditions+3
—
not collected
#95

Targeted Maximum Likelihood Learning: An Optimization Perspective

Diyang Li, Kyra Gan

NeurIPS 2025
—
not collected
#96

Counterfactual Identifiability via Dynamic Optimal Transport

Fabio De Sousa Ribeiro, Ainkaran Santhirasekaram, Ben Glocker

NeurIPS 2025
—
not collected
#97

Transfer Faster, Price Smarter: Minimax Dynamic Pricing under Cross-Market Preference Shift

Yi Zhang, Elynn Chen, Yujun Yan

NeurIPS 2025arXiv:2505.17203
—
not collected
#98

Improved Confidence Regions and Optimal Algorithms for Online and Offline Linear MNL Bandits

Yuxuan Han, Jose Blanchet, Zhengyuan Zhou

NeurIPS 2025
assortment optimizationlinear mnl banditsconfidence regionspessimistic-based algorithms+4
—
not collected
#99

Solving Discrete (Semi) Unbalanced Optimal Transport with Equivalent Transformation Mechanism and KKT-Multiplier Regularization

Weiming Liu, Xinting Liao, Jun Dan et al.

NeurIPS 2025
optimal transportunbalanced optimal transportsemi-unbalanced optimal transportkkt-multiplier regularization+4
—
not collected
#100

Trust Region Constrained Measure Transport in Path Space for Stochastic Optimal Control and Inference

Denis Blessing, Julius Berner, Lorenz Richter et al.

NeurIPS 2025arXiv:2508.12511
stochastic optimal controlpath space measuretrust region optimizationmeasure transport+4
—
not collected