Efficient Value Iteration for s-rectangular Robust Markov Decision Processes
Navdeep Kumar, Kaixin Wang, Kfir Yehuda Levy, Shie Mannor
Abstract
We focus on s-rectangular robust Markov decision processes (MDPs), which capture interconnected uncertainties across different actions within each state. This framework is more general compared to sa-rectangular robust MDPs, where uncertainties in each action are independent. However, the introduced interdependence significantly amplifies the complexity of the problem. Existing methods either have slow performance guarantees or are inapplicable to even moderately large state spaces. In this work, we derive optimal robust Bellman operators in explicit forms. This leads to robust value iteration methods with significantly faster time complexities than existing approaches, which can be used in large state spaces. Further, our findings reveal that the optimal policies demonstrate a novel threshold behavior, selectively favoring a limited set of actions based on their respective advantage functions. Additionally, our study uncovers a noteworthy connection between the robustness of a policy and the variance in its value function, highlighting that policies with lower variance exhibit greater resilience. Method LP SDP Bisection Gradient Close form Algorithm Binary Search Solution Approx Approx Exact Approx Exact Exact Approx Complexity Õ S 11 2 A 9 2 S 11 2 + S 3 A S 3 A NA S 2 A S 2 A S 2 A Contraction factor γ γ γ γ(1+ √ Sβ * ) γ γ γ Valid Kernels Yes Yes Yes No Yes Yes Yes Optimal Policy Characterization No No No No Yes Yes Yes SDP and LP stand for semi-definite programming and linear programming, respectively, β * = max s∈S β s , Õ hides logarithmic factors in S, A, ϵ.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 365ea350-c934-4fb6-a6db-1d4da27b37fbCited by top-tier papers5
- Non-rectangular Robust MDPs with Normed Uncertainty SetsNavdeep Kumar, Adarsh Gupta, Maxence Mohamed Elfatihi, Giorgia Ramponi et al.NeurIPS 2025 · 3 citations
- Finite-Time Bounds for Average-Reward Fitted Q-IterationJongmin Lee, Ernest K. RyuNeurIPS 2025 · 1 citation
- Optimal Non-Asymptotic Rates of Value Iteration for Average-Reward Markov Decision ProcessesJongmin Lee, Ernest K. RyuICLR 2025
- Near-Optimal Sample Complexity for MDPs via AnchoringJongmin Lee, Mario Bravo, Roberto CominettiICML 2025
- Best-Effort Policies for Robust Markov Decision ProcessesAlessandro Abate, Thom Badings, Giuseppe De Giacomo, Francesco FabianoAAAI 2026
Builds on6
- Maximum Entropy RL (Provably) Solves Some Robust RL ProblemsBenjamin Eysenbach, Sergey LevineICLR 2022 · 244 citations
- Online Robust Reinforcement Learning with Model UncertaintyYue Wang, Shaofeng ZouNeurIPS 2021 · 157 citations
- Policy Gradient Method For Robust Reinforcement LearningYue Wang, Shaofeng ZouICML 2022 · 104 citations
- Twice regularized MDPs and the equivalence between robustness and regularizationEsther Derman, Matthieu Geist, Shie MannorNeurIPS 2021 · 68 citations
- Natural Actor-Critic for Robust Reinforcement Learning with Function ApproximationRuida Zhou, Tao Liu, Min Cheng, Dileep Kalathil et al.NeurIPS 2023 · 55 citations
Related papers
- Fast Algorithms for -constrained S-rectangular Robust MDPsBahram Behzadian, Marek Petrik, Chin Pang HoNeurIPS 2021 · 4 citations
- Robust -Divergence MDPsChin Pang Ho, Marek Petrik, Wolfram WiesemannNeurIPS 2022 · 13 citations
- Scalable First-Order Methods for Robust MDPsJulien Grand-Clément, Christian KroerAAAI 2021 · 33 citations
- Accelerated Policy Gradient for s-rectangular Robust MDPs with Large State SpacesZiyi Chen, Heng HuangICML 2024 · 3 citations
- Policy Gradient for Rectangular Robust Markov Decision ProcessesNavdeep Kumar, Esther Derman, Matthieu Geist, Kfir Y. Levy et al.NeurIPS 2023 · 45 citations
