Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Decoupling Task and Behavior: A Two-Stage Reward Curriculum in Reinforcement Learning for Robotics

About

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.

Kilian Freitag, Knut {\AA}kesson, Morteza Haghir Chehreghani• 2026

Related benchmarks

TaskDatasetResultRank
Continuous ControlDeepMind Control Suite Cheetah Run
Reward0.62
13
Continuous ControlDM Control reacher-hard
Average Reward0.91
9
Continuous ControlDeepMind Control Suite Walker Walk
Reward0.86
6
Continuous ControlDM Control reacher-easy
Average Reward0.92
4
Continuous ControlDM Control pendulum-swingup
Average Reward0.56
4
Continuous ControlDM Control fish-upright
Average Reward0.79
4
Continuous ControlDM Control finger-turn-hard
Average Reward0.56
4
Continuous ControlDM Control walker-run
Average Reward0.62
4
Continuous ControlDM Control ball-in-cup-catch
Average Reward0.94
4
Continuous ControlDM Control finger-spin
Average Reward0.83
4
Showing 10 of 17 rows

Other info

Follow for update