Volume 4, Issue 4
The Double Regularization Method for Capacity Constrained Optimal Transport

Tianhao Wu, Qihao Cheng, Zihao Wang, Chaorui Zhang, Bo Bai, Zhongyi Huang & Hao Wu

CSIAM Trans. Appl. Math., 4 (2023), pp. 776-796.

Published online: 2023-10

Export citation
  • Abstract

Capacity constrained optimal transport is a variant of optimal transport, which adds extra constraints on the set of feasible couplings in the original optimal transport problem to limit the mass transported between each pair of source and sink. Based on this setting, constrained optimal transport has numerous applications, e.g., finance, network flow. However, due to the large number of constraints in this problem, existing algorithms are both time-consuming and space-consuming. In this paper, inspired by entropic regularization for the classical optimal transport problem, we introduce a novel regularization term for capacity constrained optimal transport. The regularized problem naturally satisfies the capacity constraints and consequently makes it possible to analyze the duality. Unlike the matrix-vector multiplication in the alternate iteration scheme for solving classical optimal transport, in our algorithm, each alternate iteration step is to solve several single-variable equations. Fortunately, we find that each of these equations corresponds to a single-variable monotonic function, and we convert solving these equations into finding the unique zero point of each single-variable monotonic function with Newton’s method. Theoretical analysis further provides a convergence guarantee to our algorithm. Extensive numerical experiments demonstrate that our proposed method has a significant advantage in terms of accuracy, efficiency, and memory consumption compared with existing methods.

  • AMS Subject Headings

49M25, 65K10

  • Copyright

COPYRIGHT: © Global Science Press

  • Email address
  • BibTex
  • RIS
  • TXT
@Article{CSIAM-AM-4-776, author = {Wu , TianhaoCheng , QihaoWang , ZihaoZhang , ChaoruiBai , BoHuang , Zhongyi and Wu , Hao}, title = {The Double Regularization Method for Capacity Constrained Optimal Transport}, journal = {CSIAM Transactions on Applied Mathematics}, year = {2023}, volume = {4}, number = {4}, pages = {776--796}, abstract = {

Capacity constrained optimal transport is a variant of optimal transport, which adds extra constraints on the set of feasible couplings in the original optimal transport problem to limit the mass transported between each pair of source and sink. Based on this setting, constrained optimal transport has numerous applications, e.g., finance, network flow. However, due to the large number of constraints in this problem, existing algorithms are both time-consuming and space-consuming. In this paper, inspired by entropic regularization for the classical optimal transport problem, we introduce a novel regularization term for capacity constrained optimal transport. The regularized problem naturally satisfies the capacity constraints and consequently makes it possible to analyze the duality. Unlike the matrix-vector multiplication in the alternate iteration scheme for solving classical optimal transport, in our algorithm, each alternate iteration step is to solve several single-variable equations. Fortunately, we find that each of these equations corresponds to a single-variable monotonic function, and we convert solving these equations into finding the unique zero point of each single-variable monotonic function with Newton’s method. Theoretical analysis further provides a convergence guarantee to our algorithm. Extensive numerical experiments demonstrate that our proposed method has a significant advantage in terms of accuracy, efficiency, and memory consumption compared with existing methods.

}, issn = {2708-0579}, doi = {https://doi.org/10.4208/csiam-am.SO-2022-0044}, url = {http://global-sci.org/intro/article_detail/csiam-am/22078.html} }
TY - JOUR T1 - The Double Regularization Method for Capacity Constrained Optimal Transport AU - Wu , Tianhao AU - Cheng , Qihao AU - Wang , Zihao AU - Zhang , Chaorui AU - Bai , Bo AU - Huang , Zhongyi AU - Wu , Hao JO - CSIAM Transactions on Applied Mathematics VL - 4 SP - 776 EP - 796 PY - 2023 DA - 2023/10 SN - 4 DO - http://doi.org/10.4208/csiam-am.SO-2022-0044 UR - https://global-sci.org/intro/article_detail/csiam-am/22078.html KW - Capacity constraint, optimal transport, regularization, Sinkhorn algorithm. AB -

Capacity constrained optimal transport is a variant of optimal transport, which adds extra constraints on the set of feasible couplings in the original optimal transport problem to limit the mass transported between each pair of source and sink. Based on this setting, constrained optimal transport has numerous applications, e.g., finance, network flow. However, due to the large number of constraints in this problem, existing algorithms are both time-consuming and space-consuming. In this paper, inspired by entropic regularization for the classical optimal transport problem, we introduce a novel regularization term for capacity constrained optimal transport. The regularized problem naturally satisfies the capacity constraints and consequently makes it possible to analyze the duality. Unlike the matrix-vector multiplication in the alternate iteration scheme for solving classical optimal transport, in our algorithm, each alternate iteration step is to solve several single-variable equations. Fortunately, we find that each of these equations corresponds to a single-variable monotonic function, and we convert solving these equations into finding the unique zero point of each single-variable monotonic function with Newton’s method. Theoretical analysis further provides a convergence guarantee to our algorithm. Extensive numerical experiments demonstrate that our proposed method has a significant advantage in terms of accuracy, efficiency, and memory consumption compared with existing methods.

Tianhao Wu, Qihao Cheng, Zihao Wang, Chaorui Zhang, Bo Bai, Zhongyi Huang & Hao Wu. (2023). The Double Regularization Method for Capacity Constrained Optimal Transport. CSIAM Transactions on Applied Mathematics. 4 (4). 776-796. doi:10.4208/csiam-am.SO-2022-0044
Copy to clipboard
The citation has been copied to your clipboard