RePrompt: Reasoning-Augmented Reprompting for Text-to-Image Generation
  via Reinforcement Learning
Authors: name: Mingrui Wu; name: Lu Wang; name: Pu Zhao; name: Fangkai Yang; name: Jianjin Zhang; name: Jianfeng Liu; name: Yuefeng Zhan; name: Weihao Han; name: Hao Sun; name: Jiayi Ji; name: Xiaoshuai Sun; name: Qingwei Lin; name: Weiwei Deng; name: Dongmei Zhang; name: Feng Sun; name: Qi Zhang; name: Rongrong Ji
Published: 2025-05-23 06:44:26

Abstract: Despite recent progress in text-to-image (T2I) generation, existing models
often struggle to faithfully capture user intentions from short and
under-specified prompts. While prior work has attempted to enhance prompts
using large language models (LLMs), these methods frequently generate stylistic
or unrealistic content due to insufficient grounding in visual semantics and
real-world composition. Inspired by recent advances in reasoning for language
model, we propose RePrompt, a novel reprompting framework that introduces
explicit reasoning into the prompt enhancement process via reinforcement
learning. Instead of relying on handcrafted rules or stylistic rewrites, our
method trains a language model to generate structured, self-reflective prompts
by optimizing for image-level outcomes. The tailored reward models assesse the
generated images in terms of human preference, semantic alignment, and visual
composition, providing indirect supervision to refine prompt generation. Our
approach enables end-to-end training without human-annotated data. Experiments
on GenEval and T2I-Compbench show that RePrompt significantly boosts spatial
layout fidelity and compositional generalization across diverse T2I backbones,
establishing new state-of-the-art results.