Deterministic Policies for Constrained Reinforcement Learning in Polynomial Time
Jeremy McMahan
Abstract
We present a novel algorithm that efficiently computes near-optimal deterministic policies for constrained reinforcement learning (CRL) problems. Our approach combines three key ideas: (1) value-demand augmentation, (2) action-space approximate dynamic programming, and (3) time-space rounding. Our algorithm constitutes a fully polynomial-time approximation scheme (FPTAS) for any time-space recursive (TSR) cost criteria. A TSR criteria requires the cost of a policy to be computable recursively over both time and (state) space, which includes classical expectation, almost sure, and anytime constraints. Our work answers three open questions spanning two long-standing lines of research: polynomial-time approximability is possible for 1) anytime-constrained policies, 2) almost-sure-constrained policies, and 3) deterministic expectation-constrained policies.
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 31b64b00-b57d-40a6-9e2e-6622a1bc1c3dCited by top-tier papers3
- Deterministic Policy Gradient Primal-Dual Methods for Continuous-Space Constrained MDPsSergio Rozada, Dongsheng Ding, Antonio G. Marques, Alejandro RibeiroAAAI 2025 · 2 citations
- Anytime-Constrained Equilibria in Polynomial TimeJeremy McMahanICML 2025
- Polynomial-Time Approximability of Constrained Reinforcement LearningJeremy McMahanICML 2025
Builds on6
- Safe Reinforcement Learning by Imagining the Near FutureGarrett Thomas, Yuping Luo, Tengyu MaNeurIPS 2021 · 118 citations
- Enforcing Hard Constraints with Soft Barriers: Safe Reinforcement Learning in Unknown Stochastic EnvironmentsYixuan Wang, Simon Sinong Zhan, Ruochen Jiao, Zhilu Wang et al.ICML 2023 · 81 citations
- Constrained episodic reinforcement learning in concave-convex and knapsack settingsKianté Brantley, Miroslav Dudík, Thodoris Lykouris, Sobhan Miryoosefi et al.NeurIPS 2020 · 56 citations
- Near-Optimal Sample Complexity Bounds for Constrained MDPsSharan Vaswani, Lin Yang, Csaba SzepesváriNeurIPS 2022 · 52 citations
- Learning with Safety Constraints: Sample Complexity of Reinforcement Learning for Constrained MDPsAria HasanzadeZonuzy, Archana Bura, Dileep M. Kalathil, Srinivas ShakkottaiAAAI 2021 · 46 citations
Related papers
- Reward Penalties on Augmented States for Solving Richly Constrained RL EffectivelyHao Jiang, Tien Mai, Pradeep Varakantham, Huy HoangAAAI 2024 · 2 citations
- Efficient Action-Constrained Reinforcement Learning via Acceptance-Rejection Method and Augmented MDPsWei Hung, Shao-Hua Sun, Ping-Chun HsiehICLR 2025
- Anytime-Competitive Reinforcement Learning with Policy PriorJianyi Yang, Pengfei Li, Tongxin Li, Adam Wierman et al.NeurIPS 2023 · 3 citations
- ACPO: A Policy Optimization Algorithm for Average MDPs with ConstraintsAkhil Agnihotri, Rahul Jain, Haipeng LuoICML 2024 · 2 citations
- Planning with Participation ConstraintsHanrui Zhang, Yu Cheng, Vincent ConitzerAAAI 2022 · 4 citations
