Read It Back: Pretrained MLLMs Are Zero-Shot Reward Models for Text-to-Image Generation· 预训练 MLLM 作为文本到图像生成的零样本奖励模型
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/
预训练模型直接用于图像生成的奖励函数,无需额外训练。
- 核心方法
- SpectraReward 通过衡量生成图像条件下原始文本的平均对数似然性作为奖励,无需偏好标签或微调。Self-SpectraReward 则利用模型的自身理解分支为生成分支提供奖励,形成闭环自优化框架。
- 适合谁读
- 研究者、工程师
- 要解决的问题
- 如何利用预训练的多模态语言模型(MLLM)作为文本到图像生成的零样本奖励模型。
- 关键实验
- 实验覆盖了两种扩散模型、三种强化学习算法、九种不同参数量的 MLLM 主干,以及五个超出分布范围的文本到图像基准。
- 主要贡献
- 提出了无需训练的 SpectraReward 和 Self-SpectraReward,显著提升图像生成性能,并超越现有基于 MLLM 的奖励训练方法。
- 意义与局限
- 证明了在图像生成的强化学习中,奖励模型与策略的对齐比单纯使用大型模型更为重要,为未来研究提供了新的方向。局限性在于未探讨不同任务间的泛化能力。