Dynamic Programming for Epistemic Uncertainty in Markov Decision Processes
Axel Benyamine, Julien Grand-Clément, Marek Petrik, Michael I. Jordan, Alain Durmus
TL;DR
This work develops a unified theory of MDPs under epistemic uncertainty by treating transition kernels as random and evaluating policy return with a risk measure. It extends the dynamic programming framework through the ambiguity-averse value $V^{\pi,\nu,\rho}$ and Bellman operators $T^{\pi,\nu,\rho}$ and $T^{\nu,\rho}$, proving contraction and monotonicity under suitable conditions. A key contribution is the complete characterization of law-invariant risk measures compatible with DP: for static kernels, $\rho$ must be $\operatorname{ess\,inf}$ or $\operatorname{ess\,sup}$, and for resampled kernels it can also be $\mathbb{E}^{\nu}$; with $W^1$-continuity, only $\mathbb{E}^{\nu}$ remains. The results illuminate when DP-based methods yield stationary optimal policies and where one must go beyond DP (e.g., augmented state spaces or nested risk measures) to capture broader risk criteria in AMDPs.
Abstract
In this paper, we propose a general theory of ambiguity-averse MDPs, which treats the uncertain transition probabilities as random variables and evaluates a policy via a risk measure applied to its random return. This ambiguity-averse MDP framework unifies several models of MDPs with epistemic uncertainty for specific choices of risk measures. We extend the concepts of value functions and Bellman operators to our setting. Based on these objects, we establish the consequences of dynamic programming principles in this framework (existence of stationary policies, value and policy iteration algorithms), and we completely characterize law-invariant risk measures compatible with dynamic programming. Our work draws connections among several variants of MDP models and fully delineates what is possible under the dynamic programming paradigm and which risk measures require leaving it.
