Wasserstein Distributionally Robust Optimization II: The Gelbrich Ambiguity Set and Elliptical TractabilityTextbook
Motivation
Distributionally robust optimization (DRO) hedges a decision against every distribution within some ambiguity set around an estimated (nominal) distribution, rather than trusting the estimate exactly. When the ambiguity set is a ball of radius around the empirical distribution in the type- Wasserstein metric, the resulting worst-case risk problem inherits attractive statistical guarantees (Mohajerin Esfahani & Kuhn 2018) but is, in general, an optimization problem over an infinite-dimensional space of measures. Kuhn, Mohajerin Esfahani, Nguyen & Shafieezadeh-Abadeh's 2019 INFORMS TutORials chapter surveys when this problem becomes computationally tractable. One route — the subject of this mission — discards everything about the nominal distribution except its mean vector and covariance matrix and replaces the Wasserstein ball with a set built only from these two moments, the Gelbrich hull. The construction is due to Gelbrich (1990), who first bounded the Wasserstein distance between two distributions using only their means and covariances.
Setting
Fix , a nominal distribution , a radius and an exponent . The type- Wasserstein distance between two probability measures on is
the infimum over couplings (probability measures on with marginals and ) of the -th root of the expected -th power of Euclidean distance. The Wasserstein ambiguity set is , and the worst-case risk of a loss function is .
Suppose has mean vector and covariance matrix (the positive semidefinite matrices). The mean-covariance uncertainty set is
where is the positive-semidefinite square root. The Gelbrich hull is : the distributions on whose own mean and covariance lie in . An elliptical distribution has density for a density generator and normalizing constant ; two elliptical distributions "have the same density generator" when their coincide (e.g. both Gaussian, both Student- for the same ).
Formalization targets
Goal (Theorem 13, Gelbrich hull). For every ,
This is an outer approximation: every distribution within of in Wasserstein distance has a mean and covariance inside , so optimizing over the Gelbrich hull instead of the Wasserstein ball can only enlarge the feasible set, never shrink it below the truth.
Supporting results. Theorem 4 (Gelbrich bound) gives the moment-only lower bound on that Theorem 13 is built from, with equality for elliptical distributions sharing a generator. Proposition 1 sharpens the goal's containment to an equality on the mean-covariance projection itself, under the same two conditions (, elliptical). Corollary 1 propagates the goal's set containment to the risk level: for every , where is the Gelbrich risk.
Significance
Theorem 13 is the hinge between an intractable infinite-dimensional worst-case-risk problem and a tractable one: the paper goes on (Theorem 16, outside this mission's scope) to show that for quadratic loss functions and elliptical nominal distributions the Gelbrich risk itself equals the optimal value of a semidefinite program with two linear matrix inequality constraints — and that, under those same conditions, the Wasserstein worst-case risk, the Gelbrich risk and the SDP value all coincide. Corollary 1 is what makes the Gelbrich risk usable as a conservative surrogate even outside that special case: it upper-bounds the true worst-case risk for any loss function and any , at the cost of discarding all but first- and second-order information about the nominal distribution. Formalizing the goal and Corollary 1 gives the exact scope in which this moment-relaxation is licensed — the restriction and the outer-approximation direction are both easy to get backwards, and this mission's Lean encoding fixes both irreversibly.
Difficulty
The obvious first argument is to prove containment pointwise: fix and show its mean and covariance land in . That reduces Theorem 13 to Proposition 1's containment half, which in turn reduces to the Gelbrich bound (Theorem 4) applied to the pair — the inequality direction of Theorem 4 suffices for containment; only the sharper equality direction (needed for Proposition 1's own equality clause) requires the elliptical hypothesis. The non-obvious step is Theorem 4 itself: bounding below by a closed-form expression in the two distributions' first two moments only, for arbitrary with those moments, requires an argument that survives every coupling — the paper's proof goes through a lower bound on the coupling's cross-covariance term via the eigenvalues of , not a direct manipulation of 's definition.
Formalization scope
is EuclideanSpace ℝ (Fin m); a "distribution" is a MeasureTheory.Measure on it
constrained by Q Set.univ = 1 (probability) and Q Ξᶜ = 0 (support in Ξ). The Wasserstein
distance is ENNReal-valued (Definition 1's infimum over couplings, matching 01-duality's
convention); the worst-case and Gelbrich risks are EReal-valued suprema restricted to loss
functions integrable under the candidate distribution, avoiding Mathlib's junk value for a
non-integrable Bochner integral. is the positive-semidefinite matrix square root,
picked by choice from its defining existential and applied in this mission only to matrices
hypothesized (or, per Section 2.3's standing assumption, given) positive semidefinite. Elliptical
distributions (IsElliptical) are represented by the paper's own density formula (a measure
equal to volume.withDensity of C·det(Σ)⁻¹·g((ξ-μ)ᵀΣ⁻¹(ξ-μ)) for some C>0), together with
the mean/covariance facts every theorem in this chunk reads off directly; an earlier draft kept
only the latter, under which "same density generator" held vacuously for any moment-matched
pair — corrected after moderation flagged it (see STATUS.md).
Because 01-duality (Wasserstein distance, ambiguity set, worst-case risk) is not yet a published
mission, this chunk redefines those objects locally in its own namespace rather than importing an
unpublished draft, per the series' definition-reuse policy; a future upload can retire the
duplication once 01-duality is live. A formalization that dropped Theorem 4's "same density
generator" condition from its equality clause, or that stated the goal's containment for all rather than , would be trivializing or simply false — both are explicit hypotheses in
the Lean statements. Matrix.PosSemidef and its Loewner order carry the constraints; no
elliptical-distribution or Gelbrich-hull infrastructure exists elsewhere on the platform, so this
mission's definitions are original contributions reusable by any later extension (Theorem 16/17,
Lemma 1/2's SDP representations) of this series.
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. https://doi.org/10.1287/educ.2019.0198
- Gelbrich, M. (1990). On a formula for the L2 Wasserstein metric between measures on Euclidean and Hilbert spaces. Mathematische Nachrichten, 147(1), 185–203.
- Mohajerin Esfahani, P., & Kuhn, D. (2018). Data-driven distributionally robust optimization using the Wasserstein metric: performance guarantees and tractable reformulations. Mathematical Programming, 171(1), 115–166.