用新方法让启发式奖励真正提升策略,减少人工调参
Going Beyond Heuristics by Imposing Policy Improvement as a Constraint
- 以策略提升为约束,重新设计启发式奖励的使用方式
- 在标准基准上表现优于现有方法,且对不完美奖励也有效
- 适合想降低奖励设计难度的研究者和工程应用
在强化学习中,加入体现人类先验知识的启发式奖励对获得理想性能至关重要。然而,由于这些启发式通常非最优,需大量人力与计算资源平衡任务奖励与启发式奖励。理论严谨的方法依赖策略不变性,但实践中无法带来策略提升,表现不佳。本文提出新的框架HEPO(启发式增强策略优化),以最大化策略提升为目标,而非策略不变性。该方法能有效利用启发式,避免先前方法的奖励欺骗问题。在标准基准上,HEPO在精心设计的奖励函数下表现更优;更令人意外的是,即使启发式由非专家设计且不够完善,也能实现良好性能,显著降低奖励设计的人力成本。
原文摘要 · Abstract (English)
In many reinforcement learning (RL) applications, augmenting the task rewards with heuristic rewards that encode human priors about how a task should be solved is crucial for achieving desirable performance. However, because such heuristics are usually not optimal, much human effort and computational resources are wasted in carefully balancing tasks and heuristic rewards. Theoretically rigorous ways of incorporating heuristics rely on the idea of \textit{policy invariance}, which guarantees that the performance of a policy obtained by maximizing heuristic rewards is the same as the optimal policy with respect to the task reward. However, in practice, policy invariance doesn't result in policy improvement, and such methods are known to empirically perform poorly. We propose a new paradigm to mitigate reward hacking and effectively use heuristics based on the practical goal of maximizing policy improvement instead of policy improvement. Our framework, Heuristic Enhanced Policy Optimization (HEPO), effectively leverages heuristics while avoiding the pitfall of prior methods for mitigating reward hacking. HEPO achieves superior performance on standard benchmarks with well-engineered reward functions. More surprisingly, HEPO allows policy optimization to achieve good performance even when heuristics are not well-engineered and designed by non-expert humans, showcasing HEPO's ability to reduce human effort in reward design. % HEPO is a plug-and-play optimization method for leveraging heuristics in reinforcement learning. Code is available at https://github.com/Improbable-AI/hepo.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。