Papers
Topics
Authors
Recent
Search
2000 character limit reached

Wasserstein Distributionally Robust Regret Optimization

Updated 4 July 2026
  • Wasserstein DRRO is a robust decision framework that employs Wasserstein ambiguity sets and regret measures to evaluate performance against the best possible action under uncertainty.
  • It encompasses various formulations including single-stage linear problems, static ex-ante stochastic optimization, control systems, and RLHF, each distinguished by its timing and geometry of regret benchmarking.
  • Research in DRRO highlights computational challenges and tractable cases, emphasizing its potential to align robust optimization with relative performance metrics.

Searching arXiv for papers on Wasserstein distributionally robust regret optimization and closely related formulations. arxiv_search(query="Wasserstein distributionally robust regret optimization", max_results=10) arxiv_search(query="Distributionally Robust Regret Minimization Wasserstein", max_results=10) Wasserstein Distributionally Robust Regret Optimization (DRRO) denotes a family of minimax decision models in which ambiguity about a probability law is represented by a Wasserstein ball, while performance is measured by regret rather than raw loss or cost. The literature now contains several non-equivalent variants: single-stage ex post regret for linear objective uncertainty over type-1 Wasserstein balls; ex-ante regret over type-pp Wasserstein balls in static stochastic optimization; finite- and infinite-horizon control formulations over type-2 Wasserstein disturbance laws; and an RLHF formulation in which ambiguity is imposed on reward functions and reduced to promptwise 1\ell_1 perturbations (Bitar, 2024, Fiechtner et al., 15 Apr 2025, Taha et al., 2023, Hajar et al., 2023, Kargin et al., 2024, Yan et al., 13 Aug 2025, Wang et al., 30 Apr 2026). Across these variants, the defining feature is that the adversarial distribution is evaluated relative to the best action or policy for that same distribution, which separates DRRO from standard Wasserstein DRO.

1. Scope, terminology, and model classes

The expression “Wasserstein DRRO” does not identify a single canonical optimization problem. It covers a collection of models that share two ingredients: a Wasserstein ambiguity set and a regret benchmark. What changes across papers is the timing of the benchmark, the uncertainty object, and the decision class.

Setting Regret benchmark Ambiguity geometry
Single-stage linear decision problem realized cost vs perfect-information minimum type-1 Wasserstein ball around P0P_0
Static ex-ante stochastic optimization expected loss vs distribution-specific Bayes optimum Bδp(P0)B_\delta^p(\mathbb{P}_0)
Finite- or infinite-horizon control causal policy vs optimal noncausal controller type-2 Wasserstein ball on disturbance laws
RLHF promptwise model policy value vs best policy under same perturbed reward 1-Wasserstein / promptwise 1\ell_1 ambiguity

This variation matters because ex post and ex ante regret are distinct objects. In the single-stage linear formulation, regret is realized after the uncertain coefficients are revealed; in the ex-ante static formulation, regret is defined as the gap in expected loss relative to the best decision for the same adversarial distribution; in control, regret is usually measured against a clairvoyant noncausal controller; and in RLHF, regret is measured against the best policy under the same plausible reward perturbation (Bitar, 2024, Fiechtner et al., 15 Apr 2025, Taha et al., 2023, Wang et al., 30 Apr 2026).

A recurring misconception is that replacing worst-case risk by worst-case regret is only a cosmetic change. A conceptually related non-Wasserstein learning-theoretic result shows that worst-case risk need not imply uniformly small regret under distribution shift: in the paper’s counterexample, supPQRegP(fDRO)=0.21\sup_{P\in\mathcal Q}\mathrm{Reg}_P(f_{\mathrm{DRO}})=0.21 whereas supPQRegP(fMRO)=0.03\sup_{P\in\mathcal Q}\mathrm{Reg}_P(f_{\mathrm{MRO}})=0.03 (Agarwal et al., 2022). This supports the central DRRO viewpoint that robustification should be applied to relative performance, not merely to absolute loss.

2. Canonical mathematical formulations

A basic ex post Wasserstein DRRO model is the single-stage linear decision problem with feasible set XRn\mathcal X\subseteq\mathbb R^n, uncertain objective coefficients wRnw\in\mathbb R^n, and realized cost wxw^\top x. The regret of choosing 1\ell_10 after 1\ell_11 is realized is

