Papers
arxiv:2610.16105

Beyond Policy Rollouts: Embedding-Space Directional Exploration with Anchored Noise for LLM RLVR

Published on Oct 12
· Submitted by
Dong Gong
on Oct 16
Authors:
,
,
,

Abstract

EDEN improves reinforcement learning with verifiable rewards by injecting anchored embedding noise into rollout conditioning to increase diversity and strengthen group-relative advantage signals without altering inference.

Reinforcement learning with verifiable rewards (RLVR) typically uses small rollout groups per prompt, where repeated sampling from the same policy can yield highly similar trajectories, reducing within-group reward variation and weakening group-relative advantage signals. Existing remedies such as higher sampling temperature, entropy bonuses, asymmetric clipping, and larger rollout budgets broaden sampling or modify policy updates but leave rollout conditioning unchanged. In this paper, we propose EDEN (Embedding-space Directional Exploration via anchored Noise), which intervenes at the rollout source through anchored input perturbations. EDEN explores beyond the policy's own rollouts. Within the same rollout budget, it replaces a small fraction of standard rollouts in each group with rollouts that the same policy generates from the prompt plus a single appended, perturbed embedding position, rather than from the prompt alone. Gaussian noise drives variation, while a shared anchor provides a common conditioning direction within each prompt group. Selected perturbed rollouts contribute to GRPO updates with downweighted advantages, and the model architecture, discrete decoding, and test-time inference remain unchanged. Across model scales and mathematical reasoning benchmarks, EDEN outperforms GRPO and the exploration controls evaluated in the same pipeline, with the largest gains in sampled solution coverage. Analyses show that more prompt groups provide non-zero group-relative learning signals and that noise and anchoring play complementary roles. Intervening at the rollout source is thus an effective lever for exploration in RLVR.

Community

Paper submitter

We introduce EDEN, which helps LLM RL post-training explore by adding anchored noise to the rollout’s input in embedding space, so the model can reach solutions beyond its usual sampling.
In RLVR, a policy sampling the same prompt several times often produces nearly identical trajectories, which leaves little reward variation for group-relative learning. We tackle this at the input itself. For each prompt, EDEN first samples standard rollouts, then generates an extra set of rollouts from inputs perturbed in embedding space. The Gaussian noise creates variation, and a shared anchor embedding gives every rollout in the group a common conditioning direction. This makes the exploration directional rather than random. We bring the most informative perturbed rollouts into training through an advantage-gated filter and noise-aware advantage reweighting. On Qwen2.5-Math-1.5B and 7B, EDEN outperforms GRPO on average across English and Chinese math benchmarks, from grade-school to olympiad level. It raises average Pass@256 by 6.67 points, and by 13.33 points on AIME’25 at 7B. Our analyses show that EDEN gives more prompt groups a non-zero learning signal, and that the noise and the anchor play complementary roles.

Sign up or log in to comment

Get this paper in your agent:

hf papers read 2610.16105
Don't have the latest CLI?
curl -LsSf https://hf.co/cli/install.sh | bash

Models citing this paper 0

No model linking this paper

Cite arxiv.org/abs/2610.16105 in a model README.md to link it from this page.

Datasets citing this paper 0

No dataset linking this paper

Cite arxiv.org/abs/2610.16105 in a dataset README.md to link it from this page.

Spaces citing this paper 0

No Space linking this paper

Cite arxiv.org/abs/2610.16105 in a Space README.md to link it from this page.

Collections including this paper 0

No Collection including this paper

Add this paper to a collection to link it from this page.