mlsanityData to 7 Oct 2026

Globally Solving Unbalanced Optimal Transport and Density Control for Gaussian Distributions

arXiv 2605.04246 ↗First posted 5 May 2026math.OC cs.LG cs.ROcited by 1

Not on the map. Why?

Abstract (arXiv)

In this article, we study unbalanced optimal transport (UOT) and establish a control-theoretic dynamical extension, which we call the unbalanced density control (UDC), for a class of Gaussian reference measures. In the static setting, we consider UOT with quadratic transport cost and Kullback--Leibler penalties on the marginals relative to prescribed Gaussian measures. We show that the infinite-dimensional variational problem admits an exact Gaussian reduction, yielding a finite-dimensional optimization over masses, means, and covariances, together with a closed-form expression for the optimal transported mass. We then formulate UDC for discrete-time linear systems, where the initial and terminal state measures are imposed softly through KL penalties and the intermediate evolution is governed by controlled linear dynamics with quadratic control cost. For this problem, we prove that any feasible solution can be replaced, without loss of optimality, by a Gaussian initial measure and an affine-Gaussian control policy. This leads to an exact finite-dimensional reformulation and, after a standard covariance-steering lifting, to an SDP-based optimization for fixed mass, again coupled with a closed-form mass update. We further establish existence of optimal solutions and identify a sufficient condition under which the affine-Gaussian UDC policy is deterministic. These results provide globally optimal solution methods for both Gaussian UOT and Gaussian UDC. Finally, we illustrate our results with several numerical examples.

Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Builds on 18 of its 50 references that are in the map. First posted on arXiv 5 May 2026.

Builds on

18 of 50 references.

References · 18

  1. 2015Optimal Entropy-Transport problems and a new Hellinger-Kantorovich distance between positive measurescited by 138
  2. 2014On the relation between optimal transport and Schr"odinger bridges: A stochastic control viewpointcited by 99
  3. 2020On Unbalanced Optimal Transport: An Analysis of Sinkhorn Algorithmcited by 61
  4. 2022Unbalanced Optimal Transport, from Theory to Numericscited by 45
  5. 2021Unbalanced Optimal Transport through Non-negative Penalized Linear Regressioncited by 31
  6. 2014Optimal steering of a linear stochastic system to a final probability distributioncited by 22
  7. 2014Optimal steering of a linear stochastic system to a final probability distribution, part IIcited by 17
  8. 2021The most likely evolution of diffusing and vanishing particles: Schrodinger Bridges with unbalanced marginalscited by 14
  9. 2016Optimal steering of a linear stochastic system to a final probability distribution, Part IIIcited by 12
  10. 2022An Optimal Transport Perspective on Unpaired Image Super-Resolutioncited by 12
  11. 2023Light Unbalanced Optimal Transportcited by 8
  12. 2018Optimal Covariance Control for Stochastic Systems Under Chance Constraintscited by 7
6 more
  1. 2022Maximum entropy optimal density control of discrete-time linear systems and Schr"odinger bridgescited by 6
  2. 2024Optimal Transport on the Lie Group of Roto-translationscited by 4
  3. 2024Reachability and Controllability Analysis of the State Covariance for Linear Stochastic Systemscited by 3
  4. 2023Computationally Efficient Covariance Steering for Systems Subject to Parametric Disturbances and Chance Constraintscited by 2
  5. 2019Input Hard Constrained Optimal Covariance Steeringcited by 1
  6. 2022Exact SDP Formulation for Discrete-Time Covariance Steering with Wasserstein Terminal Costcited by 1

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 1.

1 paper.