atlas

Research map / Reinforcement learning / RL from feedback

Leftover Lunch: Advantage-based Offline Reinforcement Learning for Language Models

ICLR 2024 · Reinforcement learning › RL from feedback · approach: Theory of RLHF · 0 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes A-LoL, a family of advantage-weighted offline policy gradient methods that treat whole LM outputs as single actions and train only on positive-advantage pre-existing data, compared against PPO, DPO, PRO, wBC and GOLD across four generation tasks.

Problem

RLHF with PPO is unstable and data-hungry, requiring fresh LM-generated data each step; the paper asks whether reward-based learning can be done purely offline on pre-existing language data.

Approach

Treat the full output sequence as a single action and derive an offline policy gradient weighted by the reference policy's advantage (reward minus a learned value estimate from frozen reference LM features) and an importance weight, training only on positive-advantage examples.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use advantage-based training with full-sequence importance weights instead of reward-based LoLLLaMA-7B on HHA · HHA testAvg. Reward: 55.1_{\pm0.6 } -> 57.3_{\pm0.3 }Helped
Set the importance weight to 1 (ablation)A-LoL on LLaMA-7B · HHA testAvg. Reward: 57.3_{\pm0.3 } -> 56.4_{\pm0.6 }Hurt
Replace full-sequence importance weights with per-token importance weights (ablation)A-LoL on LLaMA-7B · HHA testDistinct-1/2/3: $.094$/$ .403$/$ .657$ -> $.105$/$\mathbf{.464}$/$\mathbf{.727}$Helped
Replace importance weighting with a weighted KL penalty (ablation)A-LoL on LLaMA-7B · HHA validationrewardHurt
Use per-token importance weights instead of full-sequence importance weights (ablation)A-LoL on DialoGPT-medium · Reddit upvoted replies testReward: $3.15$ -> $\mathbf{3.28}$Helped
Use advantage-based training rather than reward-based LoLDialoGPT-medium on Reddit upvoted replies · Reddit upvoted replies testReward: $2.95$ -> $3.15$Helped
Use per-token importance weights instead of full-sequence importance weights (ablation)A-LoL on DialoGPT-medium · Reddit downvoted replies testReward: $3.14$ -> $\mathbf{3.18}$Helped
Use advantage-based A-LoL training on downvoted data rather than the reference policyDialoGPT-medium · Reddit downvoted replies testReward: $2.87$ -> $3.14$Helped
Use A-LoL sequence instead of Reward LoLCOMET DIS TIL · Commonsense Reasoning ATOMIC20 testCOMET Critic: $89.4$ -> $\mathbf{93.0}$Helped
Use advantage-based A-LoL instead of reward-based LoLDialoGPT-medium on WoW training · Knowledge-grounded dialog FaithDial testReward: $2.72$ -> $2.83$Helped
Use priority sampling instead of random sampling (ablation)A-LoL · HHA validationrewardHelped
Remove negative-advantage instances instead of sampling the entire dataset (ablation)A-LoL · HHA validationrewardHelped

Surprising findings

Limitations

Experimental setup

Datasets: Helpful and Harmless Assistant (HH-RLHF) Reddit comment-response pairs ATOMIC10x Wizard of Wikipedia FaithDial

Benchmarks: HH-RLHF test set Reddit response test set ATOMIC 2020 test set FaithDial test set

Metrics: average reward from reward model Distinct-1/2/3 average length human and GPT-4 win rate COMET critic score coverage density

Baselines: PPO DPO DPO (ref. free) PRO wBC Reward GOLD R-LoL NLL

Builds on: PPO DPO PRO GOLD weighted behavior cloning

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy