Reward Yourself: Efficient Self Rewards for Trustworthy Sampling

Mingjie Li, Wai Man Si, Michael Backes, Yang Zhang


Abstract
As high-quality data becomes harder to obtain, reward models are increasingly important. Beyond the costly RLHF stage, they are now used at inference time to guide LLM generation and in data selection for post-training. These methods bring efficiency and performance gains, but current reward models often fail to prevent untrustworthy behaviors such as privacy leaks and stereotypes. Re-training reward models to address these issues is expensive, since it requires large-scale human preference data. We propose SelfRW, a lightweight intrinsic reward that needs no extra fine-tuning or auxiliary models. By pruning current LLMs to approximate an “trust” and an “untrust” token distribution, we compute the log-probability difference as an auxiliary reward. When integrated into reward-guided sampling, SelfRW significantly reduces untrustworthy outputs while preserving task performance. It also improves reward-guided data selection, yielding better post-trained models. Experiments with two reward models and four LLMs on privacy, bias, and stereotype benchmarks show that combining SelfRW consistently improves trustworthiness (over 10% in privacy tasks and 20% in bias tasks) with minimal impact on general utility benchmarks.
Anthology ID:
2026.findings-acl.1554
Volume:
Findings of the Association for Computational Linguistics: ACL 2026
Month:
July
Year:
2026
Address:
San Diego, California, United States
Editors:
Maria Liakata, Viviane P. Moreira, Jiajun Zhang, David Jurgens
Venue:
Findings
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
31070–31086
Language:
URL:
https://preview.aclanthology.org/ingest-acl/2026.findings-acl.1554/
DOI:
Bibkey:
Cite (ACL):
Mingjie Li, Wai Man Si, Michael Backes, and Yang Zhang. 2026. Reward Yourself: Efficient Self Rewards for Trustworthy Sampling. In Findings of the Association for Computational Linguistics: ACL 2026, pages 31070–31086, San Diego, California, United States. Association for Computational Linguistics.
Cite (Informal):
Reward Yourself: Efficient Self Rewards for Trustworthy Sampling (Li et al., Findings 2026)
Copy Citation:
PDF:
https://preview.aclanthology.org/ingest-acl/2026.findings-acl.1554.pdf
Checklist:
 2026.findings-acl.1554.checklist.pdf