Safe Reinforcement Learning with Learned Non-Markovian Safety Constraints
<|reference_start|>Safe Reinforcement Learning with Learned Non-Markovian Safety Constraints: In safe Reinforcement Learning (RL), safety cost is typically defined as a function dependent on the immediate state and actions. In practice, safety constraints can often be non-Markovian due to the insufficient fidelity of state representation, and safety cost may not be known. We therefore address a general setting where safety labels (e.g., safe or unsafe) are associated with state-action trajectories. Our key contributions are: first, we design a safety model that specifically performs credit assignment to assess contributions of partial state-action trajectories on safety. This safety model is trained using a labeled safety dataset. Second, using RL-as-inference strategy we derive an effective algorithm for optimizing a safe policy using the learned safety model. Finally, we devise a method to dynamically adapt the tradeoff coefficient between reward maximization and safety compliance. We rewrite the constrained optimization problem into its dual problem and derive a gradient-based method to dynamically adjust the tradeoff coefficient during training. Our empirical results demonstrate that this approach is highly scalable and able to satisfy sophisticated non-Markovian safety constraints.<|reference_end|>
@article{low2024safe,
    title={Safe Reinforcement Learning with Learned Non-Markovian Safety
  Constraints},
    author={Siow Meng Low and Akshat Kumar},
    journal={arXiv preprint arXiv:2405.03005},
    year={2024},
    archivePrefix={arXiv},
    eprint={2405.03005},
    primaryClass={cs.LG cs.AI}
}