Decoupling Task and Behavior: A Two-Stage Reward Curriculum in Reinforcement Learning for Robotics
AI 摘要
提出一种两阶段奖励课程学习方法,解耦任务目标和行为规范,提升机器人强化学习效果。
主要贡献
- 提出两阶段奖励课程学习框架
- 分析不同阶段的过渡策略
- 验证方法在多个机器人控制环境中的有效性
方法论
首先使用简化任务奖励训练,然后引入包含行为相关项的完整奖励。复用各阶段样本以提高训练稳定性。
原文摘要
Deep Reinforcement Learning is a promising tool for robotic control, yet practical application is often hindered by the difficulty of designing effective reward functions. Real-world tasks typically require optimizing multiple objectives simultaneously, necessitating precise tuning of their weights to learn a policy with the desired characteristics. To address this, we propose a two-stage reward curriculum where we decouple task-specific objectives from behavioral terms. In our method, we first train the agent on a simplified task-only reward function to ensure effective exploration before introducing the full reward that includes auxiliary behavior-related terms such as energy efficiency. Further, we analyze various transition strategies and demonstrate that reusing samples between phases is critical for training stability. We validate our approach on the DeepMind Control Suite, ManiSkill3, and a mobile robot environment, modified to include auxiliary behavioral objectives. Our method proves to be simple yet effective, substantially outperforming baselines trained directly on the full reward while exhibiting higher robustness to specific reward weightings.