This model is trained using
Pref-GRPO on the training dataset of
UniGenBench.
Recent advancements highlight the importance of GRPO-based reinforcement learning methods and benchmarking in enhancing text-to-image (T2I) generation. However, current methods using pointwise reward models (RM) for scoring generated images are susceptible to reward hacking. We reveal that this happens when minimal score differences between images are amplified after normalization, creating illusory advantages that drive the model to over-optimize for trivial gains, ultimately destabilizing the image generation process. To address this, we propose Pref-GRPO, a pairwise preference reward-based GRPO method that shifts the optimization objective from score maximization to preference fitting, ensuring more stable training. In Pref-GRPO, images are pairwise compared within each group using preference RM, and the win rate is used as the reward signal. Extensive experiments demonstrate that PREF-GRPO differentiates subtle image quality differences, providing more stable advantages and mitigating reward hacking. Additionally, existing T2I benchmarks are limited by coarse evaluation criteria, hindering comprehensive model assessment. To solve this, we introduce UniGenBench, a unified T2I benchmark comprising 600 prompts across 5 main themes and 20 subthemes. It evaluates semantic consistency through 10 primary and 27 sub-criteria, leveraging MLLM for benchmark construction and evaluation. Our benchmarks uncover the strengths and weaknesses of both open and closed-source T2I models and validate the effectiveness of Pref-GRPO.
1import torch
2from diffusers import FluxPipeline
3
4pipe = FluxPipeline.from_pretrained("CodeGoat24/FLUX.1-dev-PrefGRPO", torch_dtype=torch.bfloat16)
5pipe.enable_model_cpu_offload() #save some VRAM by offloading the model to CPU. Remove this if you have enough GPU power
6
7prompt = "A cat holding a sign that says hello world"
8image = pipe(
9 prompt,
10 height=1024,
11 width=1024,
12 guidance_scale=3.5,
13 num_inference_steps=30,
14 max_sequence_length=512,
15 generator=torch.Generator("cpu").manual_seed(3407)
16).images[0]
17image.save("flux-dev.png")
18
1@article{Pref-GRPO&UniGenBench,
2 title={Pref-GRPO: Pairwise Preference Reward-based GRPO for Stable Text-to-Image Reinforcement Learning},
3 author={Wang, Yibin and Li, Zhimin and Zang, Yuhang and Zhou, Yujie and Bu, Jiazi and Wang, Chunyu and Lu, Qinglin, and Jin, Cheng and Wang, Jiaqi},
4 journal={arXiv preprint arXiv:2508.20751},
5 year={2025}
6}
