Conditional Optimal Transport on Function Spaces
Not on the map. Why?
Abstract (arXiv)
We present a systematic study of conditional triangular transport maps in function spaces from the perspective of optimal transportation and with a view towards amortized Bayesian inference. More specifically, we develop a theory of constrained optimal transport problems that describe block-triangular Monge maps that characterize conditional measures along with their Kantorovich relaxations. This generalizes the theory of optimal triangular transport to separable infinite-dimensional function spaces with general cost functions. We further tailor our results to the case of Bayesian inference problems and obtain regularity estimates on the conditioning maps from the prior to the posterior. Finally, we present numerical experiments that demonstrate the computational applicability of our theoretical results for amortized and likelihood-free inference of functional parameters.
Key facts. Cited by 26 papers in the map (data through 7 Oct 2026). Builds on 66 of its 122 references that are in the map. First posted on arXiv 9 Nov 2023.
Builds on
66 of 122 references.
References · 66
- 2015Deep Unsupervised Learning using Nonequilibrium Thermodynamicscited by 7,005
- 2017Improved Training of Wasserstein GANscited by 2,223
- 2015Variational Inference with Normalizing Flowscited by 2,161
- 2014Conditional Generative Adversarial Netscited by 2,152
- 2021Image Super-Resolution via Iterative Refinementcited by 1,506
- 2016Variational Inference: A Review for Statisticianscited by 1,466
- 2019Normalizing Flows for Probabilistic Modeling and Inferencecited by 1,315
- 2021Variational Diffusion Modelscited by 1,145
- 2020Improved Techniques for Training Score-Based Generative Modelscited by 1,036
- 2023Stochastic Interpolants: A Unifying Framework for Flows and Diffusionscited by 897
- 2018Computational Optimal Transportcited by 765
- 2018FFJORD: Free-form Continuous Dynamics for Scalable Reversible Generative Modelscited by 584
38 more
- 2016NIPS 2016 Tutorial: Generative Adversarial Networkscited by 284
- 2020Model Reduction and Neural Networks for Parametric PDEscited by 257
- 2017Advances in Variational Inferencecited by 236
- 2018Sequential Neural Likelihood: Fast Likelihood-free Inference with Autoregressive Flowscited by 217
- 2019Optimal transport mapping via input convex neural networkscited by 202
- 2021Conditional Image Generation with Score-Based Diffusion Modelscited by 195
- 2016Stochastic Optimization for Large-scale Optimal Transportcited by 188
- 2019Learning Likelihoods with Conditional Normalizing Flowscited by 170
- 2020OT-Flow: Fast and Accurate Continuous Normalizing Flows via Optimal Transportcited by 143
- 2021Do Neural Optimal Transport Solvers Work? A Continuous Wasserstein-2 Benchmarkcited by 95
- 2021Generative Modeling with Optimal Transport Mapscited by 80
- 2023Score-based Diffusion Models in Function Spacecited by 80
- 2021Plugin Estimation of Smooth Optimal Transport Mapscited by 67
- 2015Bayesian Optimization for Likelihood-Free Inference of Simulator-Based Statistical Modelscited by 61
- 2018Likelihood-free inference with emulator networkscited by 61
- 2018Deep Bayesian Inversioncited by 60
- 2022Supervised Training of Conditional Monge Mapscited by 60
- 2021Score-based Generative Neural Networks for Large-Scale Optimal Transportcited by 59
- 2018Conditional Density Estimation with Bayesian Normalising Flowscited by 54
- 2021Rates of Estimation of Optimal Transport Maps using Plug-in Estimators via Barycentric Projectionscited by 52
- 2022Generative Adversarial Neural Operatorscited by 51
- 2022Diffusion Generative Models in Infinite Dimensionscited by 49
- 2019Coupling techniques for nonlinear ensemble filteringcited by 48
- 2020Conditional Sampling with Monotone GANs: from Generative Models to Likelihood-Free Inferencecited by 40
- 2023Conditional score-based diffusion models for Bayesian inference in infinite dimensionscited by 37
- 2014Dimension-independent likelihood-informed MCMCcited by 36
- 2023The Monge Gap: A Regularizer to Learn All Transport Mapscited by 32
- 2022Kernel Neural Optimal Transportcited by 31
- 2022On amortizing convex conjugates for optimal transportcited by 30
- 2019Quantitative stability of optimal transport maps and linearization of the 2-Wasserstein spacecited by 25
- 2022Reliable amortized variational inference with physics-based latent distribution correctioncited by 25
- 2021Wasserstein Generative Learning of Conditional Distributioncited by 24
- 2022Optimal transport map estimation in general function spacescited by 20
- 2014The Geometric Foundations of Hamiltonian Monte Carlocited by 19
- 2019Subspace Detours: Building Transport Plans that are Optimal on Subspace Projectionscited by 19
- 2021Preconditioned training of normalizing flows for variational inference in inverse problemscited by 19
- 2023Efficient Neural Network Approaches for Conditional Optimal Transport with Applications in Bayesian Inferencecited by 19
- 2022On minimax density estimation via measure transportcited by 18
50 most cited of 66 All 66 (JSON)
Built on by
Citing papers per year: 2023: 1; 2024: 8; 2025: 10; 2026: 7.
26 papers.
Sort:
By area:
- 2023How Smooth Is Attention?cited by 44
- 2024Conditional Wasserstein Distances with Applications in Bayesian OT Flow Matchingcited by 34
- 2024Dynamic Conditional Optimal Transport through Simulation-Free Flowscited by 24
- 2025Flow Matching: Markov Kernels, Stochastic Processes and Transport Planscited by 16
- 2025Flow Matching with Semidiscrete Couplingscited by 12
- 2024Conditional simulation via entropic optimal transport: Toward non-parametric estimation of conditional Brenier mapscited by 9
- 2025Probabilistic operator learning: generative modeling and uncertainty quantification for foundation models of differential equationscited by 9
- 2025Fast Flow-based Visuomotor Policies via Conditional Optimal Transport Couplingscited by 8
- 2024Understanding the training of infinitely deep and wide ResNets with Conditional Optimal Transportcited by 7
- 2024Taming Score-Based Diffusion Priors for Infinite-Dimensional Nonlinear Inverse Problemscited by 7
- 2025Preconditioned Langevin Dynamics with Score-Based Generative Models for Infinite-Dimensional Linear Bayesian Inverse Problemscited by 5
- 2025Fast filtering of non-Gaussian models using Amortized Optimal Transport Mapscited by 4
14 more
- 2024Sequential Conditional Transport on Probabilistic Graphs for Interpretable Counterfactual Fairnesscited by 3
- 2024Learning conditional distributions on continuous spacescited by 2
- 2025A Geometric Approach to Optimal Experimental Designcited by 2
- 2026Error Analysis of Bayesian Inverse Problems with Generative Priorscited by 2
- 2024Sequential transport maps using SoS density estimation and α-divergencescited by 1
- 2025Pairwise Optimal Transports for Training All-to-All Flow-Based Condition Transfer Modelcited by 1
- 2025Geometric Autoencoder Priors for Bayesian Inversion: Learn First Observe Latercited by 1
- 2026Conditional Sampling via Wasserstein Autoencoders and Triangular Transportcited by 1
- 2026LatentFlow: A General Framework for Conditioning Stochastic Processescited by 1
- 2025Estimation of Stochastic Optimal Transport Mapscited by 0
- 2026Conditional Unbalanced Optimal Transport Maps: An Outlier-Robust Framework for Conditional Generative Modelingcited by 0
- 2026Optimal Transport Q-Learning for Flow Policy Steering and Accelerationcited by 0
- 2026Approximating Measures on Function Spaces: Transport and Truncationcited by 0
- 2026Improving Function Space Flow Matching with Kernel Optimal Transportcited by 0
- 2026Improving Function Space Flow Matching with Kernel Optimal Transport29 Sep 2026
- 2026Approximating Measures on Function Spaces: Transport and Truncation15 Sep 2026
- 2026LatentFlow: A General Framework for Conditioning Stochastic Processes14 Jul 2026
- 2026Optimal Transport Q-Learning for Flow Policy Steering and Acceleration7 Jul 2026
- 2026Conditional Sampling via Wasserstein Autoencoders and Triangular Transport3 Apr 2026
- 2026Conditional Unbalanced Optimal Transport Maps: An Outlier-Robust Framework for Conditional Generative Modeling7 Mar 2026
- 2026Error Analysis of Bayesian Inverse Problems with Generative Priors24 Jan 2026
- 2025Estimation of Stochastic Optimal Transport Maps10 Dec 2025
- 2025A Geometric Approach to Optimal Experimental Design16 Oct 2025
- 2025Flow Matching with Semidiscrete Couplings29 Sep 2025
- 2025Geometric Autoencoder Priors for Bayesian Inversion: Learn First Observe Later24 Sep 2025
- 2025Probabilistic operator learning: generative modeling and uncertainty quantification for foundation models of differential equations5 Sep 2025
14 more
- 2025Preconditioned Langevin Dynamics with Score-Based Generative Models for Infinite-Dimensional Linear Bayesian Inverse Problems23 May 2025
- 2025Fast Flow-based Visuomotor Policies via Conditional Optimal Transport Couplings2 May 2025
- 2025Pairwise Optimal Transports for Training All-to-All Flow-Based Condition Transfer Model4 Apr 2025
- 2025Fast filtering of non-Gaussian models using Amortized Optimal Transport Maps16 Mar 2025
- 2025Flow Matching: Markov Kernels, Stochastic Processes and Transport Plans28 Jan 2025
- 2024Conditional simulation via entropic optimal transport: Toward non-parametric estimation of conditional Brenier maps11 Nov 2024
- 2024Sequential Conditional Transport on Probabilistic Graphs for Interpretable Counterfactual Fairness6 Aug 2024
- 2024Learning conditional distributions on continuous spaces13 Jun 2024
- 2024Taming Score-Based Diffusion Priors for Infinite-Dimensional Nonlinear Inverse Problems24 May 2024
- 2024Dynamic Conditional Optimal Transport through Simulation-Free Flows5 Apr 2024
- 2024Conditional Wasserstein Distances with Applications in Bayesian OT Flow Matching27 Mar 2024
- 2024Understanding the training of infinitely deep and wide ResNets with Conditional Optimal Transport19 Mar 2024
- 2024Sequential transport maps using SoS density estimation and α-divergences27 Feb 2024
- 2023How Smooth Is Attention?22 Dec 2023
- 2024Conditional Wasserstein Distances with Applications in Bayesian OT Flow Matchingcited by 34
- 2025Flow Matching: Markov Kernels, Stochastic Processes and Transport Planscited by 16
- 2025Flow Matching with Semidiscrete Couplingscited by 12
- 2025Fast Flow-based Visuomotor Policies via Conditional Optimal Transport Couplingscited by 8
- 2024Taming Score-Based Diffusion Priors for Infinite-Dimensional Nonlinear Inverse Problemscited by 7
- 2025Preconditioned Langevin Dynamics with Score-Based Generative Models for Infinite-Dimensional Linear Bayesian Inverse Problemscited by 5
- 2026Improving Function Space Flow Matching with Kernel Optimal Transportcited by 0
- 2026Improving Function Space Flow Matching with Kernel Optimal Transport29 Sep 2026
- 2025Flow Matching with Semidiscrete Couplings29 Sep 2025
- 2025Preconditioned Langevin Dynamics with Score-Based Generative Models for Infinite-Dimensional Linear Bayesian Inverse Problems23 May 2025
- 2025Fast Flow-based Visuomotor Policies via Conditional Optimal Transport Couplings2 May 2025
- 2025Flow Matching: Markov Kernels, Stochastic Processes and Transport Plans28 Jan 2025
- 2024Taming Score-Based Diffusion Priors for Infinite-Dimensional Nonlinear Inverse Problems24 May 2024
- 2024Conditional Wasserstein Distances with Applications in Bayesian OT Flow Matching27 Mar 2024
- 2023How Smooth Is Attention?cited by 44
- 2024Dynamic Conditional Optimal Transport through Simulation-Free Flowscited by 24
- 2024Conditional simulation via entropic optimal transport: Toward non-parametric estimation of conditional Brenier mapscited by 9
- 2025Probabilistic operator learning: generative modeling and uncertainty quantification for foundation models of differential equationscited by 9
- 2024Understanding the training of infinitely deep and wide ResNets with Conditional Optimal Transportcited by 7
- 2025Fast filtering of non-Gaussian models using Amortized Optimal Transport Mapscited by 4
- 2024Sequential Conditional Transport on Probabilistic Graphs for Interpretable Counterfactual Fairnesscited by 3
- 2024Learning conditional distributions on continuous spacescited by 2
- 2025A Geometric Approach to Optimal Experimental Designcited by 2
- 2026Error Analysis of Bayesian Inverse Problems with Generative Priorscited by 2
- 2024Sequential transport maps using SoS density estimation and α-divergencescited by 1
- 2025Pairwise Optimal Transports for Training All-to-All Flow-Based Condition Transfer Modelcited by 1
12 most cited of 18 All 18 (JSON)
- 2026Approximating Measures on Function Spaces: Transport and Truncation15 Sep 2026
- 2026LatentFlow: A General Framework for Conditioning Stochastic Processes14 Jul 2026
- 2026Conditional Sampling via Wasserstein Autoencoders and Triangular Transport3 Apr 2026
- 2026Conditional Unbalanced Optimal Transport Maps: An Outlier-Robust Framework for Conditional Generative Modeling7 Mar 2026
- 2026Error Analysis of Bayesian Inverse Problems with Generative Priors24 Jan 2026
- 2025Estimation of Stochastic Optimal Transport Maps10 Dec 2025
- 2025A Geometric Approach to Optimal Experimental Design16 Oct 2025
- 2025Geometric Autoencoder Priors for Bayesian Inversion: Learn First Observe Later24 Sep 2025
- 2025Probabilistic operator learning: generative modeling and uncertainty quantification for foundation models of differential equations5 Sep 2025
- 2025Pairwise Optimal Transports for Training All-to-All Flow-Based Condition Transfer Model4 Apr 2025
- 2025Fast filtering of non-Gaussian models using Amortized Optimal Transport Maps16 Mar 2025
- 2024Conditional simulation via entropic optimal transport: Toward non-parametric estimation of conditional Brenier maps11 Nov 2024
12 newest of 18 All 18 (JSON)