Gradient Routing Localizes and Removes Unintended Behaviors in RL
On the map: LLM alignment / RLHF / preference optimisation → Reward hacking and misspecification
Abstract (arXiv)
We propose a mitigation to reward hacking in reinforcement learning. Our setup involves a misspecified reward function that gives rise to unintended behavior, and a monitor used to detect unintended behavior. As with frontier AI systems, the monitor makes systematic errors. Consequently, using the monitor for data filtering or to apply a reward penalty fails to prevent the model from learning unintended behavior ("reward hacking"). To address this problem, we introduce GRAFT: Gradient Routing Adapter Fine Tuning. Rather than penalizing or filtering flagged samples during training, GRAFT constrains gradient updates on those samples to a sacrificial set of new parameters. This has two benefits: First, because the monitor is not part of the reward function, the model does not learn to hide misbehavior. Second, ablating the sacrificial parameters reduces or entirely stops the unintended behavior while maintaining improved performance on target tasks. We demonstrate these benefits in RL environments where reward penalties cause models to learn to reward hack in undetected ways, training models ranging from 135M to 8B parameters. Additionally, we show a similar effect at a larger scale by distilling frontier LLM trajectories containing reward hacking into Qwen3-32B, without teaching that model to reward hack. Overall, our work suggests GRAFT may be a promising approach for leveraging imperfect monitors during post-training in settings where true performance is difficult to evaluate.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in Reward hacking and misspecification, one of 10 directions in LLM alignment / RLHF / preference optimisation. Builds on 38 of its 58 references that are in the map. First posted on arXiv 12 Oct 2026.
Builds on
38 of 58 references.
Closest ancestors · 6
Same direction, newest first
- 2026Debate Training Reduces Reward Hacking in RLAIF18 Aug
- 2026Countdown-Code: A Testbed for Studying The Emergence and Generalization of Reward Hacking in RLVR7 Mar
- 2025Natural Emergent Misalignment from Reward Hacking in Production RL23 Nov
- 2025Inoculation Prompting: Instructing LLMs to misbehave at train-time improves test-time alignment6 Oct
- 2025Monitoring Reasoning Models for Misbehavior and the Risks of Promoting Obfuscation14 Mar
- 2022Defining and Characterizing Reward Hacking27 Sep
Other references · 32
- 2022Training language models to follow instructions with human feedbackcited by 15,971
- 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,380
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 9,379
- 2021Evaluating Large Language Models Trained on Codecited by 8,320
- 2025DAPO: An Open-Source LLM Reinforcement Learning System at Scalecited by 2,810
- 2016Concrete Problems in AI Safetycited by 1,642
- 2022Scaling Laws for Reward Model Overoptimizationcited by 1,095
- 2023Safe RLHF: Safe Reinforcement Learning from Human Feedbackcited by 656
- 2026Terminal-Bench: Benchmarking Agents on Hard, Realistic Tasks in Command Line Interfacescited by 585
- 2023Weak-to-Strong Generalization: Eliciting Strong Capabilities With Weak Supervisioncited by 467
- 2025Persona Vectors: Monitoring and Controlling Character Traits in Language Modelscited by 345
- 2025SmolLM2: When Smol Goes Big -- Data-Centric Training of a Small Language Modelcited by 335
20 more
- 2025Chain of Thought Monitorability: A New and Fragile Opportunity for AI Safetycited by 249
- 2018Categorizing Variants of Goodhart's Lawcited by 149
- 2024ODIN: Disentangled Reward Mitigates Hacking in RLHFcited by 136
- 2021DEMix Layers: Disentangling Domains for Modular Language Modelingcited by 132
- 2021Inducing Causal Structure for Interpretable Neural Networkscited by 110
- 2024Easy-to-Hard Generalization: Scalable Alignment Beyond Human Supervisioncited by 96
- 2024Prover-Verifier Games improve legibility of LLM outputscited by 85
- 2021Consequences of Misaligned AIcited by 80
- 2025LeetCodeDataset: A Temporal Dataset for Robust Evaluation and Efficient Training of Code LLMscited by 70
- 2018Incomplete Contracting and AI Alignmentcited by 62
- 2025Inoculation Prompting: Eliciting traits from LLMs during training can suppress them at test-timecited by 43
- 2024Quantifying the Gain in Weak-to-Strong Generalizationcited by 41
- 2024Gradient Routing: Masking Gradients to Localize Computation in Neural Networkscited by 34
- 2023Making Harmful Behaviors Unlearnable for Large Language Modelscited by 20
- 2026Terminal Wrench: A Dataset of 331 Reward-Hackable Environments and 3,632 Exploit Trajectoriescited by 15
- 2025Recontextualization Mitigates Specification Gaming without Modifying the Specificationcited by 14
- 2026The Obfuscation Atlas: Mapping Where Honesty Emerges in RLVR with Deception Probescited by 12
- 2025Beyond Data Filtering: Knowledge Localization for Capability Removal in LLMscited by 11
- 2026Inoculation Adapters: Improved Selective Generalization of Capabilities with Fewer Surprising Backdoorscited by 3
- 2026Modular Pretraining Enables Access Controlcited by 3
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026Detecting and Suppressing Reward Hacking with Gradient Fingerprints
- 2026Reproducing, Analyzing, and Detecting Reward Hacking in Rubric-Based Reinforcement Learning
- 2026Gradient Regularization Mitigates Reward Hacking in Reinforcement Learning from Human Feedback and Verifiable Rewards
- 2026Adversarial Reward Auditing for Active Detection and Mitigation of Reward Hacking
- 2026Reward Hacking Benchmark: Measuring Exploits in LLM Agents with Tool Use
- 2026From Rebound to Remedy: Understanding and Mitigating Reward Hacking via Representation Engineering
- 2026Directional Alignment Mitigates Reward Hacking in Reinforcement Learning for Language Models
- 2024Correlated Proxies: A New Definition and Improved Mitigation for Reward Hacking