Safe Policy Optimization with Local Generalized Linear Function Approximations
Akifumi Wachi, Yunyue Wei, Yanan Sui
Abstract
Safe exploration is a key to applying reinforcement learning (RL) in safety-critical systems. Existing safe exploration methods guaranteed safety under the assumption of regularity, and it has been difficult to apply them to large-scale real problems. We propose a novel algorithm, SPO-LF, that optimizes an agent's policy while learning the relation between a locally available feature obtained by sensors and environmental reward/safety using generalized linear function approximations. We provide theoretical guarantees on its safety and optimality. We experimentally show that our algorithm is 1) more efficient in terms of sample complexity and computational cost and 2) more applicable to large-scale problems than previous safe RL methods with theoretical guarantees, and 3) comparably sample-efficient and safer compared with existing advanced deep RL methods with safety constraints.
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.
Cited by top-tier papers3
- Safe Exploration in Reinforcement Learning: A Generalized Formulation and AlgorithmsAkifumi Wachi, Wataru Hashimoto, Xun Shen, Kazumune HashimotoNeurIPS 2023 · 38 citations
- Generalized Linear Bandits: Almost Optimal Regret with One-Pass UpdateYu-Jie Zhang, Sheng-An Xu, Peng Zhao, Masashi SugiyamaNeurIPS 2025 · 17 citations
- HASARD: A Benchmark for Vision-Based Safe Reinforcement Learning in Embodied AgentsTristan Tomilin, Meng Fang, Mykola PechenizkiyICLR 2025
Builds on5
- Reinforcement Learning in Feature Space: Matrix Bandit, Kernels, and Regret BoundLin Yang, Mengdi WangICML 2020 · 308 citations
- IPO: Interior-Point Policy Optimization under ConstraintsYongshuai Liu, Jiaxin Ding, Xin LiuAAAI 2020 · 231 citations
- Safe Reinforcement Learning in Constrained Markov Decision ProcessesAkifumi Wachi, Yanan SuiICML 2020 · 190 citations
- Constrained Markov Decision Processes via Backward Value FunctionsHarsh Satija, Philip Amortila, Joelle PineauICML 2020 · 58 citations
- Optimism in Reinforcement Learning with Generalized Linear Function ApproximationYining Wang, Ruosong Wang, Simon Shaolei Du, Akshay KrishnamurthyICLR 2021 · 54 citations
Related papers
- Enhancing Efficiency of Safe Reinforcement Learning via Sample ManipulationShangding Gu, Laixi Shi, Yuhao Ding, Alois Knoll et al.NeurIPS 2024 · 14 citations
- Safety Representations for Safer Policy LearningKaustubh Mani, Vincent Mai, Charlie Gauthier, Annie S. Chen et al.ICLR 2025
- ActSafe: Active Exploration with Safety Constraints for Reinforcement LearningYarden As, Bhavya Sukhija, Lenart Treven, Carmelo Sferrazza et al.ICLR 2025
- CSPO: Constraint-Sensitive Policy Optimization for Safe Reinforcement LearningAyoub Belouadah, Sylvain Kubler, YVES LE TRAONICML 2026
- Safe Reinforcement Learning with Linear Function ApproximationSanae Amani, Christos Thrampoulidis, Lin YangICML 2021 · 42 citations
