让AI模型的自我解释与行为一致,提升可审计性和安全性。
Self-CTRL: Self-Consistency Training with Reinforcement Learning
- 通过强化学习优化模型自解释与行为的一致性。
- 在推理任务中,解释与行为相关性从0.24提升至0.64。
- 适用于安全对齐、透明审计,尤其适合需要可信规则的场景。
语言模型若能准确描述自身行为,将更易于被审计、理解与信任。本文提出自一致性训练与强化学习结合的方法(Self-CTRL),通过更新自解释以更好预测行为,或更新行为以更符合解释,实现两者一致性。我们在两个领域验证:一是形式化概率推理任务,模型需模仿一组有偏采样器并报告偏差;一致性训练使自报告与行为测量的隐变量偏差相关性从 $R^2=0.24$ 提升至 $R^2=0.64$,接近直接真实监督的泛化性能。二是宪法型AI任务,模型需说明何时拒绝或响应用户请求;Self-CTRL使第三方审计模型的拒绝预测准确率从36%提升至92%。反向调整行为则显著改善对齐效果,使HarmBench失败率从15.0%降至0.5%,同时未大幅增加对无害提示的拒绝。该方法为训练更安全、透明、可控的AI提供了通用范式。
原文摘要 · Abstract (English)
Language models (LMs) that faithfully describe their own behavior can more easily be audited, understood, and trusted by users. This paper describes Self-Consistency Training with Reinforcement Learning (Self-CTRL), a method that optimizes for consistency between a LM's self-explanations and behavior on related inputs by updating explanations to better predict behavior or updating behavior to better match explanations. We apply our method in two domains. First, we study a formal probabilistic reasoning task in which LMs must learn to imitate a family of biased samplers and evaluated on their ability to report the associated biases. We find that consistency training improves the correlation between self-reported and behaviorally-measured latent biases from $R^2=0.24$ to $R^2=0.64$ on a set of held-out distributions, matching the generalization of direct ground-truth supervision. Second, we study a constitutional AI domain in which LMs must describe when they will refuse or comply with user requests. Here, Self-CTRL produces rules that faithfully describe the model's behavior on held-out requests, improving the refusal predictions of a third-party auditor model from $36\%$ to $92\%$. In the other direction, behavior updates improve alignment, reducing HarmBench failure rate from $15.0\%$ to $0.5\%$ without substantially increasing refusal on harmless prompts. By aligning explanations and behavior, our work provides a general recipe for training AI models to be safer, more transparent, and more controllable.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。