ProSh: Probabilistic Shielding for Model-free Reinforcement Learning

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Court, Edwin Hamel-De le, Ohlmann, Gaspard, Belardinelli, Francesco
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866917029014929408
author Court, Edwin Hamel-De le
Ohlmann, Gaspard
Belardinelli, Francesco
author_facet Court, Edwin Hamel-De le
Ohlmann, Gaspard
Belardinelli, Francesco
contents Safety is a major concern in reinforcement learning (RL): we aim at developing RL systems that not only perform optimally, but are also safe to deploy by providing formal guarantees about their safety. To this end, we introduce Probabilistic Shielding via Risk Augmentation (ProSh), a model-free algorithm for safe reinforcement learning under cost constraints. ProSh augments the Constrained MDP state space with a risk budget and enforces safety by applying a shield to the agent's policy distribution using a learned cost critic. The shield ensures that all sampled actions remain safe in expectation. We also show that optimality is preserved when the environment is deterministic. Since ProSh is model-free, safety during training depends on the knowledge we have acquired about the environment. We provide a tight upper-bound on the cost in expectation, depending only on the backup-critic accuracy, that is always satisfied during training. Under mild, practically achievable assumptions, ProSh guarantees safety even at training time, as shown in the experiments.
format Preprint
id arxiv_https___arxiv_org_abs_2510_15720
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle ProSh: Probabilistic Shielding for Model-free Reinforcement Learning
Court, Edwin Hamel-De le
Ohlmann, Gaspard
Belardinelli, Francesco
Machine Learning
Artificial Intelligence
Safety is a major concern in reinforcement learning (RL): we aim at developing RL systems that not only perform optimally, but are also safe to deploy by providing formal guarantees about their safety. To this end, we introduce Probabilistic Shielding via Risk Augmentation (ProSh), a model-free algorithm for safe reinforcement learning under cost constraints. ProSh augments the Constrained MDP state space with a risk budget and enforces safety by applying a shield to the agent's policy distribution using a learned cost critic. The shield ensures that all sampled actions remain safe in expectation. We also show that optimality is preserved when the environment is deterministic. Since ProSh is model-free, safety during training depends on the knowledge we have acquired about the environment. We provide a tight upper-bound on the cost in expectation, depending only on the backup-critic accuracy, that is always satisfied during training. Under mild, practically achievable assumptions, ProSh guarantees safety even at training time, as shown in the experiments.
title ProSh: Probabilistic Shielding for Model-free Reinforcement Learning
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2510.15720