DVPO通过建模分值分布提升大模型后训练的鲁棒性与泛化能力。
DVPO: Distributional Value Modeling-based Policy Optimization for LLM Post-Training
- 基于分值分布建模,提供细粒度监督信号。
- 对上下尾部进行不对称正则,抑制噪声负反馈并保留探索多样性。
- 在多轮对话、数学推理等任务中优于传统方法,适合真实场景部署。
强化学习在大模型后训练中表现优异,但现实部署常面临噪声或不完整监督。此时复杂且不可靠的监督信号易导致训练不稳定并损害泛化能力。现有方法如最坏情况优化(如RFQI、CQL)和均值基方法(如PPO、GRPO)虽提升稳定性,却常忽视泛化,产生过于保守策略,导致不同场景下表现不均。为此,我们提出DVPO(基于分布值建模的风险感知策略优化),融合条件风险理论与分布值建模,更好平衡鲁棒性与泛化性。DVPO学习词元级价值分布以提供细粒度监督,并采用非对称风险正则化调控分布尾部:压缩下尾以抑制噪声负偏差,扩展上尾以保留探索多样性。在多轮对话、数学推理和科学问答等任务的大量实验中,DVPO在噪声监督下持续优于PPO、GRPO及基于鲁棒贝尔曼的PPO,展现了其在真实世界大模型后训练中的潜力。
原文摘要 · Abstract (English)
Reinforcement learning (RL) has shown strong performance in LLM post-training, but real-world deployment often involves noisy or incomplete supervision. In such settings, complex and unreliable supervision signals can destabilize training and harm generalization. While existing approaches such as worst-case optimization (e.g., RFQI, CQL) and mean-based methods (e.g., PPO, GRPO) can improve stability, they often overlook generalization and may produce overly conservative policies, leading to uneven performance across diverse real scenarios. To this end, we introduce DVPO (Distributional Value Modeling with Risk-aware Policy Optimization), a new RL framework that combines conditional risk theory with distributional value modeling to better balance robustness and generalization. DVPO learns token-level value distributions to provide fine-grained supervision, and applies an asymmetric risk regularization to shape the distribution tails: it contracts the lower tail to dampen noisy negative deviations, while expanding the upper tail to preserve exploratory diversity. Across extensive experiments and analysis in multi-turn dialogue, math reasoning, and scientific QA, DVPO consistently outperforms PPO, GRPO, and robust Bellman-based PPO under noisy supervision, showing its potential for LLM post-training in the real-world.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。