Loose lips sink ships: Mitigating Length Bias in Reinforcement Learning from Human Feedback

By Wei Shen, Rui Zheng, Wenyu Zhan, Jun Zhao, Shihan Dou, Tao Gui, Qi Zhang, Xuanjing Huang. 2023

90 citations recorded by the source corpus.

Reinforcement learning from human feedback serves as a crucial bridge, aligning large language models with human and societal values. This alignment requires a vast corpus of human feedback to learn a reward model, which is subsequently used to finetune language models. However, we have identified that the reward model often finds shortcuts to bypass its intended objectives, misleadingly assuming that humans prefer longer responses. The emergence of length bias often induces the model to favor longer outputs, yet it doesn't equate to an increase in helpful information within these outputs. In this paper, we propose an innovative solution, applying the Product-of-Experts (PoE) technique to separate reward modeling from the influence of sequence length. In our framework, the main expert concentrates on understanding human intents, while the biased expert targets the identification and capture of length bias. To further enhance the learning of bias, we introduce perturbations into the bias-focused expert, disrupting the flow of semantic information. Experimental results validate the effectiveness of our approach, indicating that language model performance is improved, irrespective of sequence length.

Subfields: Artificial Intelligence, Information Systems, Alignment & RLHF, Reinforcement Learning

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Percy Liang
  2. Tatsunori B. Hashimoto
  3. Bryan Catanzaro
  4. Stefano Ermon
  5. Tom Goldstein
  6. Jianfeng Gao
  7. Sergey Levine
  8. Mohit Bansal

In Information Systems

  1. Percy Liang
  2. Tianyi Zhou
  3. Tom Goldstein
  4. Tatsunori B. Hashimoto
  5. Bryan Catanzaro
  6. Stefano Ermon
  7. Yann Dubois
  8. Luke S. Zettlemoyer

In Alignment & RLHF

  1. Rafael Rafailov
  2. Percy Liang
  3. Tatsunori B. Hashimoto
  4. Chelsea Finn
  5. James Thorne
  6. Jiwoo Hong
  7. Noah Lee
  8. Ryan Park

In Reinforcement Learning

  1. Percy Liang
  2. Rafael Rafailov
  3. Jiuhai Chen
  4. Chen Zhu
  5. Tianyi Zhou
  6. Mohammad Shoeybi
  7. Tom Goldstein
  8. Lichang Chen

Citations in the corpus