Wasserstein Distributionally Robust Optimization V: The Wasserstein Shrinkage Estimator and Robust MMSE EstimationTextbook
Motivation
Minimum mean square error (MMSE) estimation — predicting a signal from a noisy observation by minimizing expected squared prediction error — underlies linear systems theory, linear regression, Kalman filtering, and multiple-input multiple-output signal processing. Its classical solution assumes the joint distribution of is known exactly; in practice it is estimated from data, and the estimator inherits sampling error and model risk. Kuhn, Mohajerin Esfahani, Nguyen & Shafieezadeh-Abadeh's 2019 INFORMS TutORials chapter shows that hedging the MMSE objective against every distribution in a Wasserstein ball around the empirical distribution — an infinite-dimensional worst case over an intractable set of measures, a priori — collapses to a tractable, finite-dimensional convex semidefinite program (Theorem 25, p. 29), building on the Gelbrich-hull machinery of Section 2.3. This mission formalizes that reduction.
Setting
Fix and let be a random vector: the signal to be estimated, the observation. An estimator is a measurable function ; write for the family of all estimators. The distribution of is only known to lie in a type-2 Wasserstein ball centered at an elliptical nominal distribution with nominal mean (), nominal covariance , and density generator . The distributionally robust MMSE estimation problem is
Writing blockwise, the nonlinear convex SDP
is the finite-dimensional relaxation the chapter builds toward.
Formalization targets
Goal (Theorem 25, distributionally robust MMSE estimator). If , then the optimal value of problem (35) equals the optimal value of SDP (36). Moreover, if is optimal in (36) with invertible, then the affine function
attains the outer infimum of (35) — it is a distributionally robust MMSE estimator, exhibited in closed form from an SDP optimizer.
Significance
Theorem 25 reduces an a priori infinite-dimensional, worst-case functional optimization problem (an infimum over all measurable estimators of a supremum over all distributions within a Wasserstein ball) to a finite convex program with one linear matrix inequality, one Loewner-order lower bound, and one trace/matrix-square-root constraint — solvable in polynomial time, with the optimal estimator recovered in closed form from the SDP's optimal block matrix. It shows that robustifying MMSE estimation against distributional ambiguity does not sacrifice tractability: the resulting estimator remains affine, the same functional form as the classical (non-robust) best linear unbiased estimator, only with its coefficients drawn from a regularized covariance estimate rather than the raw sample covariance. Formalizing it fixes, machine-checkably, the exact shape of that regularization — which SDP constraints are load-bearing (the Loewner lower bound in particular rules out a numerically unstable near-singular ) and which conditions (, invertible) the closed-form estimator formula actually needs.
Difficulty
The paper's own remark (p. 29) names the two nontrivial steps: first, "establishing a minimax theorem for (35) and exploiting the properties of elliptical distributions" to show the outer infimum is attained by an affine estimator — a priori (35) ranges over all measurable , and there is no obvious reason the worst case forces linearity. Second, "combining this structural insight with Theorem 16" (the SDP-representability result for indefinite quadratic losses under an elliptical nominal distribution, itself a nontrivial closed-form reduction of an infinite-dimensional worst-case risk) to convert the now-restricted problem over affine estimators into the finite SDP (36). Neither step is a routine consequence of the ambiguity-set definitions alone; each requires structural facts about elliptical distributions and quadratic losses proved earlier in the chapter.
Formalization scope
The signal-observation space is EuclideanSpace ℝ (Fin mx ⊕ Fin my), with and recovered
as the two summand projections; the block matrix is Matrix (Fin mx ⊕ Fin my) (Fin mx ⊕ Fin my) ℝ, and Matrix.toBlocks₁₁/toBlocks₁₂/toBlocks₂₁/toBlocks₂₂ give its four blocks. The
constraint "" is not stated as a separate hypothesis: it follows
automatically once is symmetric (implied by S.PosSemidef), so encoding the feasible set from
a single symmetric S rather than four independently-quantified blocks makes it structurally
impossible to drop — see pitfall 4 of BRIEF.md. The outer infimum of problem (35) ranges only
over measurable (Measurable ψ on the binder), matching the paper's own definition of
as "the family of all possible measurable estimators" (p. 29) exactly. λ_min(Σ̂) is taken
as a hypothesis parameter
characterized by the two properties that make it the minimum ("≤ every eigenvalue of Σ̂, and
attained by some eigenvalue"), rather than invoking a specific Mathlib min-eigenvalue API by name.
S_{yy}⁻¹ uses the ordinary matrix inverse (junk zero matrix when singular), matching the paper's
literal notation; S^\star_{yy} invertible is stated as an added hypothesis, not present
verbatim on the page, because the paper leaves the formula's well-definedness implicit — disclosed
per pitfall 5 rather than silently assumed away. The paper's own "which is always solvable"
clause is not asserted: Theorem 25 states, as part of itself, that SDP (36) attains its maximum
(an unconditional existence claim for an optimal ); this formalization states only the
conditional consequences of such an existing, not that one does — proving or asserting
solvability is out of this mission's scope, so the Lean statement is strictly weaker than Theorem
25's own conclusion on this point, disclosed rather than silently dropped. All risk-style suprema
are EReal-valued and
Integrable-guarded, matching the series' convention. No milestone theorem is included: the
paper's own proof sketch derives (33)'s and by extension (36)'s SDP "via Theorem 16", but Theorem
16 (indefinite quadratic loss and , eq. 23) was itself judged too heavy to state faithfully
in 02-gelbrich's time budget and is not redefined here either — see STATUS.md. Theorem 24 (the
Wasserstein shrinkage estimator, this chapter's originally recommended goal) is out of scope: its
closed-form eigenvalue transformation (eq. 34a/34b) requires transcribing nested square roots from
a rendered PDF page that this session's time budget did not allow verifying to the standard the
brief demands (pitfall 1); the brief's own documented fallback to Theorem 25 was taken instead.
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
- Nguyen, V. A., Shafieezadeh-Abadeh, S., Yue, M.-C., Kuhn, D., & Wiesemann, W. (2021). Optimistic distributionally robust optimization for nonparametric likelihood approximation. Advances in Neural Information Processing Systems, 32.
- Shafieezadeh-Abadeh, S., Nguyen, V. A., Kuhn, D., & Mohajerin Esfahani, P. (2018). Wasserstein distributionally robust Kalman filtering. Advances in Neural Information Processing Systems, 31.