Fine-Tuning Language Models from Human Preferences

Code

• github.com/openai/lm-human-preferences
  OfficialIn papertf★ 0
• github.com/volcengine/verl
  pytorch★ 20,108
• github.com/lvwerra/trl
  pytorch★ 17,754
• github.com/huggingface/trl
  pytorch★ 17,741
• github.com/naver/gdc
  pytorch★ 118
• github.com/jackaduma/Alpaca-LoRA-RLHF-PyTorch
  pytorch★ 61
• github.com/google/t5patches
  jax★ 12
• github.com/padlex/trl
  pytorch★ 0
• github.com/carperai/trlx
  pytorch★ 0

Abstract

Reward learning enables the application of reinforcement learning (RL) to tasks where reward is defined by human judgment, building a model of reward by asking humans questions. Most work on reward learning has used simulated environments, but complex information about values is often expressed in natural language, and we believe reward learning for language is a key to making RL practical and safe for real-world tasks. In this paper, we build on advances in generative pretraining of language models to apply reward learning to four natural language tasks: continuing text with positive sentiment or physically descriptive language, and summarization tasks on the TL;DR and CNN/Daily Mail datasets. For stylistic continuation we achieve good results with only 5,000 comparisons evaluated by humans. For summarization, models trained with 60,000 comparisons copy whole sentences from the input but skip irrelevant preamble; this leads to reasonable ROUGE scores and very good performance according to our human labelers, but may be exploiting the fact that labelers rely on simple heuristics.

Tasks

Reproductions