Wasserstein Distributionally Robust Optimization I: Kantorovich Duality and Strong Duality for the Worst-Case RiskTextbook
Motivation
Every data-driven decision problem faces the same trap. A decision-maker estimates a risk functional from a nominal distribution built from training samples, then optimizes a loss function against instead of the unknown true distribution . Because the optimizer adapts to the noise in , the in-sample risk of the optimizer systematically understates its true, out-of-sample risk — a phenomenon Smith and Winkler named the optimizer's curse (Smith & Winkler, Management Science, 2006). The remedy explored here is to hedge against a whole neighborhood of plausible distributions around , rather than trusting the point estimate. Kuhn, Mohajerin Esfahani, Nguyen and Shafieezadeh-Abadeh's INFORMS TutORials chapter (2019) develops this neighborhood using the Wasserstein distance, and the present mission formalizes its foundational duality theory: the machinery every later result in the chapter (finite-sample guarantees, elliptical tractability, regularization) builds on.
Setting
Fix a norm on a finite-dimensional real vector space (representing ). For , the type- Wasserstein distance between two Borel probability measures on is
where is the set of couplings of and — joint probability measures on whose marginals are and . The optimal can be read as a transportation plan moving one pile of dirt () into another () at minimum cost, which is why is also called the earth mover's distance; the underlying linear program was formalized by Kantorovich (1942) after Monge's 1781 original.
Given training samples , the empirical distribution is . Centered at , the Wasserstein ambiguity set of radius is
where is a closed set known to contain the support of the true distribution. The worst-case risk of a loss function is
and minimizing it over a class of admissible loss functions is a distributionally robust optimization problem. measures the estimation error one insures against; a larger ambiguity set gives a more conservative (and more expensive) guarantee.
Formalization targets
Goal — Theorem 7, strong duality
This is the Lagrangian dual of the worst-case risk evaluation problem, with the multiplier of the Wasserstein constraint : it converts a supremum over an infinite-dimensional space of measures into a one-dimensional minimization of the Moreau-Yosida regularization . Every tractability result later in the chapter (finite convex reformulations, SDP relaxations) specializes this duality by choosing a loss class for which is computable.
Supporting dual representations of — Theorems 1 and 2
These identify as a linear program's strong dual (Theorem 1) and, for , specialize it to the Kantorovich-Rubinstein form (Theorem 2), which is what lets the worst-case-risk analysis reason about Lipschitz loss functions directly.
Upper and lower bounds — Theorems 5 and 6
These are the tractable, easily-computed bracket that Theorems 7 and 10 later show is tight in important special cases.
Exact case — Theorem 10
Theorem 5's inequality becomes exact under convexity — the cleanest closing corollary of the duality theory, obtained from Theorem 7 by evaluating the Moreau-Yosida regularization of a convex function explicitly.
Significance
Theorem 7 is the hinge on which the entire computational program of Wasserstein distributionally robust optimization turns: every tractable reformulation in the source chapter (piecewise-concave losses via conic duality, quadratic losses via semidefinite programming, the shrinkage-estimator connection) is obtained by substituting a specific loss class into the right-hand side of Theorem 7 and showing the resulting Moreau-Yosida regularization is computable. Kuhn et al. themselves derive it as a corollary of Blanchet & Murthy (2019) and Gao & Kleywegt (2016) for the empirical case, generalized to Polish spaces by Blanchet & Murthy and Gao & Kleywegt independently — the paper cites [12] and [37] for the general statement. Formalizing it is what makes every later, more computational result in the chapter — the ones a solver is more likely to reach for next — rest on a mechanically verified foundation rather than a citation chain.
Status. The mathematical result is well established (multiple independent published proofs cited above); nothing here is open research. What this mission contributes is the first machine-checked formal statement of the duality theorem and its supporting dual representations (Theorems 1, 2, 5, 6, 10) on the Prove2Me platform — none of Wp's dual representation, the Wasserstein ambiguity set, or the worst-case risk functional exist there prior to this mission (see Formalization scope).
Difficulty
The obvious proof strategy — write down the Lagrangian of the semi-infinite program (6), swap the order of the outer supremum over and the inner minimization over the multiplier , and invoke ordinary Lagrangian strong duality — fails because (6) is an infinite- dimensional linear program over measures, not a finite convex program: there is no compact feasible set or Slater point in a form that ordinary finite-dimensional duality applies to directly. The actual proof goes through the dual representation of the Wasserstein distance itself (Theorem 1, which is why it is a prerequisite milestone), reformulating the constraint via its own dual variables and swapping the resulting sup-inf using minimax theorems for semi-infinite programs, not ordinary Lagrangian duality for finite programs.
Formalization scope
is a generic finite-dimensional real normed space (NormedAddCommGroup, NormedSpace ℝ,
Borel-measurable), representing with the paper's arbitrary fixed norm as a
parameter rather than fixing the Euclidean norm. A coupling is formalized directly via
MeasureTheory.Measure.map: π.map Prod.fst = Q ∧ π.map Prod.snd = Q'. Constrained
infima/suprema (over couplings, over the ambiguity set, over Lipschitz test functions, over
perturbation matrices) use Mathlib's guarded-binder idiom ⨅ x (_ : P x), f x, which
correctly returns (resp. ) outside the feasible set rather than a finite junk
value.
Two deliberate, disclosed conventions keep the extremal-value definitions faithful without
extended-real integration machinery, both recorded in MODERATION_NOTES.md:
worstCaseRiskand the dual representations (Theorems 1, 2) are valued inEReal, notℝ, so an unbounded supremum is recorded as rather than collapsed to Mathlib's real-valued junk value0on an unbounded family.- The goal theorem (7) and its Moreau-Yosida regularization restrict the loss function to
bounded continuous (
BoundedContinuousFunction E ℝ), narrower than the paper's general upper-semicontinuous, -integrable loss class (Assumption 1). This keeps a finite real number for every nonempty , so the right-hand side's Bochner integral is well-posed; the milestones (Theorems 5, 6, 10) keep the more general real-valued (not necessarily bounded) loss class, since their statements do not require evaluating a pointwise supremum over . Ξis required closed in Theorems 5, 6 and 7, matching the paper's own standing assumption (p. 6: "we let be a closed set that is known to contain the support of ") for the whole worst-case-risk framework, which is used silently in the paper wherever a theorem takes as an argument but was not carried into these theorems' own hypothesis lists in an earlier draft.- The goal theorem (7) additionally requires itself supported on
(, the same "supported on " convention
ambiguitySetuses for ), which the paper's framework presupposes for the nominal distribution throughout §2. Combined with bounded, this makes bounded on the full-measure set (above by unconditionally, below by itself via for ), which is what makes the right-hand side's integral genuinely well-posed rather than liable to Mathlib's non-integrable junk value .
There is no trivializing formalization risk from a vacuous hypothesis: Ξ.Nonempty and
0 < N are both required exactly where the paper's own indexing and support assumptions
require them, and every extremal value uses the extended-real convention above rather than a
convention that would make an inequality vacuously true.
No definition in this mission exists on the platform prior to this series (GET /theorems?q=Wasserstein, q=Kantorovich, q=optimal transport, q=coupling return only
unrelated discrete/finite-type constructions); all seven definitions and six theorems are
drafted fresh. WassersteinDRO.Duality.wassersteinDistance, .ambiguitySet and
.worstCaseRisk are the substrate every later mission in this five-part series (Gelbrich
tractability, finite-sample guarantees, regularization, shrinkage estimation) either imports
directly or redefines locally per the series' reuse rule.
Selected references
- Kuhn, D., Mohajerin Esfahani, P., Nguyen, V. A., & Shafieezadeh-Abadeh, S. (2019). Wasserstein Distributionally Robust Optimization: Theory and Applications in Machine Learning. INFORMS TutORials in Operations Research, 130–166. https://doi.org/10.1287/educ.2019.0198
- Villani, C. (2009). Optimal Transport: Old and New. Springer. (Cited as [108] for Theorems 1 and 2.)
- Smith, J. E., & Winkler, R. L. (2006). The optimizer's curse: Skepticism and postdecision surprise in decision analysis. Management Science, 52(3), 311–322. https://doi.org/10.1287/mnsc.1050.0451
- Gao, R., & Kleywegt, A. J. (2016). Distributionally Robust Stochastic Optimization with Wasserstein Distance. arXiv:1604.02199.
- Blanchet, J., & Murthy, K. (2019). Quantifying Distributional Model Risk via Optimal Transport. Mathematics of Operations Research, 44(2), 565–600. https://doi.org/10.1287/moor.2018.0936