用分类方法优化大模型强化学习的评判器,提升奖励信号质量。
Start Classifying: Categorical Critics for LLM Reinforcement Learning

- 将评判器从标量回归改为类别分类,用交叉熵训练离散价值分布。
- 在多个任务上优于传统PPO和DAPO,优势更稳定、方差更低。
- 适合需要精确奖励反馈的数学推理与工具调用场景。
大型语言模型的近端策略优化(PPO)通常采用均方误差(MSE)对标量价值目标进行评判器训练。然而,在可验证奖励的强化学习(RLVR)中,稀疏的二值奖励使评判器优化与校准尤为关键:微小的价值误差会直接扭曲PPO所依赖的标量优势。本文研究了基于分类的训练目标能否改善评判信号。HL-Gauss PPO 将标量MSE头替换为在离散价值支持上的分类预测器,通过交叉熵对平滑的HL-Gauss目标进行训练;输出解码为标量期望用于标准GAE和PPO,因此演员更新不变且非分布式。在数学推理、工具增强数学与Search-R1任务上,基于Qwen2.5与Qwen3模型,HL-Gauss PPO始终优于强基线。对照实验表明,输出头增大或二值分类本身并非增益主因。在常见推理前缀集合上,其提升Brier分数与校准误差,并生成更对称、低方差的优势。结果表明,类别价值学习是RLVR中PPO评判器的有效优化替代方案。
原文摘要 · Abstract (English)
Proximal Policy Optimization (PPO) for large language models typically trains its critic by mean-squared-error (MSE) regression on scalar value targets. Although scalar MSE is statistically valid for estimating the conditional expected return, sparse binary rewards in reinforcement learning with verifiable rewards (RLVR) make critic optimization and calibration especially consequential: small value errors directly distort the scalar advantages used by PPO. We study whether a classification-based training objective can improve this critic signal. HL-Gauss PPO replaces the scalar MSE head with a categorical predictor over a discretized value support, trained by cross-entropy against smoothed HL-Gauss targets. Its output is decoded to a scalar expectation for standard GAE and PPO; the actor update is therefore unchanged and is not distributional. Across mathematical reasoning, tool-augmented math, and Search-R1, and on both Qwen2.5 and Qwen3 backbones, HL-Gauss PPO consistently improves over strong PPO and DAPO baselines. Controls with one-hot, two-hot, and Bernoulli two-bin critics show that neither a larger output head nor binary classification alone explains the gains. On a common collection of reasoning prefixes, HL-Gauss improves Brier score and calibration error and yields more symmetric, lower-variance advantages. These results position categorical value learning as an effective optimization surrogate for PPO critics in RLVR.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。