1\ell_12

With a type-1 Wasserstein ball

1\ell_13

the distributionally robust regret problem is

1\ell_14

Here 1\ell_15 may be any probability measure with finite first moment, including an empirical distribution obtained from data (Bitar, 2024).

A distinct ex-ante static formulation defines regret at the distribution level. For loss 1\ell_16, decision 1\ell_17, and type-1\ell_18 Wasserstein ball

1\ell_19

the worst-case ex-ante regret is

P0P_00

and Wasserstein DRRO is P0P_01. The same paper distinguishes this from ex-post regret,

P0P_02

and notes that ex-post regret does not recover ERM when P0P_03, whereas ex-ante regret does (Fiechtner et al., 15 Apr 2025).

These formulations are related but not interchangeable. In the ex post linear model, the comparator depends on the realized scenario P0P_04. In ex-ante DRRO, the comparator is the distribution-specific optimizer P0P_05. In RLHF, the ex-ante comparator is a policy P0P_06 maximizing value under the same perturbed reward distribution, and the problem becomes

P0P_07

The formal similarity hides different tractability mechanisms and different geometric effects (Wang et al., 30 Apr 2026).

3. Regularization, geometry, and limiting behavior

A central single-stage result states that if P0P_08 is nonempty and compact, then for any P0P_09,

Bδp(P0)B_\delta^p(\mathbb{P}_0)0

Since Bδp(P0)B_\delta^p(\mathbb{P}_0)1 differs from Bδp(P0)B_\delta^p(\mathbb{P}_0)2 only by an Bδp(P0)B_\delta^p(\mathbb{P}_0)3-independent constant, the problem is equivalent to

Bδp(P0)B_\delta^p(\mathbb{P}_0)4

The induced regularizer is the radius of the smallest dual-norm ball centered at Bδp(P0)B_\delta^p(\mathbb{P}_0)5 that contains Bδp(P0)B_\delta^p(\mathbb{P}_0)6, so increasing the Wasserstein radius pulls solutions toward a feasible center rather than toward the origin. The same geometric term appears in worst-case CVaR of regret: Bδp(P0)B_\delta^p(\mathbb{P}_0)7 The paper also notes that this term coincides with the Lipschitz modulus of Bδp(P0)B_\delta^p(\mathbb{P}_0)8 with respect to the ground norm. By contrast, standard Wasserstein DRO for raw linear cost yields Bδp(P0)B_\delta^p(\mathbb{P}_0)9, which pulls toward the origin (Bitar, 2024).

Recent ex-ante analysis shows a different but related limiting picture. Under smoothness and regularity assumptions, the DRRO optimal value 1\ell_10 satisfies

1\ell_11

If the ERM solution set 1\ell_12 is a singleton, then 1\ell_13, so DRRO coincides with ERM up to first-order terms. In the convex quadratic case with Euclidean transport, the equivalence is exact for all radii: if

1\ell_14

then the DRRO-optimal policy is

1\ell_15

for every 1\ell_16, exactly as in ERM, and the regret is

1\ell_17

This establishes an exact ERM/DRRO coincidence in that class (Fiechtner et al., 15 Apr 2025).

These two strands are complementary. The ex post linear theory makes the regularization term explicit and geometric. The ex-ante static theory identifies when Wasserstein DRRO does not move the optimizer at first order, and when it never moves it at all.

4. Tractability, exact solvability, and hardness

Single-stage ex post DRRO is not uniformly easy. In general, evaluating

1\ell_18

is a norm maximization problem, and the paper remarks that such problems are NP-hard in general when 1\ell_19 is a supPQRegP(fDRO)=0.21\sup_{P\in\mathcal Q}\mathrm{Reg}_P(f_{\mathrm{DRO}})=0.210-norm with supPQRegP(fDRO)=0.21\sup_{P\in\mathcal Q}\mathrm{Reg}_P(f_{\mathrm{DRO}})=0.211 and supPQRegP(fDRO)=0.21\sup_{P\in\mathcal Q}\mathrm{Reg}_P(f_{\mathrm{DRO}})=0.212 is a compact convex polytope in halfspace form. Two tractable cases are identified. If supPQRegP(fDRO)=0.21\sup_{P\in\mathcal Q}\mathrm{Reg}_P(f_{\mathrm{DRO}})=0.213, then

