直接从人类偏好学习安全策略,避免奖励模型误差累积。
Offline Safe Policy Optimization From Heterogeneous Feedback
- 基于行为偏好与安全标签,直接优化安全策略
- 在连续控制任务中实现高奖励与高安全性,优于现有方法
- 适合需要安全性的离线强化学习场景
离线偏好强化学习(PbRL)可在无需大量奖励工程和人工交互的情况下,学习与人类偏好对齐的奖励与策略。然而,在多个领域中确保安全性仍是关键挑战。以往基于人类反馈的安全强化学习(RLHF)方法先从离线数据中学习奖励与成本模型,再通过约束强化学习优化安全策略。但在长时程连续控制任务中,奖励与成本模型的误差会累积,导致约束强化学习性能下降。为此,本文提出:(a) 不依赖间接学习策略(即不显式学习奖励与成本模型),而是直接基于对智能体行为的成对偏好及轨迹段的安全二值标签进行策略学习;(b) 提出 extsc{PreSa}(偏好与安全对齐)方法,将偏好学习模块与安全对齐结合于一个约束优化问题中,并采用拉格朗日框架求解,直接学习最大化奖励且安全的策略;(c) 在具有合成与真实人类反馈的连续控制任务上评估该方法。实验表明,该方法能有效学习高奖励、高安全性的策略,优于当前最优基线,甚至超越使用真实奖励与成本的离线安全强化学习方法。
原文摘要 · Abstract (English)
Offline Preference-based Reinforcement Learning (PbRL) learns rewards and policies aligned with human preferences without the need for extensive reward engineering and direct interaction with human annotators. However, ensuring safety remains a critical challenge across many domains and tasks. Previous works on safe RL from human feedback (RLHF) first learn reward and cost models from offline data, then use constrained RL to optimize a safe policy. While such an approach works in the contextual bandits settings (LLMs), in long horizon continuous control tasks, errors in rewards and costs accumulate, leading to impairment in performance when used with constrained RL methods. To address these challenges, (a) instead of indirectly learning policies (from rewards and costs), we introduce a framework that learns a policy directly based on pairwise preferences regarding the agent's behavior in terms of rewards, as well as binary labels indicating the safety of trajectory segments; (b) we propose \textsc{PreSa} (Preference and Safety Alignment), a method that combines preference learning module with safety alignment in a constrained optimization problem. This optimization problem is solved within a Lagrangian paradigm that directly learns reward-maximizing safe policy \textit{without explicitly learning reward and cost models}, avoiding the need for constrained RL; (c) we evaluate our approach on continuous control tasks with both synthetic and real human feedback. Empirically, our method successfully learns safe policies with high rewards, outperforming state-of-the-art baselines, and offline safe RL approaches with ground-truth reward and cost.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。