English
Related papers

Related papers: Riemannian Neural Optimal Transport

200 papers

Neural network-based optimal transport (OT) is a recent and fruitful direction in the generative modeling community. It finds its applications in various fields such as domain translation, image super-resolution, computational biology and…

Machine Learning · Computer Science 2026-02-25 Roman Tarasov , Petr Mokrov , Milena Gazdieva , Evgeny Burnaev , Alexander Korotin

Recently, Neural Topic Models (NTMs) inspired by variational autoencoders have obtained increasingly research interest due to their promising results on text analysis. However, it is usually hard for existing NTMs to achieve good document…

Information Retrieval · Computer Science 2022-06-01 He Zhao , Dinh Phung , Viet Huynh , Trung Le , Wray Buntine

Optimal Transport (OT) theory investigates the cost-minimizing transport map that moves a source distribution to a target distribution. Recently, several approaches have emerged for learning the optimal transport map for a given cost…

Machine Learning · Computer Science 2025-04-01 Jaemoo Choi , Yongxin Chen , Jaewoong Choi

The optimal transport (OT) map offers the most economical way to transfer one probability measure distribution to another. Classical OT theory does not involve a discussion of preserving topological connections and orientations in…

General Topology · Mathematics 2025-07-03 Yuping Lv , Qi Zhao , Xuebin Chang , Wei Zeng

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

Optimal transport (OT) is a powerful geometric tool used to compare and align probability measures following the least effort principle. Despite its widespread use in machine learning (ML), OT problem still bears its computational burden,…

Machine Learning · Computer Science 2023-08-14 Oliver Struckmeier , Ievgen Redko , Anton Mallasto , Karol Arndt , Markus Heinonen , Ville Kyrki

Neural representations for 3D meshes are emerging as an effective solution for compact storage and efficient processing. Existing methods often rely on neural overfitting, where a coarse mesh is stored and progressively refined through…

Computer Vision and Pattern Recognition · Computer Science 2025-11-25 Xiang Gao , Yuanpeng Liu , Xinmu Wang , Jiazhi Li , Minghao Guo , Yu Guo , Xiyun Song , Heather Yu , Zhiqiang Lao , Xianfeng David Gu

Solving large scale Optimal Transport (OT) in machine learning typically relies on sampling measures to obtain a tractable discrete problem. While the discrete solver's accuracy is controllable, the rate of convergence of the discretization…

Machine Learning · Statistics 2026-02-05 Ferdinand Genans , Olivier Wintenberger

Optimal Transport (OT) problem aims to find a transport plan that bridges two distributions while minimizing a given cost function. OT theory has been widely utilized in generative modeling. In the beginning, OT distance has been used as a…

Machine Learning · Computer Science 2024-03-08 Jaemoo Choi , Jaewoong Choi , Myungjoo Kang

Transport systems on networks are crucial in various applications, but face a significant risk of being adversely affected by unforeseen circumstances such as disasters. The application of entropy-regularized optimal transport (OT) on graph…

Machine Learning · Computer Science 2025-05-07 Koshi Oishi , Yota Hashizume , Tomohiko Jimbo , Hirotaka Kaji , Kenji Kashima

Real-world image super-resolution (SR) tasks often do not have paired datasets, which limits the application of supervised techniques. As a result, the tasks are usually approached by unpaired techniques based on Generative Adversarial…

Image and Video Processing · Electrical Eng. & Systems 2025-07-09 Milena Gazdieva , Petr Mokrov , Litu Rout , Alexander Korotin , Andrey Kravchenko , Alexander Filippov , Evgeny Burnaev

In this work, we investigate applications of no-collision transportation maps introduced in [Nurbekyan et. al., 2020] in manifold learning for image data. Recently, there has been a surge in applying transportation-based distances and…

Machine Learning · Computer Science 2024-02-20 Elisa Negrini , Levon Nurbekyan

Branched Optimal Transport (BOT) is a generalization of optimal transport in which transportation costs along an edge are subadditive. This subadditivity models an increase in transport efficiency when shipping mass along the same route,…

Machine Learning · Computer Science 2022-10-17 Peter Lippmann , Enrique Fita Sanmartín , Fred A. Hamprecht

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

The optimal transport (OT) problem aims to find the most efficient mapping between two probability distributions under a given cost function, and has diverse applications in many fields such as machine learning, computer vision and computer…

Computer Vision and Pattern Recognition · Computer Science 2025-11-04 Yan Bin Ng , Xianfeng Gu

Current graph neural network (GNN) architectures naively average or sum node embeddings into an aggregated graph representation -- potentially losing structural or semantic information. We here introduce OT-GNN, a model that computes graph…

Machine Learning · Statistics 2021-10-12 Benson Chen , Gary Bécigneul , Octavian-Eugen Ganea , Regina Barzilay , Tommi Jaakkola

We introduce an optimal transport-based model for learning a metric tensor from cross-sectional samples of evolving probability measures on a common Riemannian manifold. We neurally parametrize the metric as a spatially-varying matrix field…

Machine Learning · Computer Science 2023-03-08 Christopher Scarvelis , Justin Solomon

Recently, the pretrain-finetuning paradigm has attracted tons of attention in graph learning community due to its power of alleviating the lack of labels problem in many real-world applications. Current studies use existing techniques, such…

Machine Learning · Computer Science 2022-05-11 Jiying Zhang , Xi Xiao , Long-Kai Huang , Yu Rong , Yatao Bian

Optimal transport (OT) is a versatile framework for comparing probability measures, with many applications to statistics, machine learning, and applied mathematics. However, OT distances suffer from computational and statistical scalability…

Statistics Theory · Mathematics 2022-06-08 Ziv Goldfeld , Kengo Kato , Gabriel Rioux , Ritwik Sadhu

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…

Probability · Mathematics 2023-04-26 Stephan Eckstein , Gudmund Pammer