用约束违规信号训练可微动作生成模型,让智能体安全探索。
Leveraging Constraint Violation Signals For Action-Constrained Reinforcement Learning
- 基于违规信号构建可行动作分布,用归一化流直接学习映射
- 在多个控制任务中约束违规减少,性能与最优方法相当或更优
- 无需采样可行动作,适合需实时安全控制的场景
在许多强化学习应用中,确保智能体动作符合约束对安全性至关重要。以往动作约束强化学习(ACRL)方法通常在策略网络后使用投影层修正动作,但此类方法存在零梯度问题且运行时间长。近期方法尝试训练生成模型,通过可微映射将潜在变量映射到可行动作空间,但需从受限动作空间采样,仍具挑战。本文提出:首先,基于约束违规信号定义可行动作的目标分布,通过最小化近似分布与目标分布间的KL散度训练归一化流,无需生成可行动作样本,极大简化学习过程;其次,将学习到的流模型集成至现有深度强化学习框架,限制智能体仅在可行动作空间内探索;第三,扩展方法以处理状态相关约束,从环境中学习违规信号。实验表明,该方法在多个控制任务中显著减少约束违规,同时性能达到或优于现有最佳方法。
原文摘要 · Abstract (English)
In many RL applications, ensuring an agent's actions adhere to constraints is crucial for safety. Most previous methods in Action-Constrained Reinforcement Learning (ACRL) employ a projection layer after the policy network to correct the action. However projection-based methods suffer from issues like the zero gradient problem and higher runtime due to the usage of optimization solvers. Recently methods were proposed to train generative models to learn a differentiable mapping between latent variables and feasible actions to address this issue. However, generative models require training using samples from the constrained action space, which itself is challenging. To address such limitations, first, we define a target distribution for feasible actions based on constraint violation signals, and train normalizing flows by minimizing the KL divergence between an approximated distribution over feasible actions and the target. This eliminates the need to generate feasible action samples, greatly simplifying the flow model learning. Second, we integrate the learned flow model with existing deep RL methods, which restrict it to exploring only the feasible action space. Third, we extend our approach beyond ACRL to handle state-wise constraints by learning the constraint violation signal from the environment. Empirically, our approach has significantly fewer constraint violations while achieving similar or better quality in several control tasks than previous best methods.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。