arXiv:2508.03058cs.LGcs.AI2025-08ACL被引 10

改进大模型微调中的价值模型,提升噪声环境下的强化学习稳定性。

VRPO: Rethinking Value Modeling for Robust RL under Noisy Supervision in LLM Post-Training

  • 用语言模型的熵和困惑度设计辅助损失,增强价值模型抗噪能力。
  • 引入变分信息瓶颈,让价值模型主动过滤噪声并捕捉关键信息。
  • 在对话、数学推理等任务中显著优于PPO和GRPO,适合高噪声场景优化。

现实世界中的强化学习常面临奖励信号模糊或不完整的问题,导致策略不稳定与泛化能力下降。此类噪声可能使模型忽略关键信息甚至优势估计崩溃。我们发现,一个强健的价值模型能有效吸收不稳定的信号,生成更密集且可靠的梯度监督,优于奖励模型。为此,我们提出VRPO框架,通过(1)基于冻结语言模型熵与困惑度的辅助损失,以及(2)变分信息瓶颈机制,使价值模型具备噪声过滤与关键词捕捉能力。该设计将价值模型从被动预测者转变为主动噪声调节器,可修正噪声奖励并生成更可靠的优势估计。在多轮对话、数学推理与科学问答任务上,无论使用规则奖励还是模型生成奖励,VRPO均持续优于PPO与GRPO等基线方法。本工作强调了价值模型在鲁棒强化学习中的核心作用,并提供了在噪声监督下进行策略优化的系统性方案。

原文摘要 · Abstract (English)

Reinforcement Learning (RL) in real-world environments often suffers from ambiguous or incomplete reward supervision, which undermines policy stability and generalization. Such noise may cause models to ignore key information or even collapse in advantage estimation. We find that a strong value model is essential for absorbing unstable signals and producing reliable advantages, offering denser and more robust supervision than the reward model. To better optimize noisy supervision, we propose VRPO, a framework that enhances value modeling for robust RL in LLM post-training. VRPO integrates (1) auxiliary losses guided by entropy and perplexity from a frozen language model, and (2) a variational information bottleneck, enabling the value model to filter noise and capture key words. This design allows the value model to correct noise rewards and generate more reliable advantage estimates, transforming it from a passive predictor into an active noise regulator. Experiments on multi-turn dialogue, math reasoning, and science QA with both rule-based and model-based rewards show that VRPO consistently outperforms baselines such as PPO and GRPO. Our work highlight the central role of the value model in Robust RL and provide a principled and practical approach to policy optimization under noisy supervision.

强化学习大模型微调噪声鲁棒价值建模

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。