Search papers, labs, and topics across Lattice.
This paper introduces the Module Level Reward Evolution Framework (MLREF), which enhances reward function design in reinforcement learning by utilizing a persistent module pool for reusable reward components. By evolving this pool through reflection-based refinement, hybrid credit assignment, and a rollback merge strategy, MLREF allows for more stable and effective optimization across iterations. Experimental results demonstrate that MLREF significantly outperforms existing methods, achieving a 25.2% improvement in locomotion tasks and a 6.6% improvement in manipulation tasks.
Reward function design just got a major upgrade鈥擬LREF achieves 25.2% better performance by reusing and evolving reward components across iterations.
Reward function design remains a bottleneck in reinforcement learning. While large language models (LLMs) have enabled automated reward generation, existing methods generate and revise reward functions as monolithic programs, making it difficult to reliably preserve and reuse effective components discovered in earlier iterations, leading to unstable performance across iterations. To address this, we propose Module Level Reward Evolution Framework (MLREF). At the core of MLREF is a module pool, a persistent repository of reusable reward components. MLREF treats the module pool as the primary optimization object: the pool evolves across iterations by accumulating successful modules, refining underperforming ones, and reusing proven components; while reward functions are constructed as linear combinations of modules drawn from this pool. To drive this evolution, MLREF integrates three mechanisms: reflection-based refinement, hybrid credit assignment, and a merge strategy with rollback, which together improve the effectiveness and robustness of reward optimization. Experiments on 17 tasks show that MLREF outperforms strong baselines by 25.2% in locomotion and 6.6% in manipulation, with more stable optimization dynamics.
Tsinghua AI