稳定高效训练评论家的方法

How to Train a Critic Stably and Efficiently

精选理由

这篇论文提出了一种针对大型语言模型训练中评论家不稳定性的解决方案,结合多种优化方法,效果显著,值得一看。

AI 摘要

研究针对大型语言模型训练中评论家不稳定性的问题,提出Best-Practice Critic Optimization (BPCO)方法,结合DPPO、价值预测、蒙特卡洛价值目标等,在数学推理任务中显著提升评论家性能,并匹配或超过基于群体的基线。代码开源。

原文 · arXiv cs.AI

How to Train a Critic Stably and Efficiently

Group-based reinforcement learning methods such as GRPO for large language models avoid training a critic by sampling multiple responses for each prompt. A reliable critic could instead estimate token-level advantages from one response, but standard critic-based training recipes are often unstable. We study this instability and develop \textbf{Best-Practice Critic Optimization (BPCO)}, a recipe that combines DPPO, value predictions bounded to the reward range, Monte Carlo value targets, unnormalized policy advantages, and length-adaptive generalized advantage estimation. Because the critic is used only during training, BPCO can also condition it on reward-defining information, such as a reference answer or grading rubric, that is hidden from the policy. Controlled experiments isolate the effect of each design choice. Across mathematical reasoning tasks with models ranging from 1.5B parameters to 30B-A3B mixtures of experts, BPCO improves a strong critic-based baseline consistently, and matches or exceeds a group-based baseline while sampling one response per prompt. The same recipe also improves learning with rubric-based rewards. These results show that a carefully designed critic provides a reliable alternative to group-relative advantage estimation. Code is available at https://github.com/QPHutu/golden_critic