English

Internally Rewarded Reinforcement Learning

Machine Learning 2023-08-28 v3 Artificial Intelligence

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 Internally Rewarded Reinforcement Learning\textit{Internally Rewarded Reinforcement Learning} (IRRL) as the reward is not provided directly by the environment but internally\textit{internally} 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.

Keywords

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

R2 v1 2026-06-28T08:28:48.964Z