LGROJun 21, 2023

State-wise Constrained Policy Optimization

arXiv:2306.12594v317 citationsh-index: 29
Originality Highly original
AI Analysis

This addresses safety concerns in RL for high-dimensional robotics tasks, representing a novel advancement beyond existing constrained RL frameworks.

The paper tackles the problem of enforcing state-wise safety constraints in reinforcement learning for real-world applications like autonomous driving and robot manipulation, proposing State-wise Constrained Policy Optimization (SCPO) as the first general-purpose algorithm for this task and demonstrating its effectiveness in robot locomotion tasks where it significantly outperforms existing methods.

Reinforcement Learning (RL) algorithms have shown tremendous success in simulation environments, but their application to real-world problems faces significant challenges, with safety being a major concern. In particular, enforcing state-wise constraints is essential for many challenging tasks such as autonomous driving and robot manipulation. However, existing safe RL algorithms under the framework of Constrained Markov Decision Process (CMDP) do not consider state-wise constraints. To address this gap, we propose State-wise Constrained Policy Optimization (SCPO), the first general-purpose policy search algorithm for state-wise constrained reinforcement learning. SCPO provides guarantees for state-wise constraint satisfaction in expectation. In particular, we introduce the framework of Maximum Markov Decision Process, and prove that the worst-case safety violation is bounded under SCPO. We demonstrate the effectiveness of our approach on training neural network policies for extensive robot locomotion tasks, where the agent must satisfy a variety of state-wise safety constraints. Our results show that SCPO significantly outperforms existing methods and can handle state-wise constraints in high-dimensional robotics tasks.

Code Implementations1 repo
Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes