English

Pessimistic Risk-Aware Policy Learning in Contextual Bandits

Machine Learning 2026-05-18 v1 Machine Learning

Abstract

We study risk-aware offline policy learning, aiming to learn a decision rule from logged data that is optimal under general risk criteria. This problem is crucial in high-stakes domains where online interaction is infeasible and adverse outcomes must be carefully controlled. However, existing literature on offline contextual bandits either centers on expected-reward criteria or restricts risk considerations to policy evaluation instead of optimization. In this work, we propose a unified distributional framework for optimizing Lipschitz-continuous risk functionals, a broad class of risk measures encompassing mean-variance, entropic risk, and conditional value-at-risk, among others. By developing novel empirical concentration inequalities for importance sampling-based distributional estimators, our analysis derives data-dependent suboptimality bounds with an O~(1/n)\tilde{\mathcal{O}}(1/\sqrt{n}) rate, without relying on restrictive uniform overlap assumptions. This rate is minimax optimal and matches that of risk-neutral offline policy optimization, indicating that optimizing general Lipschitz risk criteria incurs no additional statistical cost relative to the expected-reward.

Keywords

Cite

@article{arxiv.2605.15620,
  title  = {Pessimistic Risk-Aware Policy Learning in Contextual Bandits},
  author = {Yilong Wan and Yuqiang Li and Xianyi Wu},
  journal= {arXiv preprint arXiv:2605.15620},
  year   = {2026}
}
R2 v1 2026-07-22T07:13:43.458Z