Internally Rewarded Reinforcement Learning
Abstract
We study a class of reinforcement learning problems where the reward signals for policy learning are generated by an internal reward model that is dependent on and jointly optimized with the policy. This interdependence between the policy and the reward model leads to an unstable learning process because reward signals from an immature reward model are noisy and impede policy learning, and conversely, an under-optimized policy impedes reward estimation learning. We call this learning setting (IRRL) as the reward is not provided directly by the environment but by a reward model. In this paper, we formally formulate IRRL and present a class of problems that belong to IRRL. We theoretically derive and empirically analyze the effect of the reward function in IRRL and based on these analyses propose the clipped linear reward function. Experimental results show that the proposed reward function can consistently stabilize the training process by reducing the impact of reward noise, which leads to faster convergence and higher performance compared with baselines in diverse tasks.
Cite
@article{arxiv.2302.00270,
title = {Internally Rewarded Reinforcement Learning},
author = {Mengdi Li and Xufeng Zhao and Jae Hee Lee and Cornelius Weber and Stefan Wermter},
journal= {arXiv preprint arXiv:2302.00270},
year = {2023}
}
Comments
Accepted at ICML 2023. Update: adopt the term "reward model" instead of using "critic" to prevent confusion with the term "critic" in actor-critic algorithms. Project webpage at https://ir-rl.github.io