English

Near-Optimal Policy Identification in Robust Constrained Markov Decision Processes via Epigraph Form

Machine Learning 2026-04-27 v5 Optimization and Control

Abstract

Designing a safe policy for uncertain environments is crucial in real-world control systems. However, this challenge remains inadequately addressed within the Markov decision process (MDP) framework. This paper presents the first algorithm guaranteed to identify a near-optimal policy in a robust constrained MDP (RCMDP), where an optimal policy minimizes cumulative cost while satisfying constraints in the worst-case scenario across a set of environments. We first prove that the conventional policy gradient approach to the Lagrangian max-min formulation can become trapped in suboptimal solutions. This occurs when its inner minimization encounters a sum of conflicting gradients from the objective and constraint functions. To address this, we leverage the epigraph form of the RCMDP problem, which resolves the conflict by selecting a single gradient from either the objective or the constraints. Building on the epigraph form, we propose a bisection search algorithm with a policy gradient subroutine and prove that it identifies an ε\varepsilon-optimal policy in an RCMDP with O~(ε4)\tilde{\mathcal{O}}(\varepsilon^{-4}) robust policy evaluations.

Keywords

Cite

@article{arxiv.2408.16286,
  title  = {Near-Optimal Policy Identification in Robust Constrained Markov Decision Processes via Epigraph Form},
  author = {Toshinori Kitamura and Tadashi Kozuno and Wataru Kumagai and Kenta Hoshino and Yohei Hosoe and Kazumi Kasaura and Masashi Hamaya and Paavo Parmas and Yutaka Matsuo},
  journal= {arXiv preprint arXiv:2408.16286},
  year   = {2026}
}

Comments

This manuscript contains a technical error; the main result does not hold (see also arXiv:2604.21177 for a formal invalidation)

R2 v1 2026-06-28T18:27:18.895Z