Wasserstein Distributionally Robust Optimization IV: Regularization by Robustification in Classification and RegressionTextbook
Motivation
Regularization — adding a penalty on model complexity to an empirical risk minimization objective — is one of the oldest and most reliably effective tools in statistical learning: ridge regression, LASSO, and margin-based support vector machines all fit this template. For decades the regularization weight and penalty function were chosen heuristically or by cross-validation, with only asymptotic or worst-case generalization bounds explaining why they help. Kuhn, Mohajerin Esfahani, Nguyen & Shafieezadeh-Abadeh's 2019 INFORMS TutORials chapter on Wasserstein distributionally robust optimization (DRO) gives regularization a different, non-asymptotic justification: a decision rule that is robust to adversarial perturbations of its training data, measured in Wasserstein distance, is exactly a regularized empirical risk minimizer — no approximation, no asymptotics. This mission formalizes the general form of that equivalence, Theorem 10 (p. 14), which underlies every regularization-by-robustification result the chapter derives for classification and regression alike.
Setting
Fix a normed space , a convex loss function , a radius , and training samples (). The empirical distribution is . The worst-case risk of at radius (eq. (6), p. 6) is
where is the type- Wasserstein ball of radius around (Definition 1, p. 3) — the set of probability measures on within Wasserstein distance of the empirical distribution, under a norm on and its transportation exponent . The Lipschitz modulus (possibly ) measures how fast can grow.
Formalization targets
Goal (Theorem 10, convex loss and ). If is convex and , then the worst-case risk of over the type-1 Wasserstein ball around the empirical distribution coincides with the Lipschitz-regularized empirical loss:
where is the ordinary empirical risk. The left side is the worst-case expected loss under adversarial perturbations of the training data of bounded aggregate Wasserstein cost; the right side is the ordinary empirical risk plus a regularization term proportional to the perturbation radius and the loss's own Lipschitz modulus — no approximation, an exact equality for every convex .
Significance
This is the single result underlying every specific regularization-by-robustification corollary
the chapter derives — the -regularized SVM and its / variants,
regularized logistic regression, and the regression analogue with a Lipschitz loss (Proposition 3)
— because each of those is obtained by substituting a specific convex, Lipschitz loss (hinge,
logloss, a linear-model margin loss) for the general here and reading off
in closed form. Theorem 10 is exact (, Ξ = R^m) precisely because a
type-1 transportation cost is dual to the Lipschitz modulus (a consequence of the Kantorovich-
Rubinstein duality this book's earlier duality theorems establish), whereas the corresponding
statement for (Theorem 11 elsewhere in the chapter) needs a strictly stronger hypothesis
on . Formalizing Theorem 10 fixes, machine-checkably, the exact scope of the equivalence —
which losses qualify (convex, no boundedness or smoothness needed), which transportation exponent
is required (, not any ), and that the equality is exact, not an upper or lower bound —
the single fact every classification- and regression-specific robustification corollary in the
chapter cites without re-deriving.
Difficulty
The natural first attempt treats the worst-case risk as an instance of the general finite convex
reduction (Theorem 8, which needs or a related concave/convex-conjugate structure and
applies for a general norm exponent with ) and specializes it to . This is
not the paper's own route for Theorem 10: at the dual exponent , and the general
finite-convex-program reduction (11) degenerates in a way that is more naturally derived directly
from the type-1 Wasserstein distance's own dual (Kantorovich-Rubinstein) representation — a
Lipschitz test function pairs exactly with a type-1 transportation cost, which is why the answer is
Lipschitz-regularization and not some other penalty. The Lean statement records only the final
equality (the proof itself, via Theorem 8 or the direct Kantorovich-Rubinstein route, is left as
sorry, as required for a draft item), but the choice of which milestones would support that proof
is exactly this: the type-1/Lipschitz-modulus duality is a structurally different argument from the
finite-convex-program machinery Theorem 8 needs for other , which is why this chapter's earlier
draft mistakenly tried to reach a specialization of Theorem 10 (Proposition 2, restated for a
labeled, ambiguity set) through a finite-atom shortcut that does not actually
capture the general type-1 ball Theorem 10's own proof needs — see Formalization scope.
Formalization scope
is EuclideanSpace ℝ (Fin m) (fixed to Set.univ, matching the theorem's own "Ξ = R^m"
hypothesis) with an arbitrary fixed norm as a type-class parameter, not specialized to Euclidean.
The worst-case risk, Wasserstein distance, ambiguity set, nominal risk, empirical distribution and
Lipschitz modulus are all redefined locally in this chapter's own namespace, verbatim restatements
of 01-duality's own drafts of the same objects (identical EReal/ENNReal/Integrable
conventions), because 01-duality is not yet a published mission
(CAPTAIN_ADDENDUM_WAVE2.md rule 5); a later upload can retire the duplication once it is live.
hN : 0 < N excludes the empty-sample case the paper's own "N training samples" presupposes;
hℓ : Integrable ℓ (empiricalDistribution ξhat) guards the empirical risk's Bochner integral
(automatically satisfied under any real-valued ℓ since the empirical distribution is a finite
convex combination of Dirac masses, but stated explicitly rather than assumed silently). No
constant is pinned to a numeral. This mission originally targeted Proposition 2 (regularization
by robustification for classification, p. 25), a specialization of Theorem 10 to a labeled,
κ=∞ input-output ambiguity set — but that draft's κ=∞ ambiguity set restricted admissible
perturbations to a finite family of exactly N atoms, each sample's entire mass moving as one
indivisible unit, which is a strict, non-faithful subset of the true κ=∞ Wasserstein ball
(splitting a sample's mass across several destinations is a legitimate, and for a convex loss
strictly more powerful, perturbation by Jensen's inequality). This made the goal theorem false for
at least one of the three loss functions (Table 1) the mission itself certified as covered
(logloss), not merely narrower than the paper's claim — moderation caught this and required either
a faithful general-ambiguity-set reformulation or retargeting the goal. This mission takes the
latter: Theorem 10 itself, already correctly and generally stated using the unrestricted
Wasserstein ball throughout, is now the goal, with no further milestone (no other numbered result
in this chunk's scope is needed for Theorem 10's own statement beyond the shared definitions
above). A future mission in this series can revisit Proposition 2/3 with a properly general
labeled ambiguity set once that construction (label-pinned marginal plus an aggregate coupling
condition on the input coordinate, rather than a finite atom family) is built.
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
- Shafieezadeh-Abadeh, S., Esfahani, P. M., & Kuhn, D. (2015). Distributionally robust logistic regression. Advances in Neural Information Processing Systems, 28.
- Blanchet, J., Kang, Y., & Murthy, K. (2019). Robust Wasserstein profile inference and applications to machine learning. Journal of Applied Probability, 56(3), 830–857.