CurveRL: 用于LLM推理的基于分布感知的上下文重加权原则
CurveRL: Principled Distribution-Aware Context Reweighting for LLM Reasoning
AI总结:
本文提出CurveRL方法,通过分位数坐标变换实现分布感知的提示重加权,在RLVR框架下统一优化理论并显著提升推理性能。
摘要:
上下文或提示级别的重加权已成为使用验证奖励的强化学习(RLVR)中提升大型语言模型推理能力的关键算法杠杆,但决定最优加权的原则仍不清楚。我们通过将提示重加权公式化为通过率函数空间中定义的效用泛函的泛函导数来解决这一差距,从而产生一个统一的优化框架,该框架能够容纳现有方案,包括REINFORCE和GRPO。在此优化框架的基础上,我们提出了一种基于分位数坐标变换的分布感知提示重加权方法,称为CurveRL,其中分配给每个提示的权重不取决于通过率的绝对值,而是取决于其排名和密度,以反映学习动态中通过率的分布结构。跨多个基准的大量实验表明,我们提出的CurveRL始终优于GRPO和其他RLVR基线。我们的研究将上下文分布控制确定为分析和设计提示重加权RLVR算法的原则性轴心。代码发布在https://github.com/zhyzmath/CurveRL。
英文摘要:
Context or prompt-level reweighting has emerged as a central algorithmic lever in Reinforcement Learning with Verified Rewards (RLVR) for improving the reasoning capability of large language models, yet the principle determining what constitutes an optimal weighting remains poorly understood. We address this gap by formulating prompt reweighting as a functional derivative of a utility functional defined in the pass-rate function space, yielding a unified optimality framework that accommodates existing schemes, including REINFORCE and GRPO. Building on this optimality framework, we propose a distribution-aware prompt reweighting approach, called CurveRL, based on a quantile coordinate transform, in which the weight assigned to each prompt depends not on the absolute value of pass rates but on its rank and density to reflect the distributional structure of the pass rates in the learning dynamics. Extensive experiments across multiple benchmarks demonstrate that our proposed CurveRL consistently outperforms GRPO and other RLVR baselines. Our study identifies context-distribution control as a principled axis for analyzing and designing prompt-reweighted RLVR algorithms. The code is released in https://github.com/zhyzmath/CurveRL.
机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。