atlas

Research map / Image generation / Text-to-image diffusion

ReNO: Enhancing One-step Text-to-Image Models through Reward-based Noise Optimization

NeurIPS 2024 · Image generation › Text-to-image diffusion · approach: T2I evaluation and controllable adapters · 11 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 ReNO, which improves one-step text-to-image models at inference by gradient-ascent optimization of the initial noise against an ensemble of human-preference reward models, without updating model weights.

Problem

Text-to-image models struggle with complex compositional prompts, and reward fine-tuning suffers from reward hacking and poor generalization to unseen prompts; per-prompt fine-tuning and multi-step noise optimization are too expensive.

Approach

Backpropagate reward-model gradients through a single-step distilled T2I generator to optimize the initial Gaussian noise at inference, using a weighted combination of HPSv2, PickScore, ImageReward, and CLIPScore plus a chi-distribution noise regularizer, with few iterations to limit reward hacking.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Optimize initial noise with ReNOSD-Turbo · GenEvalMean: 0.49 to 0.62Helped
Optimize initial noise with ReNOFLUX-schnell · GenEvalMean: 0.64 to 0.72Helped
Optimize initial noise with ReNOSD-Turbo · T2I-CompBench ColorColor: 55% to 78%Helped
Use all four reward models rather than CLIPScore alone (ablation)SD-Turbo + ReNO · T2I-CompBench attribute binding, first 50 promptsShape: 66.9 to 77.4Helped
Use all four reward models rather than CLIPScore alone (ablation)SD-Turbo + ReNO · T2I-CompBench attribute binding, first 50 promptsTexture: 79.6 to 82.8Helped
Optimize initial noise using CLIPScore alone (ablation)SD-Turbo · LAION aesthetic scoreAesthetic: 5.647 -> 5.475Hurt
Optimize initial noise using ImageReward alone (ablation)SD-Turbo · T2I-CompBench attribute bindingColor: 0.5513 -> 0.7720Helped
Optimize initial noise using HPSv2 alone (ablation)SD-Turbo · LAION aesthetic scoreAesthetic: 5.647 -> 5.752Helped
Optimize initial noise with ReNOSD-Turbo · Parti-Prompts, 50 seedsLPIPS similarity: 0.382 -> 0.246Helped
Optimize initial noise with ReNOSDXL-Turbo · Parti-Prompts, 50 seedsLPIPS similarity: 0.391 -> 0.291Helped
Use one-step SD-Turbo with ReNO instead of 50-step SD2.1 + DOODLNoise optimization · T2I-CompBench attribute binding, first 50 promptsColor: 38.5 (+5.1) -> 70.1 (+9.7)Helped

Surprising findings

Limitations

Experimental setup

Datasets: T2I-CompBench GenEval Parti-Prompts

Benchmarks: T2I-CompBench GenEval Parti-Prompts

Metrics: BLIP-VQA score object detection score GenEval mean LAION aesthetic score LPIPS DINO similarity user preference win rate CLIPScore

Baselines: SD-Turbo SDXL-Turbo PixArt-alpha DMD HyperSDXL SDXL SD3 DALL-E 3 DOODL DPO-SDXL FLUX-schnell

Builds on: DOODL Adversarial Diffusion Distillation Diffusion Matching Distillation ImageReward HPSv2

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