arXiv:2605.21225cs.LGcs.AI2026-05中稿 · AAMAS 2026 as a fu…

用偏好数据让强化学习模型更安全,不重训也能保持高奖励。

PREFINE: Preference-Based Implicit Reward and Cost Fine-Tuning for Safety Alignment

论文配图:PREFINE: Preference-Based Implicit Reward and Cost Fine-Tuning for Safety Alignment
图 1 · 摘自论文原文
  • 基于轨迹偏好构建反事实样本,联合优化奖励与安全约束。
  • 降低60%以上约束违规和灾难性失败,保持原始奖励水平。
  • 适合需要安全对齐的连续控制场景,效率远超传统方法。

我们解决预训练强化学习策略在不从头训练的前提下,如何通过成本约束实现安全感知的问题。尽管成本可数值编码,但本文假设更一般的情形是成本以偏好形式给出。给定一个奖励优化的策略和少量优选(低成本)与非优选(高成本)轨迹数据集,目标是微调策略生成低成本行为的同时保留高奖励。不同于语言模型中的标准RLHF(偏好针对同一提示的不同回复),本工作涉及连续控制环境中的轨迹级偏好。我们提出PREFINE:一种基于偏好的隐式奖励与成本微调方法,将广泛用于大模型微调的直接偏好优化(DPO)扩展至序列决策场景。PREFINE通过策略采样构造反事实轨迹,建立有意义的偏好对比,并联合优化奖励保持与安全对齐。实验表明,PREFINE将约束违规和灾难性失败减少超过60%,同时维持原始奖励表现。相比完整离线强化学习或模仿学习,PREFINE在数据与计算效率上均有显著提升,实现了连续领域中偏好对齐与安全策略适应的桥梁。

原文摘要 · Abstract (English)

We address the problem of making a pre-trained reinforcement learning (RL) policy safety-aware by incorporating cost constraints without retraining it from scratch. While costs could be numerically encoded, we assume a more general setting is when costs are provided as preferences. Given a reward-optimized policy and a small dataset of preferred (low-cost) and dispreferred (high-cost) trajectories, our goal is to fine-tune the policy to generate low-cost behaviors while retaining high rewards. Unlike standard RLHF in language models, where preferences are defined over responses to the same prompt, our setting involves trajectory-level preferences in continuous control environments. We introduce PREFINE: Preference-based Implicit Reward and Cost Fine-Tuning for Safety Alignment which is a preference-based fine-tuning method that adapts Direct Preference Optimization (DPO), which is now widely used for LLM fine-tuning, to the sequential decision making setting. PREFINE constructs policy-sampled counterfactual trajectories to establish meaningful preference contrasts and jointly optimizes for reward retention and safety alignment. Empirically, PREFINE reduces constraint violations and catastrophic failures by over 60% while maintaining original reward behavior. PREFINE produces policies that achieve low-cost, high-reward performance with significantly improved data and computational efficiency compared to full offline RL or imitation learning, bridging preference alignment and safe policy adaptation in continuous domains.

强化学习安全对齐偏好学习连续控制

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