Globally Solving Unbalanced Optimal Transport and Density Control for Gaussian Distributions

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Nakashima, Haruto, Ganguly, Siddhartha, Kashima, Kenji
Format: Preprint
Published: 2026
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866913092033576960
author Nakashima, Haruto
Ganguly, Siddhartha
Kashima, Kenji
author_facet Nakashima, Haruto
Ganguly, Siddhartha
Kashima, Kenji
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2605_04246
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Globally Solving Unbalanced Optimal Transport and Density Control for Gaussian Distributions
Nakashima, Haruto
Ganguly, Siddhartha
Kashima, Kenji
Optimization and Control
Machine Learning
Robotics
Systems and Control
93E20, 93E03, 58E25, 65K10
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.
title Globally Solving Unbalanced Optimal Transport and Density Control for Gaussian Distributions
topic Optimization and Control
Machine Learning
Robotics
Systems and Control
93E20, 93E03, 58E25, 65K10
url https://arxiv.org/abs/2605.04246