atlas

Research map / Reinforcement learning / RL from feedback

Degeneration-free Policy Optimization: RL Fine-Tuning for Language Models without Degeneration

ICML 2024 · Reinforcement learning › RL from feedback · approach: RL for LLM reasoning

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 Degeneration-free Policy Optimization (DfPO), an RL fine-tuning method for LMs that uses KL-masking and truncated advantages to improve task scores without KL-penalty hyperparameter search or text degeneration.

Problem

RL fine-tuning of language models with PPO and a KL-divergence penalty is highly sensitive to the penalty coefficient and often leads to text degeneration, where generated text becomes unnatural while task reward increases.

Approach

Reformulates policy gradient as separate likelihood maximization and minimization. KL-masking keeps only tokens whose likelihood can be moved in the advantage direction without leaving the reference policy (sign of log-ratio matches sign of advantage), and truncated advantages route each token to max or min, replacing the KL reward penalty with a hyperparameter-free rule.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use DfPO+Sup instead of PPO+SupT5 on CommonGen · CommonGenMETEOR: 0.329 ± 0.003 -> 0.336 ± 0.001Helped
Use DfPO+Sup instead of PPO+SupT5 on CommonGen · CommonGenBertScore: 0.930 ± 0.001 -> 0.929 ± 0.001Hurt
Use DfPO+Sup instead of PPO+SupT5 on CommonGen · CommonGenSpice: 0.447 ± 0.004 -> 0.445 ± 0.002Hurt

Surprising findings

Limitations

Experimental setup

Datasets: IMDB RealToxicityPrompts CommonGen

Benchmarks: IMDB text continuation RealToxicityPrompts text detoxification CommonGen

Metrics: sentiment score perplexity non-toxicity score METEOR ROUGE BLEU BERTScore CIDEr SPICE diversity metrics

Baselines: PPO NLPO DPO Zero Shot Supervised

Builds on: PPO NLPO DPO RL4LMs GAE

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