arXiv:2410.16790cs.LGcs.RO2024-10被引 5

用分阶段奖励策略让强化学习更好平衡复杂目标。

Curriculum Reinforcement Learning for Complex Reward Functions

  • 先学简单奖励,再过渡到复杂奖励,逐步优化。
  • 在控制套件和机器人任务中性能显著提升。
  • 适合需要多目标平衡的机器人实际应用。

强化学习在解决控制问题方面表现出强大能力,但其实际应用常因包含多个项的复杂奖励函数而受阻。尽管奖励假设认为任何目标均可由标量奖励函数表示,但在没有被利用的前提下平衡相互竞争的奖励项仍具挑战。为克服传统方法对奖励项精确调参的依赖,本文提出一种两阶段奖励课程:首先最大化简单奖励,再过渡至完整复杂奖励。我们基于智能体对评价网络的拟合程度,自动确定两阶段切换点,并引入灵活回放缓冲区,实现跨阶段样本高效复用。在修改后的DeepMind控制套件(增加约束项)及移动机器人场景(更多竞争项)中评估,相比无课程基线,本方法显著提升性能,能有效平衡任务完成与约束满足,而非仅利用约束项。结果表明,该两阶段奖励课程可提升复杂奖励环境下强化学习的效率与稳定性,推动更鲁棒、适应性强的机器人系统在真实场景中的应用。

原文摘要 · Abstract (English)

Reinforcement learning (RL) has emerged as a powerful tool for tackling control problems, but its practical application is often hindered by the complexity arising from intricate reward functions with multiple terms. The reward hypothesis posits that any objective can be encapsulated in a scalar reward function, yet balancing individual, potentially adversarial, reward terms without exploitation remains challenging. To overcome the limitations of traditional RL methods, which often require precise balancing of competing reward terms, we propose a two-stage reward curriculum that first maximizes a simple reward function and then transitions to the full, complex reward. We provide a method based on how well an actor fits a critic to automatically determine the transition point between the two stages. Additionally, we introduce a flexible replay buffer that enables efficient phase transfer by reusing samples from one stage in the next. We evaluate our method on the DeepMind control suite, modified to include an additional constraint term in the reward definitions. We further evaluate our method in a mobile robot scenario with even more competing reward terms. In both settings, our two-stage reward curriculum achieves a substantial improvement in performance compared to a baseline trained without curriculum. Instead of exploiting the constraint term in the reward, it is able to learn policies that balance task completion and constraint satisfaction. Our results demonstrate the potential of two-stage reward curricula for efficient and stable RL in environments with complex rewards, paving the way for more robust and adaptable robotic systems in real-world applications.

强化学习奖励设计机器人控制课程学习

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