Conic Programming Reformulations of Two-Stage Distributionally Robust Linear Programs over Wasserstein Balls 2: With Q = 0 and Ξ = ℝ^K, the 1-Wasserstein Problem Is the Linear Program (32)Research Paper
Two-stage decisions under an ambiguous distribution
Two-stage stochastic linear programs model decisions taken in two steps: a first-stage decision is fixed before an uncertain parameter is observed, and a second-stage (recourse) decision adapts to afterwards. They underlie capacity planning, inventory and network design. In practice the distribution of is known only through samples . Distributionally robust optimization replaces the unknown distribution by the worst case over an ambiguity set; a popular choice is a Wasserstein ball around the empirical distribution, because it contains the true distribution with high probability and yields out-of-sample guarantees (Mohajerin Esfahani and Kuhn, 2018).
Hanasusanto and Kuhn (arXiv:1609.07505, Operations Research 66(3), 2018) study when the resulting two-stage problem can be solved exactly. Section 3 of their paper gives a copositive reformulation for the 2-Wasserstein ball (the subject of a companion mission). Section 4, the subject of this mission, shows that when the uncertainty enters only the constraints of the recourse problem and the ball is a 1-Wasserstein ball with a polyhedral reference distance, the whole problem is a linear program.
Setting
Let and be the first-stage cost and feasible set, the second-stage cost, the recourse matrix, and , affine functions of . The recourse value at is
The uncertainty affects only the constraints (in the paper's notation, ), and there is no support constraint: . The problem has sufficiently expensive recourse if some satisfies .
For scaling parameters the reference distance is with
Moving a unit of mass in the positive direction of coordinate costs , in the negative direction . The 1-Wasserstein ball consists of the probability measures on whose optimal transport cost to under is at most . The worst-case expectation and the two-stage problem (1) are
The dual norm of (31) is .
Formalization targets
Goal: Theorem 6
For and every , equals the optimal value of the linear program
and consequently problem (1) has the optimal value of (32) minimized jointly over and the other variables with added to the objective.
Milestones, in the order of the paper's proof
- Theorem 1 in this instance (, gauge cost, ): equals the moment problem (5) and, for , the dual
- Display (33): .
- The dual norm of (31): .
- The constraint decomposition: the semi-infinite constraint of (33) holds if and only if, for each , there are satisfying the four linear inequalities of (32).
Significance
Theorem 6 shows that the two-stage distributionally robust linear program over a 1-Wasserstein ball, which is a priori an optimization over infinitely many distributions, has a linear-programming reformulation with blocks of variables and constraints. It covers distributionally robust least-absolute-deviation regression (Example 2 of the paper), where the reformulation becomes the empirical loss plus a dual-norm regularizer, and it contrasts with the paper's Theorem 7, which shows that for -norm reference distances with the same problem is NP-hard.
The result is proved in the paper; there is no machine-checked proof. A formalization adds a verified chain from the measure-theoretic worst-case expectation to a finite linear program, including the conventions the paper leaves implicit: the extended-real values ( may be , and (32) may then be infeasible), the restriction , and the first-stage cost in the objective.
Difficulty
Each step of the proof is classical, but the chain crosses several regimes. Theorem 1 is a Wasserstein strong duality result, and here it is applied to a loss that is convex, possibly on part of , and not upper semicontinuous, with a cost that is not symmetric; the standard published duality theorems assume an upper semicontinuous real-valued loss or a metric. The passage to (33) interchanges a supremum over with an infimum over the dual-norm ball and must handle the case where the dual recourse polyhedron is unbounded. Finally, every equivalence must hold in the extended reals: an argument that tacitly assumes finiteness proves a statement that fails exactly when the dual polyhedron has a recession direction.
Formalization scope
Vectors are Fin n → ℝ, samples are ξhat : Fin I → (Fin K → ℝ) with , and carries its product (Borel) σ-algebra. and are affine maps. The transport cost is the platform definition RWPI.SqrtLasso.transportCost (couplings with first marginal , second ), the empirical distribution is WassersteinDRO.Duality.empiricalDistribution, and expectations of the extended-real recourse value use WassersteinDRO.Duality.erealExpectation (value whenever the positive part has infinite integral). Every optimal value is an extended-real infimum or supremum over its feasible set; no Bochner integral and no real-valued sInf/sSup is used.
Conventions made explicit, with reasons:
- . The paper does not restrict in Theorem 6, but its proof starts from the dual (6), stated only for , and the identity is false at when the dual recourse polyhedron has a recession direction with : then (32) is infeasible while is the finite sample average.
- in (32). The printed objective of (32) omits the first-stage cost. The proof establishes the pointwise identity value of (32) at , which is the goal's first part; the problem-level equivalence restores .
- Asymmetric gauge. For , (31) is not a norm, and Theorem 1 of the paper is stated for a metric. The paper applies it to (31) anyway; the milestone states that application, with the gauge cost in the orientation . The case is not imposed.
- Finite first moment. Membership in is implied by finite transport cost to and is not repeated.
- The two "" factors in the paper's proof are misprints for .
A formalization in which is defined by (6), (33) or (32), in which the gauge is secretly symmetric, or in which expectations are Bochner integrals (zero for non-integrable functions) would make the goal trivial or different; the definitions here exclude each of these.
A complete development needs Wasserstein strong duality for a lower semicontinuous loss and cost (with a disintegration of couplings to pass to the moment problem (5)), LP strong duality and Farkas' lemma in the extended-real form, and a minimax interchange over a compact dual-norm ball. Wasserstein duality for general costs and extended-real LP duality are reusable well beyond this mission; contributions of either are welcome.
Selected references
- G. A. Hanasusanto, D. Kuhn, Conic Programming Reformulations of Two-Stage Distributionally Robust Linear Programs over Wasserstein Balls, Operations Research 66(3), 2018. Pinned version: arXiv:1609.07505v3. DOI 10.1287/opre.2017.1698.
- P. Mohajerin Esfahani, D. Kuhn, Data-driven distributionally robust optimization using the Wasserstein metric: performance guarantees and tractable reformulations, Mathematical Programming 171, 2018. doi:10.1007/s10107-017-1172-1
- J. Blanchet, K. Murthy, Quantifying distributional model risk via optimal transport, Mathematics of Operations Research 44(2), 2019. doi:10.1287/moor.2018.0936
- R. Gao, A. Kleywegt, Distributionally robust stochastic optimization with Wasserstein distance, Mathematics of Operations Research 48(2), 2023. doi:10.1287/moor.2022.1275
- D. P. Bertsekas, Convex Optimization Theory, Athena Scientific, 2009 (Proposition 5.5.4, the minimax theorem used in the proof).