用最优传输理论量化动作不确定性,让强化学习更安全
Optimal Transport-Guided Safety in Temporal Difference Reinforcement Learning
- 引入最优传输度量动作的不确定性
- 显著降低进入危险状态的概率
- 适合需要安全决策的机器人控制场景
强化学习通常追求最优性能而忽视安全。本文将安全定义为在环境随机性下动作后果更具可预测性,并提出一种基于最优传输理论的时序差分算法,用于量化动作不确定性。通过将该不确定性评分融入决策目标,智能体被鼓励选择结果更可预测的动作。理论证明该方法能降低进入危险状态的概率。在多种环境不确定性下的案例研究中,结果表明该方法不仅提升了安全性,同时保持了原有性能水平。代码开源地址:https://github.com/SAILRIT/OT-guided-TD-Learning。
原文摘要 · Abstract (English)
The primary goal of reinforcement learning is to develop decision-making policies that prioritize optimal performance, frequently without considering safety. In contrast, safe reinforcement learning seeks to reduce or avoid unsafe behavior. This paper views safety as taking actions with more predictable consequences under environment stochasticity and introduces a temporal difference algorithm that uses optimal transport theory to quantify the uncertainty associated with actions. By integrating this uncertainty score into the decision-making objective, the agent is encouraged to favor actions with more predictable outcomes. We theoretically prove that our algorithm leads to a reduction in the probability of visiting unsafe states. We evaluate the proposed algorithm on several case studies in the presence of various forms of environment uncertainty. The results demonstrate that our method not only provides safer behavior but also maintains the performance. A Python implementation of our algorithm is available at \href{https://github.com/SAILRIT/Risk-averse-TD-Learning}{https://github.com/SAILRIT/OT-guided-TD-Learning}.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。