Generalized unnormalized optimal transport and its fast algorithms
Generalized unnormalized optimal transport and its fast algorithms
复制标题
DOI:
10.1016/j.jcp.2020.110041
复制
发表时间:
2021-03-26
影响因子:
4.1
通讯作者:
Osher, Stanley
中科院分区:
文献类型:
--
作者:
Lee, Wonjun;Lai, Rongjie;Osher, Stanley
We introduce fast algorithms for generalized unnormalized optimal transport. To handle densities with different total mass, we consider a dynamic model, which mixes the Lpoptimal transport with Lpdistance. For p = 1, we derive the corresponding L1generalized unnormalized Kantorovich formula. We further show that the problem becomes a simple L-1 minimization which is solved efficiently by a primal-dual algorithm. For p = 2, we derive the L2generalized unnormalized Kantorovich formula, a new unnormalized Monge problem and the corresponding Monge-Ampere equation. Furthermore, we introduce a new unconstrained optimization formulation of the problem. The associated gradient flow is essentially related to an elliptic equation which can be solved efficiently. Here the proposed gradient descent procedure together with the Nesterov acceleration involves the HamiltonJacobi equation arisingfrom the KKT conditions. Several numerical examples are presented to illustrate the effectiveness of the proposed algorithms. (c) 2020 Elsevier Inc. All rights reserved.