Which Rollout Taught It That? BehaviorTrace and the Limits of Training-Data Attribution in Online RL
Not on the map. Why?
Abstract (arXiv)
When reinforcement learning teaches a language model a new behavior, can we find the training rollouts that taught it? And when an attribution method says it can, how do we know the answer is real? We study both questions on online RL fine-tuning with GRPO, using a planted behavior with a known cause. We release BehaviorTrace, an open evaluation harness that combines full-gradient sketching, the planted-behavior setup, and controls for gradient magnitude, fluency, headroom, and variation across seeds and generation draws. Across three seeds on Qwen2.5-1.5B, much of the apparent attribution signal comes from confounds. A control that ranks training steps by gradient size alone, with no behavior target, reaches 4.2 to 4.5 times chance and matches or beats the best targeted estimator on two of three seeds. At saturated checkpoints, model fluency predicts the behavior label at least as well as every gradient method we compared it with. Once fluency is controlled, the per-rollout results change from seed to seed and from one generation draw to the next, so a single run cannot settle the question. One signal does hold on all three seeds. The gradient of the trigger tokens aligns with a target built where the behavior actually occurs. We turn these findings into a checklist for evaluating attribution in RL. We test existing estimators, including GAS (renormalized TracInCP) and a TRAK-style estimator, and do not propose a new one.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 6 of its 15 references that are in the map. First posted on arXiv 7 Oct 2026.
Builds on
6 of 15 references.
References · 6
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
- 2024Qwen2.5 Technical Reportcited by 4,713
- 2026Probe-Based Data Attribution: Discovering and Mitigating Undesirable Behaviors in LLM Post-Trainingcited by 5
- 2026Reinforcement Learning Can Amplify Emergent Misalignment from Harmless Rewardscited by 2
- 2026Data Attribution of Emergent Misalignment with Persona Featurescited by 1
- 2026Sycophantic Agreement Transfers with Neutral Data via Contrastive Preference Optimizationcited by 1
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2025Behavior Injection: Preparing Language Models for Reinforcement Learning
- 2025A Snapshot of Influence: A Local Data Attribution Framework for Online Reinforcement Learning
- 2026Spend Your Rollouts Where It Counts: Rollout Allocation for Group-Based RL Post-Training
- 2026Demystifying Reinforcement Learning Post-Training of Language Models
- 2026Towards Better RL Training Data Utilization via Second-Order Rollout
- 2025Posterior Behavioral Cloning: Pretraining BC Policies for Efficient RL Finetuning
- 2025RL in Name Only? Analyzing the Structural Assumptions in RL post-training for LLMs
- 2026Dynamics-Predictive Sampling for Active RL Finetuning of Large Reasoning Models