Constrained Hellinger-Kantorovich barycenters: least-cost soft and conic multi-marginal formulations
Maciej Buze
TL;DR
This work addresses computing barycenters under the Hellinger-Kantorovich ($HK$) metric in unbalanced settings by introducing a constrained least-cost soft multi-marginal formulation with a one-sided hard marginal constraint. It then develops a conic multi-marginal reformulation based on a single joint perspective cost, and connects this constrained framework to unbalanced multi-marginal optimal transport through the notion of least cost. The approach is demonstrated with Dirac masses and Gaussian measures, illustrating numerical viability and highlighting the natural extension of Wasserstein barycenters to the unbalanced setting. Overall, the paper provides a principled, computationally tractable framework for least-cost HK barycenters that extends classical barycenters to unbalanced measures with potential wide-ranging applications.
Abstract
We show that the problem of finding the barycenter in the Hellinger-Kantorovich setting admits a least-cost soft multi-marginal formulation, provided that a one-sided hard marginal constraint is introduced. The constrained approach is then shown to admit a conic multi-marginal reformulation based on defining a single joint multi-marginal perspective cost function in the conic multi-marginal formulation, as opposed to separate two-marginal perspective cost functions for each two-marginal problem in the coupled-two-marginal formulation, as was studied previously in literature. We further establish that, as in the Wasserstein metric, the recently introduced framework of unbalanced multi-marginal optimal transport can be reformulated using the notion of the least cost. Subsequently, we discuss an example when input measures are Dirac masses and numerically solve an example for Gaussian measures. Finally, we also explore why the constrained approach can be seen as a natural extension of a Wasserstein space barycenter to the unbalanced setting.
