Toward Evaluative Thinking: Meta Policy Optimization with Evolving Reward Models

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Kim, Zae Myung, Park, Chanwoo, Raheja, Vipul, Kim, Suin, Kang, Dongyeop
Format: Preprint
Publié: 2025
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866912380635578368
author Kim, Zae Myung
Park, Chanwoo
Raheja, Vipul
Kim, Suin
Kang, Dongyeop
author_facet Kim, Zae Myung
Park, Chanwoo
Raheja, Vipul
Kim, Suin
Kang, Dongyeop
contents Reward-based alignment methods for large language models (LLMs) face two key limitations: vulnerability to reward hacking, where models exploit flaws in the reward signal; and reliance on brittle, labor-intensive prompt engineering when LLMs are used as reward models. We introduce Meta Policy Optimization (MPO), a framework that addresses these challenges by integrating a meta-reward model that dynamically refines the reward model's prompt throughout training. In MPO, the meta-reward model monitors the evolving training context and continuously adjusts the reward model's prompt to maintain high alignment, providing an adaptive reward signal that resists exploitation by the policy. This meta-learning approach promotes a more stable policy optimization, and greatly reduces the need for manual reward prompt design. It yields performance on par with or better than models guided by extensively hand-crafted reward prompts. Furthermore, we show that MPO maintains its effectiveness across diverse tasks, from essay writing to mathematical reasoning, without requiring specialized reward designs. Beyond standard RLAIF, MPO's meta-learning formulation is readily extensible to higher-level alignment frameworks. Overall, this method addresses theoretical and practical challenges in reward-based RL alignment for LLMs, paving the way for more robust and adaptable alignment strategies. The code and data can be accessed at: https://github.com/minnesotanlp/mpo
format Preprint
id arxiv_https___arxiv_org_abs_2504_20157
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Toward Evaluative Thinking: Meta Policy Optimization with Evolving Reward Models
Kim, Zae Myung
Park, Chanwoo
Raheja, Vipul
Kim, Suin
Kang, Dongyeop
Computation and Language
Reward-based alignment methods for large language models (LLMs) face two key limitations: vulnerability to reward hacking, where models exploit flaws in the reward signal; and reliance on brittle, labor-intensive prompt engineering when LLMs are used as reward models. We introduce Meta Policy Optimization (MPO), a framework that addresses these challenges by integrating a meta-reward model that dynamically refines the reward model's prompt throughout training. In MPO, the meta-reward model monitors the evolving training context and continuously adjusts the reward model's prompt to maintain high alignment, providing an adaptive reward signal that resists exploitation by the policy. This meta-learning approach promotes a more stable policy optimization, and greatly reduces the need for manual reward prompt design. It yields performance on par with or better than models guided by extensively hand-crafted reward prompts. Furthermore, we show that MPO maintains its effectiveness across diverse tasks, from essay writing to mathematical reasoning, without requiring specialized reward designs. Beyond standard RLAIF, MPO's meta-learning formulation is readily extensible to higher-level alignment frameworks. Overall, this method addresses theoretical and practical challenges in reward-based RL alignment for LLMs, paving the way for more robust and adaptable alignment strategies. The code and data can be accessed at: https://github.com/minnesotanlp/mpo
title Toward Evaluative Thinking: Meta Policy Optimization with Evolving Reward Models
topic Computation and Language
url https://arxiv.org/abs/2504.20157