TensorX
返回文献探索

Paper · arXiv 2607.11886

Read It Back: Pretrained MLLMs Are Zero-Shot Reward Models for Text-to-Image Generation

Runhui Huang, Qihui Zhang, Zhe Liu, Yu Gao, Jie Wu, Hengshuang Zhao

87 upvotesJuly 13, 2026arXiv 预印本
AI 摘要

SpectraReward leverages pretrained multimodal language models as training-free reward models for image-generation reinforcement learning by measuring prompt recovery likelihood, with a self-improving variant that uses the policy's own understanding branch to close the loop.

multimodal large language modelsreward modelsreinforcement learningdiffusion modelsimage-conditioned prompt log-likelihoodSelf-SpectraRewardreward-policy alignment

Abstract

In this paper, we propose SpectraReward, a training-free reward function that turns pretrained MLLMs into off-the-shelf reward models for image-generation reinforcement learning. Instead of asking the MLLM to judge a generated image or answer decomposed verification questions, SpectraReward measures how well the original prompt can be recovered from the generated image through a single image-conditioned, teacher-forced forward pass. We use the average image-conditioned prompt log-likelihood as the reward, directly reusing the MLLM's pretrained image-text alignment ability without preference labels, reward-model fine-tuning. We further introduce Self-SpectraReward, a special case for unified multimodal models where the policy's own understanding branch serves as the reward model for its generation branch, forming a closed-loop self-improving framework without external reward models or external knowledge. Extensive experiments validate SpectraReward through a broad image-generation RL study covering two diffusion models, three RL algorithms, nine reward MLLM backbones from four MLLM families spanning 4B to 235B parameters, and five out-of-distribution text-to-image benchmarks. Results show that both SpectraReward and Self-SpectraReward significantly and consistently improve generation performance and outperform prior MLLM-derived reward training methods. Further analysis reveals that larger reward MLLMs are not always better, while Self-SpectraReward can match or surpass much larger external reward models, suggesting that reward-policy alignment is a key factor for effective image-generation RL. Project Page: https://huangrh99.github.io/SpectraReward/

北京市昌平区探索星信息技术及软件开发工作室

京ICP备2026059466号
Read It Back: Pretrained MLLMs Are Zero-Shot Reward Models for Text-to-Image Generation | TensorX