Saved in:
Bibliographic Details
Main Authors: Wu, Tianhao, Cheng, Qihao, Wang, Zihao, Zhang, Chaorui, Bai, Bo, Huang, Zhongyi, Wu, Hao
Format: Preprint
Published: 2022
Subjects:
Online Access:https://arxiv.org/abs/2212.00354
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866913686837264384
author Wu, Tianhao
Cheng, Qihao
Wang, Zihao
Zhang, Chaorui
Bai, Bo
Huang, Zhongyi
Wu, Hao
author_facet Wu, Tianhao
Cheng, Qihao
Wang, Zihao
Zhang, Chaorui
Bai, Bo
Huang, Zhongyi
Wu, Hao
contents 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. Extensive numerical experiments demonstrate that our proposed method has a significant advantage in terms of accuracy, efficiency, and memory consumption compared with existing methods.
format Preprint
id arxiv_https___arxiv_org_abs_2212_00354
institution arXiv
publishDate 2022
record_format arxiv
spellingShingle The Double Regularization Method for Capacity Constrained Optimal Transport
Wu, Tianhao
Cheng, Qihao
Wang, Zihao
Zhang, Chaorui
Bai, Bo
Huang, Zhongyi
Wu, Hao
Optimization and Control
49M25, 65K10
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. Extensive numerical experiments demonstrate that our proposed method has a significant advantage in terms of accuracy, efficiency, and memory consumption compared with existing methods.
title The Double Regularization Method for Capacity Constrained Optimal Transport
topic Optimization and Control
49M25, 65K10
url https://arxiv.org/abs/2212.00354