supPQRegP(fDRO)=0.21\sup_{P\in\mathcal Q}\mathrm{Reg}_P(f_{\mathrm{DRO}})=0.214

and DRRO becomes a finite-dimensional convex program; with polyhedral norms it is an LP, with Euclidean norm an SOCP. If supPQRegP(fDRO)=0.21\sup_{P\in\mathcal Q}\mathrm{Reg}_P(f_{\mathrm{DRO}})=0.215, the regularizer admits a support-function reformulation valid for any nonempty compact supPQRegP(fDRO)=0.21\sup_{P\in\mathcal Q}\mathrm{Reg}_P(f_{\mathrm{DRO}})=0.216, yielding another finite-dimensional convex program when the support function is computable (Bitar, 2024).

For ex-ante Wasserstein DRRO, the computational picture is sharper. In the scalar newsvendor with loss supPQRegP(fDRO)=0.21\sup_{P\in\mathcal Q}\mathrm{Reg}_P(f_{\mathrm{DRO}})=0.217, the paper proves that the regret function supPQRegP(fDRO)=0.21\sup_{P\in\mathcal Q}\mathrm{Reg}_P(f_{\mathrm{DRO}})=0.218 is concave on supPQRegP(fDRO)=0.21\sup_{P\in\mathcal Q}\mathrm{Reg}_P(f_{\mathrm{DRO}})=0.219 and on supPQRegP(fMRO)=0.03\sup_{P\in\mathcal Q}\mathrm{Reg}_P(f_{\mathrm{MRO}})=0.030. Hence

supPQRegP(fMRO)=0.03\sup_{P\in\mathcal Q}\mathrm{Reg}_P(f_{\mathrm{MRO}})=0.031

so exact regret evaluation reduces to maximizing two one-dimensional concave functions. This extends Wasserstein newsvendor tractability to the regret setting (Fiechtner et al., 15 Apr 2025).

Outside special structures, exact ex-ante computation is hard. For empirical reference distributions and max-affine losses, the same paper proves that evaluating supPQRegP(fMRO)=0.03\sup_{P\in\mathcal Q}\mathrm{Reg}_P(f_{\mathrm{MRO}})=0.032 is NP-hard in general, even when supPQRegP(fMRO)=0.03\sup_{P\in\mathcal Q}\mathrm{Reg}_P(f_{\mathrm{MRO}})=0.033, the ground norm is supPQRegP(fMRO)=0.03\sup_{P\in\mathcal Q}\mathrm{Reg}_P(f_{\mathrm{MRO}})=0.034, and the loss contains no bilinear supPQRegP(fMRO)=0.03\sup_{P\in\mathcal Q}\mathrm{Reg}_P(f_{\mathrm{MRO}})=0.035-supPQRegP(fMRO)=0.03\sup_{P\in\mathcal Q}\mathrm{Reg}_P(f_{\mathrm{MRO}})=0.036 cross terms. The source of hardness is the supremum over the benchmark decision supPQRegP(fMRO)=0.03\sup_{P\in\mathcal Q}\mathrm{Reg}_P(f_{\mathrm{MRO}})=0.037, not the Wasserstein worst-case expectation alone. To address this, the paper proposes a convex relaxation obtained by replacing bilinear products supPQRegP(fMRO)=0.03\sup_{P\in\mathcal Q}\mathrm{Reg}_P(f_{\mathrm{MRO}})=0.038 with auxiliary variables supPQRegP(fMRO)=0.03\sup_{P\in\mathcal Q}\mathrm{Reg}_P(f_{\mathrm{MRO}})=0.039 and perspective constraints. The relaxation is upper bounded by ex-post regret, and the paper reports that it improves over recent alternatives while remaining convex and broadly computable (Fiechtner et al., 15 Apr 2025).

A plausible implication is that Wasserstein DRRO inherits the classical tractability of Wasserstein DRO only when the regret benchmark preserves convex structure. Once the comparator is itself optimized against the same adversarial law, exact solvability becomes substantially more delicate.

