mlsanityData to 7 Oct 2026

CausalRM: Causal-Theoretic Reward Modeling for RLHF from Observational User Feedbacks

arXiv 2603.18736 ↗First posted 19 Mar 2026cs.LG cs.AI cs.CLcited by 6

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

  1. 2025On Symmetric Losses for Robust Policy Optimization with Noisy Preferences30 May
  2. 2024Interpretable Preferences via Multi-Objective Reward Modeling and Mixture-of-Experts18 Jun
  3. 2024RLHF Workflow: From Reward Modeling to Online RLHF13 May
  4. 2024InfoRM: Mitigating Reward Hacking in RLHF via Information-Theoretic Reward Modeling14 Feb
  5. 2024ODIN: Disentangled Reward Mitigates Hacking in RLHF11 Feb

Other references · 35

  1. 2023GPT-4 Technical Reportcited by 18,132
  2. 2022Training language models to follow instructions with human feedbackcited by 15,712
  3. 2017Proximal Policy Optimization Algorithmscited by 14,441
  4. 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 12,865
  5. 2025DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learningcited by 6,146
  6. 2025Gemini 2.5: Pushing the Frontier with Advanced Reasoning, Multimodality, Long Context, and Next Generation Agentic Capabilitiescited by 3,655
  7. 2024HarmBench: A Standardized Evaluation Framework for Automated Red Teaming and Robust Refusalcited by 1,330
  8. 2018Co-teaching: Robust Training of Deep Neural Networks with Extremely Noisy Labelscited by 1,053
  9. 2025Group Sequence Policy Optimizationcited by 788
  10. 2016Making Deep Neural Networks Robust to Label Noise: a Loss Correction Approachcited by 663
  11. 2023Bias and Fairness in Large Language Models: A Surveycited by 626
  12. 2024Generative Verifiers: Reward Modeling as Next-Token Predictioncited by 465
23 more
  1. 2024WildGuard: Open One-Stop Moderation Tools for Safety Risks, Jailbreaks, and Refusals of LLMscited by 460
  2. 2023UltraFeedback: Boosting Language Models with Scaled AI Feedbackcited by 394
  3. 2024PKU-SafeRLHF: Towards Multi-Level Safety Alignment for LLMs with Human Preferencecited by 252
  4. 2023HelpSteer: Multi-attribute Helpfulness Dataset for SteerLMcited by 140
  5. 2024Provably Robust DPO: Aligning Language Models with Noisy Feedbackcited by 133
  6. 2025DeepAgent: A General Reasoning Agent with Scalable Toolsetscited by 69
  7. 2024Towards Robust Alignment of Language Models: Distributionally Robustifying Direct Preference Optimizationcited by 52
  8. 2022ESCM2: Entire Space Counterfactual Multi-Task Model for Post-Click Conversion Rate Estimationcited by 47
  9. 2019Large-scale Causal Approaches to Debiasing Post-click Conversion Rate Estimation with Multi-task Learningcited by 39
  10. 2023Optimal Transport for Treatment Effect Estimationcited by 32
  11. 2020Information Theoretic Counterfactual Learning from Missing-Not-At-Random Feedbackcited by 30
  12. 2022StableDR: Stabilized Doubly Robust Learning for Recommendation on Data Missing Not at Randomcited by 24
  13. 2025Early Stopping Against Label Noise Without Validation Datacited by 21
  14. 2026OmniGAIA: Towards Native Omni-Modal AI Agentscited by 21
  15. 2025User Feedback in Human-LLM Dialogues: A Lens to Understand Users But Noisy as a Learning Signalcited by 19
  16. 2023BadLabel: A Robust Perspective on Evaluating and Enhancing Label-noise Learningcited by 16
  17. 2023Learnable Behavior Control: Breaking Atari Human World Records via Sample-Efficient Behavior Selectioncited by 14
  18. 2022Entire Space Counterfactual Learning for Reliable Content Recommendationscited by 13
  19. 2022Instance-dependent Label Distribution Estimation for Learning with Label Noisecited by 9
  20. 2024Debiased Recommendation with Noisy Feedbackcited by 8
  21. 2024Proximity Matters: Local Proximity Enhanced Balancing for Treatment Effect Estimationcited by 7
  22. 2025SelectMix: Enhancing Label Noise Robustness through Targeted Sample Mixingcited by 5
  23. 2026TourPlanner: A Competitive Consensus Framework with Constraint-Gated Reinforcement Learning for Travel Planningcited by 3

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 6.

6 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2025Beyond Reward Hacking: Causal Rewards for Large Language Model Alignment
  2. 2026Reinforcement Learning from Human Feedback: A Statistical Perspective
  3. 2024Secrets of RLHF in Large Language Models Part II: Reward Modeling
  4. 2024How to Evaluate Reward Models for RLHF
  5. 2025Mitigating Length Bias in RLHF through a Causal Lens
  6. 2026Factored Causal Representation Learning for Robust Reward Modeling in RLHF
  7. 2024A Theoretical Framework for Partially Observed Reward-States in RLHF
  8. 2025Debiasing Reward Models by Representation Learning with Guarantees