Beyond Binary Rewards: A Comparative Study of Reward Design for Reinforcement Unlearning
On the map: LLM safety and security → Machine unlearning
Abstract (arXiv)
Machine unlearning seeks to selectively remove specific knowledge from trained language models without full retraining, a growing necessity under privacy regulations such as GDPR and the EU AI Act. Recent work has reformulated unlearning as a Reinforcement Learning with Verifiable Rewards (RLVR) problem, where models are optimized against verifiable rewards computed directly from their outputs. However, existing methods rely on sparse binary rewards that provide minimal learning signal, indicating only whether forbidden content was avoided, and limiting convergence speed. In this paper, we study how reward design affects unlearning efficiency within the Reinforcement Unlearning (RUL) framework. We introduce a principled reward decomposition framework that decouples verifiability from sparsity, and propose two new reward functions: an exponential reward that provides graded penalties based on the count of forbidden-concept occurrences, and a PageRank inspired reward that weights penalties by semantic importance. We conduct experiments on the Real World Knowledge Unlearning (RWKU) benchmark, demonstrating that both rewards consistently outperform the binary setting, while reaching similar forgetting performance up to 3× faster and preserving general model utility. Our results show that reward design is a key driver of unlearning efficiency offering a practical path toward scalable and efficient machine unlearning.
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Placed in Machine unlearning, one of 12 directions in LLM safety and security. Builds on 30 of its 39 references that are in the map. First posted on arXiv 30 Jul 2026.
Builds on
30 of 39 references.
Closest ancestors · 20
Same direction, newest first
- 2026Reinforcement Unlearning via Group Relative Policy Optimization28 Jan
- 2025OpenUnlearning: Accelerating LLM Unlearning via Unified Benchmarking of Methods and Metrics14 Jun
- 2025RULE: Reinforcement UnLEarning Achieves Forget-Retain Pareto Optimality8 Jun
- 2025A Comprehensive Survey of Machine Unlearning Techniques for Large Language Models22 Feb
- 2024LLM Unlearning via Loss Adjustment with Only Forget Data14 Oct
- 2024A Closer Look at Machine Unlearning for Large Language Models10 Oct
- 2024A Probabilistic Perspective on Unlearning and Alignment for Large Language Models4 Oct
- 2024Alternate Preference Optimization for Unlearning Factual Knowledge in Large Language Models20 Sep
- 2024Machine Unlearning in Generative AI: A Survey30 Jul
- 2024MUSE: Machine Unlearning Six-Way Evaluation for Language Models8 Jul
- 2024RWKU: Benchmarking Real-World Knowledge Unlearning for Large Language Models16 Jun
- 2024Negative Preference Optimization: From Catastrophic Collapse to Effective Unlearning8 Apr
8 more
- 2024Rethinking Machine Unlearning for Large Language Models13 Feb
- 2024TOFU: A Task of Fictitious Unlearning for LLMs11 Jan
- 2023Who's Harry Potter? Approximate Unlearning in LLMs3 Oct
- 2022Knowledge Unlearning for Mitigating Privacy Risks in Language Models4 Oct
- 2022A Survey of Machine Unlearning6 Sep
- 2022Quark: Controllable Text Generation with Reinforced Unlearning26 May
- 2021Remember What You Want to Forget: Algorithms for Machine Unlearning4 Mar
- 2021Machine Unlearning via Algorithmic Stability25 Feb
Other references · 10
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 9,379
- 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 8,065
- 2019Machine Unlearningcited by 1,061
- 2024Tulu 3: Pushing Frontiers in Open Language Model Post-Trainingcited by 1,053
- 2019Certified Data Removal from Machine Learning Modelscited by 535
- 2024The WMDP Benchmark: Measuring and Reducing Malicious Use With Unlearningcited by 509
- 2019Making AI Forget You: Data Deletion in Machine Learningcited by 465
- 2020Descent-to-Delete: Gradient-Based Methods for Machine Unlearningcited by 311
- 2023ProPILE: Probing Privacy Leakage in Large Language Modelscited by 129
- 2025The European Union general data protection regulation: what it is and what it meanscited by 31
Built on by
Citing papers per year: 2026: 1.
1 paper.
Similar, not cited
Close in wording, no citation either way
- 2026Binary Rewards and Reinforcement Learning: Fundamental Challenges
- 2025Exploration vs Exploitation: Rethinking RLVR through Clipping, Entropy, and Spurious Reward
- 2025Position: The Hidden Costs and Measurement Gaps of Reinforcement Learning with Verifiable Rewards
- 2026The Unlearnability Phenomenon in RLVR for Language Models
- 2025ConfClip: Confidence-Weighted and Clipped Reward for Reinforcement Learning in LLMs
- 2025The Choice of Divergence: A Neglected Key to Mitigating Diversity Collapse in Reinforcement Learning with Verifiable Reward
- 2025Extending RLVR to Open-Ended Tasks via Verifiable Multiple-Choice Reformulation
- 2025Crossing the Reward Bridge: Expanding RL with Verifiable Rewards Across Diverse Domains