论文

KAPPO 提出用单次采样实现 @k 训练效果,降低 RL 后训练锐化

Controlling Sharpening in LLM Post-Training

精选理由

这篇论文把 pass@k 训练的机制讲透了,还给出一个只用单次采样就能达到多采样效果的省钱方法,做 RL 训练的值得看看推导部分。

arXiv 论文研究 RL 后训练中的锐化(sharpening)现象,即准确率提升伴随行为覆盖下降。作者推导出 KL 正则化下 pass@k 与 max@k 目标的最优策略闭式解,发现 @k 优化会根据题目难度重新分配学习压力。基于此提出 KAPPO,通过提示自适应正则化或奖励变换,把现有 @1 优化方法改造成 @k 版本。在数学推理和代码生成实验中,KAPPO 单次采样训练即可匹配多次采样的 @k 优化器性能,成本大幅降低。

原文 · arXiv cs.LG

Controlling Sharpening in LLM Post-Training

LLM post-training via reinforcement learning has shown consistent improvements in accuracy at the cost of reducing behavioral coverage - a phenomenon known as sharpening that can stall progress on difficult tasks. Repeated sampling objectives such as $\mathrm{pass@k}$ and $\mathrm{max@k}$ are known to mitigate sharpening, but the mechanism behind it remains poorly understood. Moreover, existing approaches for $\mathrm{@k}$ optimization use dedicated algorithms, and no general framework exists for adapting established policy optimization methods from $\mathrm{@}1$ to $\mathrm{@k}$. We derive the optimal policies for KL-regularized $\mathrm{pass@k}$ and $\mathrm{max@k}$ objectives in closed form. Our explicit solutions reveal that $\mathrm{@k}$ optimization naturally redistributes learning pressure based on prompt difficulty, directing it toward harder problems. Crucially, we show that standard $\mathrm{@}1$ objectives achieve these exact same optimal policies through prompt-adaptive regularization or reward transformation. Ultimately, this yields an optimizer-independent explanation for why $\mathrm{@k}$ objectives reduce sharpening on easy tasks while prioritizing success on harder ones. Building on these insights, we introduce $\mathrm{@k}$ Adaptive Pressure for Policy Optimization (KAPPO). Following a one-time reference calibration, KAPPO seamlessly adapts existing $\mathrm{@}1$ methods to $\mathrm{@k}$ via prompt-adaptive regularization or reward transformation, without otherwise altering their training procedures. Experiments with several RL methods on mathematical reasoning and code generation demonstrate that KAPPO provides direct control over sharpening. Crucially, we demonstrate that $\mathrm{@k}$ pressure can be fully realized with single completion training, matching the performance of multi-completion $\mathrm{@k}$ optimizers at a fraction of the cost.