CausalRM: Causal-Theoretic Reward Modeling for RLHF from Observational User Feedbacks
On the map: LLM alignment / RLHF / preference optimisation → Reward modeling and RLHF
Abstract (arXiv)
Despite the success of reinforcement learning from human feedback (RLHF) in aligning language models, current reward modeling heavily relies on experimental feedback data collected from human annotators under controlled and costly conditions. In this work, we introduce observational reward modeling -- learning reward models with observational user feedback (e.g., clicks, copies, and upvotes) -- as a scalable and cost-effective alternative. We identify two fundamental challenges in this setting: (1) observational feedback is noisy due to annotation errors, which deviates it from true user preference; (2) observational feedback is biased by user preference, where users preferentially provide feedback on responses they feel strongly about, which creats a distribution shift between training and inference data. To address these challenges, we propose CausalRM, a causal-theoretic reward modeling framework that aims to learn unbiased reward models from observational feedback. To tackle challenge (1), CausalRM introduces a noise-aware surrogate loss term that is provably equivalent to the primal loss under noise-free conditions by explicitly modeling the annotation error generation process. To tackle challenge (2), CausalRM uses propensity scores -- the probability of a user providing feedback for a given response -- to reweight training samples, yielding a loss function that eliminates user preference bias. Extensive experiments across diverse LLM backbones and benchmark datasets validate that CausalRM effectively learns accurate reward signals from noisy and biased observational feedback and delivers substantial performance improvements on downstream RLHF tasks -- including a 49.2% gain on WildGuardMix and a 32.7% improvement on HarmBench. Code is available on our project website.
Key facts. Cited by 6 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 40 of its 60 references that are in the map. First posted on arXiv 19 Mar 2026.
Builds on
40 of 60 references.
Closest ancestors · 5
Same direction, newest first
- 2025On Symmetric Losses for Robust Policy Optimization with Noisy Preferences30 May
- 2024Interpretable Preferences via Multi-Objective Reward Modeling and Mixture-of-Experts18 Jun
- 2024RLHF Workflow: From Reward Modeling to Online RLHF13 May
- 2024InfoRM: Mitigating Reward Hacking in RLHF via Information-Theoretic Reward Modeling14 Feb
- 2024ODIN: Disentangled Reward Mitigates Hacking in RLHF11 Feb
Other references · 35
- 2023GPT-4 Technical Reportcited by 18,132
- 2022Training language models to follow instructions with human feedbackcited by 15,712
- 2017Proximal Policy Optimization Algorithmscited by 14,441
- 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 12,865
- 2025DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learningcited by 6,146
- 2025Gemini 2.5: Pushing the Frontier with Advanced Reasoning, Multimodality, Long Context, and Next Generation Agentic Capabilitiescited by 3,655
- 2024HarmBench: A Standardized Evaluation Framework for Automated Red Teaming and Robust Refusalcited by 1,330
- 2018Co-teaching: Robust Training of Deep Neural Networks with Extremely Noisy Labelscited by 1,053
- 2025Group Sequence Policy Optimizationcited by 788
- 2016Making Deep Neural Networks Robust to Label Noise: a Loss Correction Approachcited by 663
- 2023Bias and Fairness in Large Language Models: A Surveycited by 626
- 2024Generative Verifiers: Reward Modeling as Next-Token Predictioncited by 465
23 more
- 2024WildGuard: Open One-Stop Moderation Tools for Safety Risks, Jailbreaks, and Refusals of LLMscited by 460
- 2023UltraFeedback: Boosting Language Models with Scaled AI Feedbackcited by 394
- 2024PKU-SafeRLHF: Towards Multi-Level Safety Alignment for LLMs with Human Preferencecited by 252
- 2023HelpSteer: Multi-attribute Helpfulness Dataset for SteerLMcited by 140
- 2024Provably Robust DPO: Aligning Language Models with Noisy Feedbackcited by 133
- 2025DeepAgent: A General Reasoning Agent with Scalable Toolsetscited by 69
- 2024Towards Robust Alignment of Language Models: Distributionally Robustifying Direct Preference Optimizationcited by 52
- 2022ESCM2: Entire Space Counterfactual Multi-Task Model for Post-Click Conversion Rate Estimationcited by 47
- 2019Large-scale Causal Approaches to Debiasing Post-click Conversion Rate Estimation with Multi-task Learningcited by 39
- 2023Optimal Transport for Treatment Effect Estimationcited by 32
- 2020Information Theoretic Counterfactual Learning from Missing-Not-At-Random Feedbackcited by 30
- 2022StableDR: Stabilized Doubly Robust Learning for Recommendation on Data Missing Not at Randomcited by 24
- 2025Early Stopping Against Label Noise Without Validation Datacited by 21
- 2026OmniGAIA: Towards Native Omni-Modal AI Agentscited by 21
- 2025User Feedback in Human-LLM Dialogues: A Lens to Understand Users But Noisy as a Learning Signalcited by 19
- 2023BadLabel: A Robust Perspective on Evaluating and Enhancing Label-noise Learningcited by 16
- 2023Learnable Behavior Control: Breaking Atari Human World Records via Sample-Efficient Behavior Selectioncited by 14
- 2022Entire Space Counterfactual Learning for Reliable Content Recommendationscited by 13
- 2022Instance-dependent Label Distribution Estimation for Learning with Label Noisecited by 9
- 2024Debiased Recommendation with Noisy Feedbackcited by 8
- 2024Proximity Matters: Local Proximity Enhanced Balancing for Treatment Effect Estimationcited by 7
- 2025SelectMix: Enhancing Label Noise Robustness through Targeted Sample Mixingcited by 5
- 2026TourPlanner: A Competitive Consensus Framework with Constraint-Gated Reinforcement Learning for Travel Planningcited by 3
Built on by
Citing papers per year: 2026: 6.
6 papers.
- 2026Uncertainty-Aware Reward Modeling for Stable RLHFcited by 4
- 2026ImplicitRM: Unbiased Reward Modeling from Implicit Preference Data for LLM alignmentcited by 2
- 2026Deep Time-Series Forecasting in 10 Years: A Surveycited by 0
- 2026Rubrics on Trial: Evolving Rubrics from a Single Query via Synthetic Pairwise Evidencecited by 0
- 2026What do Reward Models Memorize?cited by 0
- 2026S3: Spectral Null-Space Swap Makes Reasoning Models Efficientcited by 0
- 2026S3: Spectral Null-Space Swap Makes Reasoning Models Efficient29 Sep 2026
- 2026What do Reward Models Memorize?27 Jul 2026
- 2026Rubrics on Trial: Evolving Rubrics from a Single Query via Synthetic Pairwise Evidence16 Jul 2026
- 2026Uncertainty-Aware Reward Modeling for Stable RLHF18 Jun 2026
- 2026ImplicitRM: Unbiased Reward Modeling from Implicit Preference Data for LLM alignment24 Mar 2026
- 2026Deep Time-Series Forecasting in 10 Years: A Survey20 Mar 2026
Similar, not cited
Close in wording, no citation either way
- 2025Beyond Reward Hacking: Causal Rewards for Large Language Model Alignment
- 2026Reinforcement Learning from Human Feedback: A Statistical Perspective
- 2024Secrets of RLHF in Large Language Models Part II: Reward Modeling
- 2024How to Evaluate Reward Models for RLHF
- 2025Mitigating Length Bias in RLHF through a Causal Lens
- 2026Factored Causal Representation Learning for Robust Reward Modeling in RLHF
- 2024A Theoretical Framework for Partially Observed Reward-States in RLHF
- 2025Debiasing Reward Models by Representation Learning with Guarantees