atlas

Research map / Reinforcement learning / RL from feedback

REBEL: Reinforcement Learning via Regressing Relative Rewards

NeurIPS 2024 · Reinforcement learning › RL from feedback · approach: RL fine-tuning of generative models · 2 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 REBEL, a minimalist RL algorithm that fine-tunes generative models by iteratively regressing the relative reward between two completions onto the policy's log-ratio difference, with theory linking it to mirror descent/NPG.

Problem

PPO is complex and sensitive to implementation details (value networks, clipping, four models in memory) when fine-tuning large generative models, so a simpler scalable RL algorithm is sought.

Approach

Reduce policy optimization to iterated squared-loss regression: the log-probability ratio difference of the policy between two samples is regressed onto the reward difference, which cancels the partition function and implicitly parameterizes the reward via the policy; with on-policy y and base-distribution y', it recovers mirror descent/NPG-style updates without critics or clipping.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Replace PPO with REBELPythia 1.4B fine-tuning · TL;DRwinrate: 51.7 ($\pm1.42$) -> 55.1 ($\pm1.35$)Helped
Replace PPO with REBELPythia 2.8B fine-tuning · TL;DRwinrate: 67.4 ($\pm1.30$) -> 70.2 ($\pm1.32$)Helped
Replace PPO with REBELPythia 6.9B fine-tuning · TL;DRwinrate: 77.6$^\ddag$ -> 78.1 ($\pm1.74$)Helped
Replace APA with REBELOpenChat-3.5 fine-tuning · AlpacaEval 2.0LC win rate: 14.7 -> 17.3Helped
Replace APA with REBELOpenChat-3.5 fine-tuning · AlpacaEval 2.0win rate: 14.2 -> 12.8Hurt
Replace APA with REBELOpenChat-3.5 fine-tuning · MT-Benchscore: 7.43 -> 8.06Helped
Fine-tune with REBELOpenChat-3.5 · GSM8K (5-shot)accuracy: 68.5 -> 68.8Helped
Fine-tune with REBELOpenChat-3.5 · Arc (25-shot)accuracy: 64.9 -> 64.3Hurt
Use one REBEL iteration instead of DPO on online best-of-5/worst-of-5 data (ablation)Meta-Llama-3-8B-Instruct · UltraFeedbackLC win rate: 44.9 -> 48.3Helped
Continue REBEL from iteration 1 to iteration 2 (ablation)Meta-Llama-3-8B-Instruct · UltraFeedbackLC win rate: 48.3 -> 50.0Helped
Continue REBEL from iteration 2 to iteration 3 (ablation)Meta-Llama-3-8B-Instruct · UltraFeedbackLC win rate: 50.0 -> 49.7Hurt
Replace PPO's clipped objective with relative-reward regressionLatent consistency model · Text-guided image generationLAION aesthetic rewardMixed

Surprising findings

Limitations

Experimental setup

Datasets: TL;DR summarization dataset Nectar UltraFeedback LAION aesthetic predictor data Open LLM Leaderboard tasks

Benchmarks: TL;DR summarization winrate vs human reference AlpacaEval 2.0 MT-Bench Open LLM Leaderboard (MMLU, GSM8K, Arc, Winogrande, TruthfulQA, HellaSwag) Arena Hard LAION aesthetic score

Metrics: winrate reward model score KL divergence length-controlled win rate MT-Bench score inter-quartile mean reward

Baselines: PPO DPO Iterative DPO REINFORCE RLOO APA

Builds on: PPO Natural Policy Gradient DPO RLOO Mirror Descent

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