arXiv:2506.08266cs.LGcs.AI2025-06被引 7

让AI更安全地听人话,不靠妥协,而是用数学保证不犯错。

Reinforcement Learning from Human Feedback with High-Confidence Safety Constraints

  • 分离有用性和安全性,分别训练奖励模型和成本模型。
  • 两步优化:先保守求解,再用置信区间验证是否真安全。
  • 理论证明出错概率可控,适合医疗、金融等高风险场景使用。

现有语言模型对齐方法常将安全性视为与帮助性之间的权衡,可能导致敏感领域出现不可接受的回应。为确保此类场景下的可靠表现,我们提出高置信度安全强化学习人类反馈(HC-RLHF)方法,可在最大化帮助性的同时提供高置信度的安全保障。与以往方法类似,HC-RLHF显式地将人类偏好分解为帮助性与无害性(安全性),分别通过训练奖励模型和成本模型进行学习。随后采用两步流程寻找安全解:第一步在成本约束的刻意悲观版本下优化奖励函数;第二步对训练好的模型进行安全测试,验证其性能是否保持在实际成本约束的上置信界内。我们提供了HC-RLHF的理论分析,包括证明其以不超过用户设定阈值的概率返回不安全解。在实证部分,我们将HC-RLHF应用于三个不同语言模型(Qwen2-1.5B、Qwen2.5-3B 和 LLaMa3.2-3B)的人类偏好对齐。结果表明,HC-RLHF能在高概率下生成安全模型,并在无害性和帮助性上优于先前方法。

原文摘要 · Abstract (English)

Existing approaches to language model alignment often treat safety as a tradeoff against helpfulness, which can lead to unacceptable responses in sensitive domains. To ensure reliable performance in such settings, we propose High-Confidence Safe Reinforcement Learning from Human Feedback (HC-RLHF), a method that provides high-confidence safety guarantees while maximizing helpfulness. Similar to previous methods, HC-RLHF explicitly decouples human preferences into helpfulness and harmlessness (safety), which are learned by training a reward model and a cost model, respectively. It then employs a two-step process to find safe solutions. In the first step, it optimizes the reward function under an intentionally pessimistic version of the cost constraint. In the second step, the trained model undergoes a safety test to verify whether its performance stays within an upper-confidence bound of the actual cost constraint. We provide a theoretical analysis of HC-RLHF, including proof that it will not return an unsafe solution with a probability greater than a user-specified threshold. For our empirical analysis, we apply HC-RLHF to align three different language models (Qwen2-1.5B, Qwen2.5-3B, and LLaMa3.2-3B) with human preferences. Our results demonstrate that HC-RLHF produces safe models with high probability and can improve harmlessness and helpfulness compared to previous methods.

强化学习安全对齐可信AI大模型

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