Search papers, labs, and topics across Lattice.
The paper introduces In-Context RLVR, a reinforcement learning method that improves reasoning in large language models by weighting training examples based on the quality of their reasoning traces. It leverages the policy model's in-context learning ability to estimate the "Demonstration Utility" of a solution, quantified as "Evidence Gain," which implicitly reweights rewards to favor high-quality reasoning traces. Experiments on mathematical benchmarks demonstrate that In-Context RLVR outperforms standard RLVR in both accuracy and reasoning quality.
Stop training LLMs on lucky guesses: this new RL method uses the model's own in-context learning ability to identify and upweight high-quality reasoning traces, leading to better performance.
Reinforcement Learning with Verifiable Rewards (RLVR) improves reasoning in large language models but treats all correct solutions equally, potentially reinforcing flawed traces that get correct answers by chance. We observe that better reasoning are better teachers: high-quality solutions serve as more effective demonstrations than low-quality ones. We term this teaching ability Demonstration Utility, and show that the policy model's own in-context learning ability provides an efficient way to measure it, yielding a quality signal termed Evidence Gain. To employ this signal during training, we introduce In-Context RLVR. By Bayesian analysis, we show that this objective implicitly reweights rewards by Evidence Gain, assigning higher weights to high-quality traces and lower weights to low-quality ones, without requiring costly computation or external evaluators. Experiments on mathematical benchmarks show improvements in both accuracy and reasoning quality over standard RLVR.
