ProxT2I: Efficient Reward-Guided Text-to-Image Generation via Proximal Diffusion

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Fang, Zhenghan, Zheng, Jian, Gao, Qiaozi, Gao, Xiaofeng, Sulam, Jeremias
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866917107327827968
author Fang, Zhenghan
Zheng, Jian
Gao, Qiaozi
Gao, Xiaofeng
Sulam, Jeremias
author_facet Fang, Zhenghan
Zheng, Jian
Gao, Qiaozi
Gao, Xiaofeng
Sulam, Jeremias
contents Diffusion models have emerged as a dominant paradigm for generative modeling across a wide range of domains, including prompt-conditional generation. The vast majority of samplers, however, rely on forward discretization of the reverse diffusion process and use score functions that are learned from data. Such forward and explicit discretizations can be slow and unstable, requiring a large number of sampling steps to produce good-quality samples. In this work we develop a text-to-image (T2I) diffusion model based on backward discretizations, dubbed ProxT2I, relying on learned and conditional proximal operators instead of score functions. We further leverage recent advances in reinforcement learning and policy optimization to optimize our samplers for task-specific rewards. Additionally, we develop a new large-scale and open-source dataset comprising 15 million high-quality human images with fine-grained captions, called LAION-Face-T2I-15M, for training and evaluation. Our approach consistently enhances sampling efficiency and human-preference alignment compared to score-based baselines, and achieves results on par with existing state-of-the-art and open-source text-to-image models while requiring lower compute and smaller model size, offering a lightweight yet performant solution for human text-to-image generation.
format Preprint
id arxiv_https___arxiv_org_abs_2511_18742
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle ProxT2I: Efficient Reward-Guided Text-to-Image Generation via Proximal Diffusion
Fang, Zhenghan
Zheng, Jian
Gao, Qiaozi
Gao, Xiaofeng
Sulam, Jeremias
Computer Vision and Pattern Recognition
Artificial Intelligence
Machine Learning
Diffusion models have emerged as a dominant paradigm for generative modeling across a wide range of domains, including prompt-conditional generation. The vast majority of samplers, however, rely on forward discretization of the reverse diffusion process and use score functions that are learned from data. Such forward and explicit discretizations can be slow and unstable, requiring a large number of sampling steps to produce good-quality samples. In this work we develop a text-to-image (T2I) diffusion model based on backward discretizations, dubbed ProxT2I, relying on learned and conditional proximal operators instead of score functions. We further leverage recent advances in reinforcement learning and policy optimization to optimize our samplers for task-specific rewards. Additionally, we develop a new large-scale and open-source dataset comprising 15 million high-quality human images with fine-grained captions, called LAION-Face-T2I-15M, for training and evaluation. Our approach consistently enhances sampling efficiency and human-preference alignment compared to score-based baselines, and achieves results on par with existing state-of-the-art and open-source text-to-image models while requiring lower compute and smaller model size, offering a lightweight yet performant solution for human text-to-image generation.
title ProxT2I: Efficient Reward-Guided Text-to-Image Generation via Proximal Diffusion
topic Computer Vision and Pattern Recognition
Artificial Intelligence
Machine Learning
url https://arxiv.org/abs/2511.18742