5. Control formulations

Control-theoretic Wasserstein DRRO is organized around linear-quadratic systems and regret relative to a noncausal benchmark. In the finite-horizon full-information formulation, the system is

XRn\mathcal X\subseteq\mathbb R^n0

the cost is quadratic, and the regret of a strictly causal linear disturbance-feedback controller XRn\mathcal X\subseteq\mathbb R^n1 is

XRn\mathcal X\subseteq\mathbb R^n2

where XRn\mathcal X\subseteq\mathbb R^n3 is the optimal noncausal controller with perfect knowledge of the disturbance trajectory. With a type-2 Wasserstein ball around a nominal law XRn\mathcal X\subseteq\mathbb R^n4 on the stacked disturbance trajectory, the minimax expected-regret problem admits an exact semidefinite reformulation, and the worst-case distribution is the push-forward of XRn\mathcal X\subseteq\mathbb R^n5 through the linear map XRn\mathcal X\subseteq\mathbb R^n6 (Taha et al., 2023).

Under partial observability, the benchmark cannot be a universally pointwise-optimal noncausal controller, and the measurement-feedback formulation instead uses the noncausal controller minimizing the Frobenius norm of the closed-loop transfer operator. The resulting finite-horizon Wasserstein DRRO-MF problem combines Wasserstein quadratic duality with a Youla-type parameterization and a suboptimal Nehari approximation, yielding an SDP for fixed XRn\mathcal X\subseteq\mathbb R^n7 whose size is proportional to the time horizon (Hajar et al., 2023).

The infinite-horizon formulation replaces finite stacked disturbances by a stationary disturbance process and studies

XRn\mathcal X\subseteq\mathbb R^n8

Here the ambiguity set allows time correlation. In the stationary limit, the problem becomes a saddle-point problem over covariance operators constrained by a Bures-Wasserstein ball,

XRn\mathcal X\subseteq\mathbb R^n9

The optimal controller is causal, LTI, and stabilizing, but generally non-rational; the paper nevertheless shows that it is characterized by a finite-dimensional parameter and computes it through a frequency-domain Frank–Wolfe method with an wRnw\in\mathbb R^n0 rate up to discretization error (Kargin et al., 2024).

A later output-feedback formulation uses purified outputs wRnw\in\mathbb R^n1 and places a type-2 Wasserstein ball on the joint law of initial state, disturbances, and measurement noise. The paper derives strong duality results for general quadratic objectives, then eliminates first wRnw\in\mathbb R^n2 and wRnw\in\mathbb R^n3, and later the structured feedback variable wRnw\in\mathbb R^n4, yielding a lower-dimensional SDP and an equivalent distributed optimization reformulation (Yan et al., 13 Aug 2025).

A closely related multistage ex-ante DRRO model studies finite-horizon LQR under a Gelbrich ball on the common stage-law moments. That paper states that the Gelbrich distance is a lower bound on squared wRnw\in\mathbb R^n5-Wasserstein distance and coincides with it for elliptical distributions, in particular Gaussian laws. This suggests a moment-based Wasserstein-type relaxation rather than a full Wasserstein ball, but the resulting SDP and the strictly causal empirical-mean correction it identifies are directly relevant to multistage regret robustness (Fiechtner et al., 7 Apr 2026).

6. RLHF and reward misspecification

Wasserstein DRRO has also been formulated for reinforcement learning from human feedback, where the reward model is treated as a proxy for an unobserved true human-preference reward. In that setting, a policy wRnw\in\mathbb R^n6 is evaluated under a distribution wRnw\in\mathbb R^n7 over reward functions through

wRnw\in\mathbb R^n8

and ex-ante DRRO is

wRnw\in\mathbb R^n9

A key simplification is that wxw^\top x0 depends on wxw^\top x1 only through the mean reward wxw^\top x2. Starting from a 1-Wasserstein ambiguity set over reward-function distributions,

wxw^\top x3

the paper shows that wxw^\top x4, and then specializes to a promptwise ambiguity model

wxw^\top x5

