部署后无需重训,就能让强化学习模型更公平
Inference-Time Policy Alignment for Fair Reinforcement Learning
- 推理时用福利评分动态调整动作概率,不改原模型参数
- 多领域实验显示公平性显著提升,任务性能基本不变
- 适合需快速适配新公平性要求的落地场景
深度强化学习(RL)代理通过优化标量奖励函数获得优异表现。然而,部署后其策略往往僵化且难以适应新的性能标准。例如,以最大化累积奖励训练的代理无法应对未知的利益相关方偏好。现有实现公平性的方法通常假设偏好已知,需在公平性指标下完全重训策略。受大语言模型推理时对齐的启发,我们研究在不更新基础策略参数的前提下,于推理时引导预训练的RL策略向基于福利的公平目标对齐。我们将该问题形式化为策略塑形问题,提出一种乘法式策略塑形框架,通过依赖动作的福利评分调整动作概率,无需修改基础策略。该框架通用,兼容任意深度RL代理。在多个领域的大量实验表明,推理时策略塑形能显著提升基于福利的公平性指标,同时保持核心任务性能。
原文摘要 · Abstract (English)
Deep reinforcement learning (RL) agents achieve strong performance by optimizing scalar reward functions. However, once deployed, the policies of these RL agents are often rigid and costly to adapt to new performance criteria. For instance, an agent trained to maximize expected cumulative reward may not accommodate previously unknown stakeholder preferences. Existing approaches to achieve fairness, a type of preference, in RL typically assume that such preferences are known a priori and require complete retraining of the policy under a fairness-oriented metric. Inspired by inference-time alignment in large language models, we investigate the problem of steering a pretrained RL policy toward welfare-based fairness objectives at inference time without updating the base policy's parameters. We formalize inference-time fairness alignment as a policy shaping problem and propose a multiplicative policy shaping framework that adjusts action probabilities using action-dependent welfare scores, thus requiring no modification to the base policy. Our framework is general and compatible with any deep RL agent. Through extensive experiments across multiple domains, we demonstrate that inference-time policy shaping substantially improves welfare-based fairness objectives while preserving core task performance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。