arXiv:2503.05238cs.LG2025-03AAAI被引 7

用概率模型检测强化学习中的异常状态,确保部署时可靠预警。

Guaranteeing Out-Of-Distribution Detection in Deep RL via Transition Estimation

  • 基于状态转移概率建模,定义异常为训练与实际分布不一致的转移。
  • 采用条件变分自编码器估计转移动态,重建误差实现可保证的检测。
  • 适合关注安全部署的RL系统,尤其适用于高风险场景

深度强化学习代理在部署时的可靠性常受质疑,因其训练环境未必反映真实场景。为应对这一问题,系统通常配备分布外(OOD)检测机制,在遇到未知状态或不确定性时发出警报。现有研究对强化学习中OOD执行的定义缺乏共识。本文基于马尔可夫决策过程,将每个状态-动作对映射到下一状态的概率视为转移分布,并定义:若实际部署中某转移概率与训练阶段不同,则该转移为OOD。为此,我们使用条件变分自编码器(CVAE)拟合训练环境的转移动态,结合重构损失构建基于符合性的检测器,可在预设置信水平下保证OOD检测。我们在适配的基准上评估该方法,并与现有强化学习OOD检测模型对比。

原文摘要 · Abstract (English)

An issue concerning the use of deep reinforcement learning (RL) agents is whether they can be trusted to perform reliably when deployed, as training environments may not reflect real-life environments. Anticipating instances outside their training scope, learning-enabled systems are often equipped with out-of-distribution (OOD) detectors that alert when a trained system encounters a state it does not recognize or in which it exhibits uncertainty. There exists limited work conducted on the problem of OOD detection within RL, with prior studies being unable to achieve a consensus on the definition of OOD execution within the context of RL. By framing our problem using a Markov Decision Process, we assume there is a transition distribution mapping each state-action pair to another state with some probability. Based on this, we consider the following definition of OOD execution within RL: A transition is OOD if its probability during real-life deployment differs from the transition distribution encountered during training. As such, we utilize conditional variational autoencoders (CVAE) to approximate the transition dynamics of the training environment and implement a conformity-based detector using reconstruction loss that is able to guarantee OOD detection with a pre-determined confidence level. We evaluate our detector by adapting existing benchmarks and compare it with existing OOD detection models for RL.

强化学习分布外检测安全部署概率建模

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。