Dual-Evolving Reinforcement Learning for Automated Reward Design
Organizations: University of Waterloo · The University of Hong Kong · The Chinese University of Hong Kong, Shenzhen · Duke University
Abstract
Crafting effective reward signals remains a central challenge in reinforcement learning (RL) for complex reasoning. Existing methods for automated reward optimization search with a fixed proposer which never learns from its costly trials. We present Dual-Evolving Reinforcement Learning (DERL), a dual-training framework for autonomous discovery of reward structures. An outer-loop proposer composes a reward from task primitives, and an inner policy trains with this reward. The proposer itself learns by policy gradient from the validation performance of the inner policy. Thus, DERL utilizes past-trial feedback to shape its reward-proposal distribution, an empirical meta-gradient of task success. We validate DERL on embodied (ALFWorld), scientific (ScienceWorld), and mathematical (GSM8K, MATH) reasoning. On multi-turn agent benchmarks, DERL outperforms strong baselines, significantly on out-of-distribution tasks. In mathematics, DERL improves on the outcome reward where directly added primitives hurt. Further analyses under matched compute show that our proposer effectively captures the meta-gradient to discover useful reward signals.
Figures & tables
| Method | ALFWorld | ScienceWorld | ||||
|---|---|---|---|---|---|---|
| L0 | L1 | L2 (OOD) | L0 | L1 | L2 (OOD) | |
| GRPO w/ Outcome | 76.6 | 71.1 | 29.7 | 21.1 | 13.7 | 10.9 |
| GRPO w/ Average(primitives) | 88.1 | 85.4 | 30.5 | 37.9 | 31.3 | 18.0 |
| GiGPO | 86.7 | 83.2 | 48.0 | 25.8 | 15.2 | 4.7 |
| RLVMR | 89.1 | 87.9 | 56.3 | 46.9 | 34.4 | 26.5 |
| DERL -base | 91.0 | 89.1 | 65.0 | 47.7 | 43.0 | 30.1 |
| Reward Function | GSM8K | MATH |
|---|---|---|
| Trained on MATH | ||
| Outcome | 82.9 | 59.1 |
| Outcome + Format | 83.9 | 56.8 |
| Average(primitives) | 83.6 | 54.9 |
| DERL -base | 83.2 | 60.5 |
| DERL -full | 84.1 | 60.9 |
Appendix figures & tables6 assets
Supplementary material from the paper’s appendix.
Appendix
| Outer step | Proposed program (verbatim) | Val score |
|---|---|---|
| 0 | g1 * (g2 - 1) / 2 + (g3 + 1) * (g4 - 1) * 2 / 3 | 0 |
| g1 + 0.5 * (g2 + 0.5 * (g3 + 0.5 * (g4)))) - 0.5 * (g2 + 0.5 * (g3 + ... | 0 | |
| g1 + 0.01 * (g2 - 0.001) + 0.0001 * (g3 - 0.0001) + 0.000005 * (g4 - ... | 0.850 | |
| g1 * 0.5 + 0.2 * (g2 + 0.1) - 0.3 / 2 + 0.4 * (g3 * 0.1) + 0.1 * (g4 * ... | 0.879 | |
| g1 * (g2 + (g3 - 1.0) * (g4 - 0.0)) | 0.023 | |
| g1 + 0.5 * (g2 + 0.5 * (g3 + 0.5 * (g4 + 0.5))) | 0.885 |
| Method | Train | Test |
|---|---|---|
| GRPO baselines | ||
| GRPO w/ Outcome | 89.45 | 75.82 |
| GRPO w/ Average(primitives) | 89.74 | 76.12 |
| Static proposers | ||
| Random search (best over steps) | 89.59 | 76.50 |
| RLAIF (prompted GPT-4o) | 89.23 | 77.18 |
| Outer step | DERL -base | Random search | RLAIF |
|---|---|---|---|
| 0 | 75.13 | 76.08 | 76.88 |
| 1 | 76.60 | 74.90 | 76.67 |
| 2 | 77.69 | 75.97 | 76.40 |
| 3 | 77.13 | 54.58 | 76.85 |
| 4 | 77.29 | 76.34 | 76.45 |
| 5 | 78.28 | 75.16 | 76.17 |
| Method | L0 | L1 | L2 |
|---|---|---|---|
| ALFWorld | |||
| GRPO w/ Outcome | 76.56 1.91 [73.52, 79.61] | 71.09 2.89 [66.50, 75.69] | 29.69 2.12 [26.32, 33.05] |
| GRPO w/ Average(primitives) | 88.09 1.33 [85.96, 90.21] | 85.35 1.33 [83.23, 87.47] | 30.47 2.63 [26.28, 34.65] |
| DERL -base | 91.02 2.07 [87.73, 94.30] | 89.06 1.42 [86.80, 91.32] | 65.04 2.64 [60.83, 69.25] |
| DERL -full | 91.80 1.35 [89.64, 93.95] | 88.28 1.78 [85.46, 91.11] | 76.37 4.05 [69.92, 82.82] |
| ScienceWorld | |||
Reward Design for RL
Reward design for reinforcement learning studies how to specify, learn, and adapt signals that guide agents toward intended behavior.
See all 259 papers