arXiv:2605.19416cs.CL2026-05

用配对偏好替代平均值,让大模型推理更精准。

LambdaPO: A Lambda Style Policy Optimization for Reasoning Language Models

  • 将优势估计从单一均值改为动态加权的配对偏好结构。
  • 在数学推理任务上显著优于基线方法,提升模型优化精度。
  • 适合需要精细推理对齐的复杂任务,如数学证明与问答。

组相对策略优化(GRPO)已成为现代强化学习对齐的核心方法,因其无需显式价值评判器,通过采样轨迹组间的奖励归一化实现高效对齐。然而,该方法依赖单一统计基准(如组均值),将轨迹空间的相对拓扑压缩为标量,丢失了复杂、排名敏感奖励景观中至关重要的细粒度偏好信息。为此,我们提出新型框架LambdaPO,通过将优势估计从标量重构为分解的配对偏好结构,解决这一信息瓶颈。具体而言,任意轨迹的优势被定义为相对于同组其他轨迹的奖励差值之和,每一对比较结果根据策略自身对偏好判断的置信度进行动态衰减。为进一步缓解二元监督信号稀疏问题,我们引入语义密度奖励,基于生成推理路径与真实解之间的精确率-召回率对齐计算。实验结果表明,该方法能从一组采样轨迹中挖掘更丰富的优化信号,在多个挑战性数学推理与问答任务中表现优于基线方法。

原文摘要 · Abstract (English)

Group Relative Policy Optimization(GRPO) has become a cornerstone of modern reinforcement learning alignment, prized for its efficacy in foregoing an explicit value-critic by leveraging reward normalization across sampled trajectory cohorts. However, the method's reliance on a monolithic statistical baseline, such as the group mean, collapses the relational topology of the trajectory space into a single scalar, thereby erasing the fine-grained preference information essential for navigating complex, rank-sensitive reward landscapes. To address this issue, we introduce a novel framework, Lambda Policy Optimization (LambdaPO), that addresses this information-theoretic bottleneck by re-conceptualizing advantage estimation from a scalar value to a decomposed, pairwise preference structure. Specifically, the advantage for any given trajectory is formulated as the integrated sum of reward differentials against all peers in its cohort, where each pairwise comparison is dynamically attenuated by the policy's own probabilistic confidence in the established preference. To further mitigate the sparsity of binary outcome supervision, we augment the objective with a semantic density reward, derived from the precision-recall alignment between generated reasoning traces and ground-truth solutions. As a result, our method can mine more fine-grained optimization signals from a group of rollouts, guiding the LLM to a better optima. Experimental results across challenging math reasoning and question-answering tasks demonstrates that LambdaPO improves performance compared to the baseline methods.

大模型对齐强化学习推理优化

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