arXiv:2603.23086cs.LGcs.CV2026-03中稿 · ECCV

用新奖励机制让自回归图像模型生成更多样高质量图片。

Policy-based Tuning of Autoregressive Image Models with Instance- and Distribution-Level Rewards

  • 将自回归图像生成建模为马尔可夫决策过程,用分组相对策略优化训练。
  • 引入分布级留一法FID奖励,提升样本多样性,防止模式崩溃。
  • 适合想提升图像生成质量与多样性的研究者,尤其关注推理效率者。

自回归(AR)模型在图像生成中表现优异,但其标准最大似然训练无法直接优化样本质量与多样性。尽管强化学习(RL)已用于对齐扩散模型,但常导致输出多样性下降。类似地,现有针对AR模型的RL方法仅依赖实例级奖励,往往以牺牲分布覆盖为代价换取质量。为此,我们提出一种轻量级RL框架,将基于标记的AR合成视为马尔可夫决策过程,通过分组相对策略优化(GRPO)进行优化。核心贡献是提出一种新型分布级留一法FID(LOO-FID)奖励;通过利用特征矩的指数移动平均,显式鼓励样本多样性并防止策略更新中的模式崩溃。我们结合复合实例级奖励(CLIP与HPSv2)确保语义与感知保真度,并通过自适应熵正则化项稳定多目标学习。在LlamaGen与VQGAN架构上的大量实验表明,仅需数百次调优迭代即可在标准质量与多样性指标上实现显著提升。结果还显示,该模型可在无需分类器无关引导(Classifier-Free Guidance)的情况下生成具有竞争力的样本,从而规避其2倍推理开销。

原文摘要 · Abstract (English)

Autoregressive (AR) models are highly effective for image generation, yet their standard maximum-likelihood estimation training lacks direct optimization for sample quality and diversity. While reinforcement learning (RL) has been used to align diffusion models, these methods typically suffer from output diversity collapse. Similarly, concurrent RL methods for AR models rely strictly on instance-level rewards, often trading off distributional coverage for quality. To address these limitations, we propose a lightweight RL framework that casts token-based AR synthesis as a Markov Decision Process, optimized via Group Relative Policy Optimization (GRPO). Our core contribution is the introduction of a novel distribution-level Leave-One-Out FID (LOO-FID) reward; by leveraging an exponential moving average of feature moments, it explicitly encourages sample diversity and prevents mode collapse during policy updates. We integrate this with composite instance-level rewards (CLIP and HPSv2) for strict semantic and perceptual fidelity, and stabilize the multi-objective learning with an adaptive entropy regularization term. Extensive experiments on LlamaGen and VQGAN architectures demonstrate clear improvements across standard quality and diversity metrics within only a few hundred tuning iterations. The results also show that the model can be updated to produce competitive samples even without Classifier-Free Guidance, and bypass its 2x inference cost.

图像生成强化学习多样性优化自回归模型

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。