Optimal Transport-Based Distributionally Robust Optimization: Structural Properties and Iterative Schemes 4: A Worst-Case Distribution Is the Law of X + √δ G A(X)⁻¹β, Spending the Whole Budget δResearch Paper
Why the worst case matters
Distributionally robust optimization (DRO) replaces the expected loss of a linear decision rule by its worst case over all data laws within transport distance of the baseline . When is an empirical distribution this covers regularized regression and classification (Blanchet, Kang, Murthy 2019; Esfahani, Kuhn 2018). The optimal value is one number; a practitioner also wants to know which distribution realizes it, the adversary's best response. It shows which data points the robust model guards against, it gives stress-test scenarios, and its structure is what the paper's comparative statics (Theorem 7) and iterative schemes build on.
J. Blanchet, K. Murthy and F. Zhang, Optimal Transport-Based Distributionally Robust Optimization: Structural Properties and Iterative Schemes (arXiv:1810.02403v3; Math. Oper. Res. 47(2), 2022), answer this for a state-dependent Mahalanobis cost. Theorem 6 of that paper describes the worst-case distribution through the solution of the one-dimensional dual problem. This mission formalizes Theorem 6.
Setting
The data take values in with law , a Borel probability measure. The decision ranges over a convex set (a standing assumption of the paper). Moving mass from to costs
where is positive definite. Assumption 1 asks that be lower semicontinuous and that for -almost every , with . The optimal transport cost is the least value of over couplings of and . For the worst-case objective is
Assumption 2 asks that the loss be convex, that the growth exponent be finite, and that .
The dual side is built from
the robust loss , the dual objective and the maximizer set . By the paper's Theorem 1, (6) equals , and a minimizer , the dual optimizer, exists. The threshold is ; below it .
Formalization targets
Goal: Theorem 6(c)
Let and . Let be measurable selections with the largest and smallest square, , , and let be independent of with . Then
and the law of attains the supremum in (6).
Milestones
- Lemma 1: and when , and below that; hence .
- Lemma 3(a): for once .
- Proposition 2 (a) and its -part: the one-sided -derivatives of are the minimum and the maximum of over .
- The one-sided -derivatives of on :
- [Blanchet–Murthy 2019, Theorem 1]: strong duality and complementary slackness (already proved on the platform).
Companion statements
Theorem 6(a): unless is constant. Theorem 6(b): . Theorem 6(d), as the paper's example: for , , and , and no distribution attains the value . Theorem 6(e), with Lemma 8: if and , then is a singleton almost surely and the law of is the unique worst case.
Significance
The result reduces a search over all probability measures to a scalar problem per data point. Each point is pushed along the single direction , by an amount read off the maximizers of , and the budget is always spent completely. When the maximizer is not unique, randomization between two extreme maximizers is needed and suffices. Part (d) shows that the threshold condition cannot be removed, and part (e) gives a deterministic transport map (a Monge solution) under smoothness. Theorem 7 of the paper, the monotone growth of worst-case displacements in , is a statement about the random variable produced here.
None of these statements has been machine-checked before. The general strong duality with complementary slackness of Blanchet–Murthy (arXiv:1604.01446) is proved on the platform, so this mission connects that abstract theorem to an explicit worst-case law. A formal proof also has to settle one point the printed text leaves loose: which extreme maximizers enter the mixture (see Formalization scope).
Difficulty
The first idea is to take the pointwise maximizers of at and push forward. That works only when the maximizer is unique and the budget constraint happens to be tight. In general the maximizer set is not a singleton, and the derivative of at only lies between a left and a right derivative. Turning the first-order condition into an exact cost needs three things. The one-sided derivatives must be identified as expectations over , an envelope theorem under an integral with a non-compact parameter set. The extreme maximizers must be compact and measurably selectable. Finally the resulting coupling must be certified optimal through complementary slackness. Each step needs a hypothesis that is easy to lose: the growth control of Lemma 3, the essential supremum in , measurability of selections.
Formalization scope
Points of are EuclideanSpace ℝ (Fin d), is the inner product, and is quadInv A β x. and are extended-real valued; uses the published extended integral ModelRiskOT.Duality.extIntegral. The worst-case objective (6) is the published primal problem primalValue: a supremum over couplings with first marginal and (primalFeasible). A distribution attains (6) when a coupling with that second marginal has value primalValue. The random worst case is encoded by the joint law of , a mixture of two graph couplings, because depends on the pair. A dual optimizer is a real minimizing on .
Standing hypotheses on every statement: a probability measure, Assumptions 1 and 2, , convex. The following departures from the printed text are disclosed in each item:
- In (c), are the selections of extreme square used in the proof (p. 40), not and as printed; a two-point example shows the printed choice can fail.
- In (e), "for every " is read as "for -almost every ", which is what the proof gives.
- Lemma 8 is stated for .
- Proposition 2 is formalized for part (a) and the -derivatives only.
Every expectation used in the goal (, ) carries an integrability conjunct. The cost identity and optimality are asserted for the measure itself, so no junk value of a non-integrable integral can make the goal hold vacuously. A statement asserting only first-order conditions for would not be Theorem 6. The goal asserts primal attainment and the exact cost.
Needed infrastructure, reusable beyond this mission: envelope (Danskin) theorems for one-sided derivatives of suprema, differentiation of convex integral functionals, measurable selection of argmax sets, and complementary slackness through the published Blanchet–Murthy theorem. Contributions of any of these as standalone lemmas are welcome.
Selected references
- J. Blanchet, K. Murthy, F. Zhang, Optimal Transport-Based Distributionally Robust Optimization: Structural Properties and Iterative Schemes, Math. Oper. Res. 47(2), 2022. arXiv:1810.02403v3, doi:10.1287/moor.2021.1178
- J. Blanchet, K. Murthy, Quantifying Distributional Model Risk via Optimal Transport, Math. Oper. Res. 44(2), 2019. arXiv:1604.01446
- J. Blanchet, Y. Kang, K. Murthy, Robust Wasserstein Profile Inference and Applications to Machine Learning, J. Appl. Probab. 56(3), 2019. doi:10.1017/jpr.2019.49
- P. Mohajerin Esfahani, D. Kuhn, Data-driven distributionally robust optimization using the Wasserstein metric, Math. Program. 171, 2018. doi:10.1007/s10107-017-1172-1
- D. P. Bertsekas, S. E. Shreve, Stochastic Optimal Control: The Discrete-Time Case, Academic Press, 1978 (Proposition 7.50, measurable selection).