TRACE compiles user corrections into runtime enforcement rules for coding agents, cutting preference violations from 100% to 37.6% in-distribution and 2% out-of-distribution on ClawArena tasks while matching memory baselines on task success.
Too Correct to Learn: Reinforcement Learning on Saturated Reasoning Data
1 Pith paper cite this work. Polarity classification is still indexing.
abstract
Reinforcement Learning (RL) enhances LLM reasoning, yet a paradox emerges as models scale: strong base models saturate standard benchmarks (e.g., MATH), yielding correct but homogeneous solutions. In such environments, the lack of failure cases causes the advantage signal in group-relative algorithms (e.g., GRPO) to vanish, driving policies into mode collapse. To address this, we propose Constrained Uniform Top-K Sampling (CUTS), a parameter-free decoding strategy enforcing structure-preserving exploration. Unlike standard sampling that follows model biases, CUTS flattens the local optimization landscape by sampling uniformly from constrained high-confidence candidates. We integrate this into Mixed-CUTS, a training framework synergizing exploitative and exploratory rollouts to amplify intra-group advantage variance. Experiments on Qwen3 models demonstrate that our approach prevents policy degeneration and significantly boosts out-of-domain generalization. Notably, Mixed-CUTS improves Pass@1 accuracy on the challenging AIME25 benchmark by up to 15.1% over standard GRPO, validating that maintaining diversity within the semantic manifold is critical for rigorous reasoning.
fields
cs.LG 1years
2026 1verdicts
CONDITIONAL 1representative citing papers
citing papers explorer
-
Getting Better at Working With You: Compiling User Corrections into Runtime Enforcement for Coding Agents
TRACE compiles user corrections into runtime enforcement rules for coding agents, cutting preference violations from 100% to 37.6% in-distribution and 2% out-of-distribution on ClawArena tasks while matching memory baselines on task success.