RewardUQ: A Unified Framework for Uncertainty-Aware Reward Models
On the map: LLM alignment / RLHF / preference optimisation → Reward modeling and RLHF
Abstract (arXiv)
Reward models are central to aligning large language models (LLMs) with human preferences. Yet most approaches rely on pointwise reward estimates that overlook the epistemic uncertainty in reward models arising from limited human feedback. Recent work suggests that quantifying this uncertainty can reduce the costs of human annotation via uncertainty-guided active learning and mitigate reward overoptimization in LLM post-training. However, uncertainty-aware reward models have so far been adopted without thorough comparison, leaving them poorly understood. This work introduces a unified framework, RewardUQ, to systematically evaluate uncertainty quantification for reward models. We compare common methods along standard metrics measuring accuracy and calibration, and we propose a new ranking strategy incorporating both dimensions for a simplified comparison. Our experimental results suggest that model size and initialization have the most meaningful impact on performance, and most prior work could have benefited from alternative design choices. To foster the development and evaluation of new methods and aid the deployment in downstream applications, we release our open-source framework as a Python package. Our code is available at https://github.com/lasgroup/rewarduq.
Key facts. Cited by 3 papers in the map (data through 7 Oct 2026). Placed in Reward modeling and RLHF, one of 10 directions in LLM alignment / RLHF / preference optimisation. Builds on 47 of its 60 references that are in the map. First posted on arXiv 27 Feb 2026.
Builds on
47 of 60 references.
Closest ancestors · 25
Same direction, newest first
- 2025Uncertainty Quantification for Large Language Model Reward Learning under Heterogeneous Human Feedback2 Dec
- 2025Efficient Reinforcement Learning from Human Feedback via Bayesian Preference Inference6 Nov
- 2025Skywork-Reward-V2: Scaling Preference Data Curation via Human-AI Synergy2 Jul
- 2024Towards Reliable Alignment: Uncertainty-aware RLHF31 Oct
- 2024Skywork-Reward: Bag of Tricks for Reward Modeling in LLMs24 Oct
- 2024Uncertainty-aware Reward Model: Teaching Reward Models to Know What is Unknown1 Oct
- 2024Reward-Robust RLHF in LLMs18 Sep
- 2024Towards Data-Centric RLHF: Simple Metrics for Preference Dataset Comparison15 Sep
- 2024RewardBench: Evaluating Reward Models for Language Modeling20 Mar
- 2024Bayesian Reward Models for LLM Alignment20 Feb
- 2024Active Preference Optimization for Sample Efficient RLHF16 Feb
- 2024Reinforcement Learning from Human Feedback with Active Queries14 Feb
13 more
- 2024Improving Reinforcement Learning from Human Feedback with Efficient Reward Model Ensemble30 Jan
- 2023Uncertainty-Penalized Reinforcement Learning from Human Feedback with Diverse Reward LoRA Ensembles30 Dec
- 2023Helping or Herding? Reward Model Ensembles Mitigate but do not Eliminate Reward Hacking14 Dec
- 2023Distributional Preference Learning: Understanding and Accounting for Hidden Context in RLHF13 Dec
- 2023Reward Model Ensembles Help Mitigate Overoptimization4 Oct
- 2023Open Problems and Fundamental Limitations of Reinforcement Learning from Human Feedback27 Jul
- 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Model29 May
- 2022Scaling Laws for Reward Model Overoptimization19 Oct
- 2022Reward Uncertainty for Exploration in Preference-based Reinforcement Learning24 May
- 2022Training a Helpful and Harmless Assistant with Reinforcement Learning from Human Feedback12 Apr
- 2022Uncertainty Estimation for Language Reward Models14 Mar
- 2019Fine-Tuning Language Models from Human Preferences18 Sep
- 2017Deep reinforcement learning from human preferences12 Jun
Other references · 22
- 2022Training language models to follow instructions with human feedbackcited by 15,712
- 2017Proximal Policy Optimization Algorithmscited by 14,441
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
- 2025Qwen3 Technical Reportcited by 8,195
- 2017On Calibration of Modern Neural Networkscited by 4,860
- 2015Dropout as a Bayesian Approximation: Representing Model Uncertainty in Deep Learningcited by 4,416
- 2020Learning to summarize from human feedbackcited by 2,562
- 2017What Uncertainties Do We Need in Bayesian Deep Learning for Computer Vision?cited by 2,335
- 2016Concrete Problems in AI Safetycited by 1,582
- 2024Tulu 3: Pushing Frontiers in Open Language Model Post-Trainingcited by 988
- 2019Aleatoric and Epistemic Uncertainty in Machine Learning: An Introduction to Concepts and Methodscited by 903
- 2023UltraFeedback: Boosting Language Models with Scaled AI Feedbackcited by 394
10 more
- 2023LoRA ensembles for large language model fine-tuningcited by 65
- 2024Active Preference Learning for Large Language Modelscited by 55
- 2024Efficient Exploration for LLMscited by 44
- 2024LoRA-Ensemble: Efficient Uncertainty Modelling for Self-Attention Networkscited by 20
- 2025Understanding Model Calibration -- A gentle introduction and visual exploration of calibration and the expected calibration error (ECE)cited by 20
- 2024Deep Bayesian Active Learning for Preference Modeling in Large Language Modelscited by 19
- 2023Sample Efficient Preference Alignment in LLMs via Active Explorationcited by 18
- 2024Sample-Efficient Alignment for LLMscited by 16
- 2022Near-optimal Policy Identification in Active Reinforcement Learningcited by 7
- 2024Uncertainty-Penalized Direct Preference Optimizationcited by 6
Built on by
Citing papers per year: 2026: 3.
3 papers.
Similar, not cited
Close in wording, no citation either way
- 2026Uncertainty-Aware Reward Modeling for Stable RLHF
- 2025RewardBench 2: Advancing Reward Model Evaluation
- 2025Establishing Reliability Metrics for Reward Models in Large Language Models
- 2024Post-hoc Reward Calibration: A Case Study on Length Bias
- 2024Improving Instruction Following in Language Models through Proxy-Based Uncertainty Estimation
- 2024RM-Bench: Benchmarking Reward Models of Language Models with Subtlety and Style
- 2025Reward Model Interpretability via Optimal and Pessimal Tokens
- 2025Two Minds Better Than One: Collaborative Reward Modeling for LLM Alignment