State-wise Constrained Policy Optimization

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Zhao, Weiye, Chen, Rui, Sun, Yifan, Wei, Tianhao, Liu, Changliu
Format: Preprint
Published: 2023
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866929388378914816
author Zhao, Weiye
Chen, Rui
Sun, Yifan
Wei, Tianhao
Liu, Changliu
author_facet Zhao, Weiye
Chen, Rui
Sun, Yifan
Wei, Tianhao
Liu, Changliu
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2306_12594
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle State-wise Constrained Policy Optimization
Zhao, Weiye
Chen, Rui
Sun, Yifan
Wei, Tianhao
Liu, Changliu
Machine Learning
Robotics
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.
title State-wise Constrained Policy Optimization
topic Machine Learning
Robotics
url https://arxiv.org/abs/2306.12594