For a fixed prompt with response simplex variable wxw^\top x6, proxy reward vector wxw^\top x7, and perturbation wxw^\top x8, the promptwise problem becomes

wxw^\top x9

The paper proves the exact inner solution

1\ell_100

with an adversary that can place all budget on a single coordinate. It then derives a water-filling optimal policy: 1\ell_101 where 1\ell_102 is defined through the threshold equations in the paper. As 1\ell_103, the solution approaches the greedy policy 1\ell_104; as 1\ell_105, it approaches the uniform distribution (Wang et al., 30 Apr 2026).

The same work translates the exact promptwise theory into a PPO/GRPO-style algorithm. In the hard version, the “dangerous” sampled response receives a bonus 1\ell_106; in the soft version, the bonus is weighted by self-normalized importance weights. A dynamic ambiguity budget,

1\ell_107

is motivated through a Donsker–Varadhan bound on reward misspecification. In the reported experiment, the main methods achieved the following peak held-out gold rewards: PPO 1\ell_108, GRPO 1\ell_109, DRO-RLHF 1\ell_110, DRRO-RLHF (hard) 1\ell_111, and DRRO-RLHF (soft + dynamic) 1\ell_112; the corresponding peak proxy rewards were 1\ell_113, 1\ell_114, 1\ell_115, 1\ell_116, and 1\ell_117, with peak KL values 1\ell_118, 1\ell_119, 1\ell_120, 1\ell_121, and 1\ell_122 (Wang et al., 30 Apr 2026).

The direct statistical theory for Wasserstein DRRO is still fragmented, but adjacent Wasserstein-DRO results provide reusable tools. A general asymptotic analysis of Wasserstein DRO shows that if regret can be treated as a loss function 1\ell_123, then the robust objective

1\ell_124

admits the standard dual reformulation

1\ell_125

The same work proposes a radius-calibration rule 1\ell_126 for squared transport cost, more precisely 1\ell_127, based on an optimal-transport projection argument. It also interprets Wasserstein robustness as adaptive variation regularization, a perspective that transfers naturally whenever regret is sufficiently smooth (Blanchet et al., 2021).

A residuals-based conditional Wasserstein DRO framework with covariates shows how a Wasserstein ball can be centered at an 1\ell_128-specific residual-generated empirical distribution,

1\ell_129

with radius 1\ell_130. This is not a regret paper, but it provides an explicit template for contextual Wasserstein DRRO by replacing cost with a benchmark-relative regret loss (Kannan et al., 2020).

Distributed Wasserstein DRO supplies another reusable template. In that setting, a samplewise loss 1\ell_131 leads to the finite-dimensional robust objective

1\ell_132

together with per-sample adversarial variables and a saddle-point dynamics. This is not specifically regret-based, but it becomes directly relevant when regret is expressible as a samplewise loss (Cherukuri et al., 2017).

By contrast, online DRO over time with shrinking ambiguity sets develops dynamic regret bounds of order 1\ell_133, but its main theory is not Wasserstein-based. It is best viewed as a template for online DRRO with evolving ambiguity rather than as a direct Wasserstein DRRO result (Aigner et al., 2023).

The current boundaries of Wasserstein DRRO are therefore sharp. Exact structure is available for single-stage linear ex-post regret, scalar newsvendor, quadratic control, and promptwise RLHF with 1\ell_134 ambiguity. General max-affine ex-ante Wasserstein DRRO is NP-hard (Fiechtner et al., 15 Apr 2025). Several control results rely critically on linear-quadratic structure and quadratic Wasserstein duality (Taha et al., 2023, Hajar et al., 2023, Kargin et al., 2024, Yan et al., 13 Aug 2025). The RLHF formulation depends critically on promptwise 1\ell_135 geometry for its coordinate-local adversary and sampled-bonus interpretation (Wang et al., 30 Apr 2026). A reasonable summary is that Wasserstein DRRO is now a well-defined research area, but not yet a uniform methodology: its most explicit results remain highly structure-dependent.

Topic to Video (Beta)

No one has generated a video about this topic yet.

Whiteboard

No one has generated a whiteboard explanation for this topic yet.

Follow Topic

Get notified by email when new papers are published related to Wasserstein Distributionally Robust Regret Optimization (DRRO).