Unbalanced Optimal Total Variation Transport: A Theoretical Approach to Spatial Resource Allocation Problems
Nhan-Phu Chung, Jinhui Han, Bohan Li, Zehao Li
Abstract
We propose and analyze a new class of unbalanced weak optimal transport (OT) problems with total variation penalties, motivated by spatial resource allocation tasks. Unlike classical OT, our framework accommodates general unbalanced nonnegative measures and incorporates cost objectives that directly capture operational trade-offs between transport cost and supply-demand mismatch. In the general setting, we establish the existence of optimal solutions and a dual formulation. We then focus on the semi-discrete setting, where one measure is discrete and the other is absolutely continuous, a structure relevant to applications such as service area partitioning for facilities like schools or medical stations. Exploiting a tessellation-based structure, we derive the corresponding explicit optimality conditions. We further address a quantization problem that jointly optimizes the locations and weights of discrete support points, applicable to facility location tasks such as the cost-efficient deployment of battery swap stations or e-commerce warehouses, informed by demand-side data. The dual-tessellation structure also yields explicit gradient expressions, enabling efficient numerical optimization in finite dimensions.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers1
Ask how each one uses itBuilds on15
- Unbalanced minibatch Optimal Transport; applications to Domain AdaptationKilian Fatras, Thibault Séjourné, Rémi Flamary, Nicolas CourtyICML 2021 · 183 citations
- Robust Optimal Transport with Applications in Generative Modeling and Domain AdaptationYogesh Balaji, Rama Chellappa, Soheil FeiziNeurIPS 2020 · 141 citations
- Unified Optimal Transport Framework for Universal Domain AdaptationWanxing Chang, Ye Shi, Hoang Tuan, Jingya WangNeurIPS 2022 · 118 citations
- On Unbalanced Optimal Transport: An Analysis of Sinkhorn AlgorithmKhiem Pham, Khang Le, Nhat Ho, Tung Pham et al.ICML 2020 · 104 citations
- Rates of Estimation of Optimal Transport Maps using Plug-in Estimators via Barycentric ProjectionsNabarun Deb, Promit Ghosal, Bodhisattva SenNeurIPS 2021 · 96 citations
Related papers
- Submodular framework for structured-sparse optimal transportPiyushi Manupriya, Pratik Jawanpuria, Karthik S. Gurumoorthy, Saketha Nath Jagarlapudi et al.ICML 2024 · 1 citation
- UOTIP: Unbalanced Optimal Transport Map for Unpaired Inverse ProblemsDonggyu Lee, Taekyung Lee, Jaewoong ChoiICML 2026
- A Combinatorial Algorithm for the Semi-Discrete Optimal Transport ProblemPankaj K. Agarwal, Sharath Raghvendra, Pouyan Shirzadian, Keegan YaoNeurIPS 2024 · 4 citations
- Semi-Dual Unbalanced Quadratic Optimal Transport: fast statistical rates and convergent algorithmAdrien Vacher, François-Xavier VialardICML 2023 · 11 citations
- Light Unbalanced Optimal TransportMilena Gazdieva, Arip Asadulaev, Evgeny Burnaev, Aleksandr KorotinNeurIPS 2024 · 9 citations
