Mitigating Length Bias in RLHF through a Causal Lens
On the map: LLM alignment / RLHF / preference optimisation → Reward modeling and RLHF
Abstract (arXiv)
Reinforcement learning from human feedback (RLHF) is widely used to align large language models (LLMs) with human preferences. However, RLHF-trained reward models often exhibit length bias -- a systematic tendency to favor longer responses by conflating verbosity with quality. We propose a causal framework for analyzing and mitigating length bias in RLHF reward modeling. Central to our approach is a counterfactual data augmentation method that generates response pairs designed to isolate content quality from verbosity. These counterfactual examples are then used to train the reward model, enabling it to assess responses based on content quality independently of verbosity. Specifically, we construct (1) length-divergent pairs with similar content and (2) content-divergent pairs of similar length. Empirical evaluations show that our method reduces length bias in reward assignment and leads to more concise, content-focused outputs from the policy model. These findings demonstrate that the proposed approach effectively reduces length bias and improves the robustness and content sensitivity of reward modeling in RLHF pipelines.
Key facts. Cited by 4 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 26 of its 43 references that are in the map. First posted on arXiv 16 Nov 2025.
Builds on
26 of 43 references.
Closest ancestors · 14
Same direction, newest first
- 2025Disentangling Length Bias In Preference Learning Via Response-Conditioned Modeling2 Feb
- 2025Beyond Reward Hacking: Causal Rewards for Large Language Model Alignment16 Jan
- 2024Post-hoc Reward Calibration: A Case Study on Length Bias25 Sep
- 2024RRM: Robust Reward Model Training Mitigates Reward Hacking20 Sep
- 2024RLHF Workflow: From Reward Modeling to Online RLHF13 May
- 2024ODIN: Disentangled Reward Mitigates Hacking in RLHF11 Feb
- 2023Contrastive Preference Learning: Learning from Human Feedback without RL20 Oct
- 2023Loose lips sink ships: Mitigating Length Bias in Reinforcement Learning from Human Feedback8 Oct
- 2023A Long Way to Go: Investigating Length Correlations in RLHF5 Oct
- 2023DeepSpeed-Chat: Easy, Fast and Affordable RLHF Training of ChatGPT-like Models at All Scales2 Aug
- 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Model29 May
- 2023RAFT: Reward rAnked FineTuning for Generative Foundation Model Alignment13 Apr
2 more
Other references · 12
- 2020Language Models are Few-Shot Learnerscited by 32,199
- 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
- 2023Judging LLM-as-a-Judge with MT-Bench and Chatbot Arenacited by 8,393
- 2020Learning to summarize from human feedbackcited by 2,562
- 2024Chatbot Arena: An Open Platform for Evaluating LLMs by Human Preferencecited by 1,151
- 2020MPNet: Masked and Permuted Pre-training for Language Understandingcited by 973
- 2018Challenging Common Assumptions in the Unsupervised Learning of Disentangled Representationscited by 929
- 2019Learning the Difference that Makes a Difference with Counterfactually-Augmented Datacited by 428
- 2022Predictability and Surprise in Large Generative Modelscited by 209
- 2023LMFlow: An Extensible Toolkit for Finetuning and Inference of Large Foundation Modelscited by 58
Built on by
Citing papers per year: 2026: 4.
4 papers.
- 2026CROP: Task Relevance via Counterfactuals for Selective On-Policy Distillationcited by 5
- 2026Reward Bias Substitution: Single-Axis Bias Mitigations Redirect Optimization Pressurecited by 3
- 2026Rethinking the Comparison Unit in Sequence-Level Reinforcement Learning: An Equal-Length Paired Training Framework from Loss Correction to Sample Constructioncited by 0
- 2026RewardExplainer: Learning Reward Model Explanations from Counterfactual Preference Feedbackcited by 0
- 2026RewardExplainer: Learning Reward Model Explanations from Counterfactual Preference Feedback27 Sep 2026
- 2026CROP: Task Relevance via Counterfactuals for Selective On-Policy Distillation13 Aug 2026
- 2026Reward Bias Substitution: Single-Axis Bias Mitigations Redirect Optimization Pressure27 May 2026
- 2026Rethinking the Comparison Unit in Sequence-Level Reinforcement Learning: An Equal-Length Paired Training Framework from Loss Correction to Sample Construction19 Apr 2026
Similar, not cited
Close in wording, no citation either way
- 2025Bias Fitting to Mitigate Length Bias of Reward Model in RLHF
- 2024Disentangling Length from Quality in Direct Preference Optimization
- 2024Secrets of RLHF in Large Language Models Part II: Reward Modeling
- 2025A Unified Pairwise Framework for RLHF: Bridging Generative Reward Modeling and Policy Optimization
- 2024How to Evaluate Reward Models for RLHF
- 2026Factored Causal Representation Learning for Robust Reward Modeling in RLHF
- 2026CausalRM: Causal-Theoretic Reward Modeling for RLHF from Observational User Feedbacks
- 2024It Takes Two: On the Seamlessness between Reward and Policy Model in RLHF