atlas

Research map / Reinforcement learning / RL from feedback

Using Human Feedback to Fine-tune Diffusion Models without Any Reward Model

CVPR 2024 · Reinforcement learning › RL from feedback · approach: Theory of RLHF · 49 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 D3PO, which adapts DPO to the multi-step denoising MDP of diffusion models so they can be fine-tuned directly from human preference comparisons without training a reward model.

Problem

RLHF for diffusion models requires training a reward model, which is costly in data, architecture, and tuning; directly applying DPO to diffusion models is infeasible due to GPU memory needed to store gradients across many latent denoising steps.

Approach

Reformulate the denoising process as a multi-step MDP, show via a Q-function/Bradley-Terry argument that the DPO objective implicitly learns the optimal reward, and apply the DPO loss to every state-action pair of sub-trajectories, using the final image's human preference for all denoising steps.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Fine-tune with human preference feedback using D3PO instead of no fine-tuningStable Diffusion v1.5 · Prompt-image alignment (300 prompts)CLIP score: 30.7 -> 31.9Helped
Fine-tune with human preference feedback using D3PO instead of no fine-tuningStable Diffusion v1.5 · Prompt-image alignment (300 prompts)BLIP score: 1.95 -> 2.06Helped
Fine-tune with human preference feedback using D3PO instead of no fine-tuningStable Diffusion v1.5 · Prompt-image alignment (300 prompts)ImageReward score: 0.04 -> 0.27Helped
Fine-tune with human preference feedback using D3PO instead of no fine-tuningStable Diffusion v1.5 · Prompt-image alignment (300 prompts)Human preference: 14.7% -> 55.7%Helped

Surprising findings

Limitations

Experimental setup

Datasets: LAION aesthetics predictor training data

Benchmarks: CLIP score BLIP score ImageReward score human preference evaluation on 300 prompts

Metrics: image size reward LAION aesthetic score normalcy rate of hand images deformity rate of anime characters safety rate CLIP score BLIP score ImageReward score human preference percentage

Baselines: DDPO DPOK Reward Weighted ReFL ImageReward-based fine-tuning

Builds on: DPO DDPO Reward Weighted regression LoRA

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