改进混合生成模型的强化学习训练,提升图像质量与稳定性。
MAR-GRPO: Stabilized GRPO for AR-diffusion Hybrid Image Generation
- 用多轨迹平均降低扩散模型带来的梯度噪声。
- 仅对不确定性高的部分进行优化,避免过度平滑。
- 通过一致性筛选确保生成内容与文本描述对齐,适合高质量图像生成任务。
强化学习在自回归(AR)和扩散模型中已取得成功,但将其扩展到混合AR-扩散框架仍面临挑战,主要源于推理过程交错及噪声化的对数概率估计。本文研究掩码自回归模型(MAR),发现扩散头在训练动态中起关键作用,常引入噪声梯度导致训练不稳定和性能过早饱和。为此,提出一种稳定化的强化学习框架。引入多轨迹期望(MTE),通过平均多个扩散轨迹来估计优化方向,从而减少扩散引起的梯度噪声。为避免过度平滑,从多轨迹中估计词元级不确定性,并仅对顶部k%不确定的词元应用多轨迹优化。此外,提出一致性感知的词元选择策略,过滤掉与最终生成内容不一致的AR词元。在多个基准测试上的实验表明,该方法在视觉质量、训练稳定性及空间结构理解方面均优于基线GRPO和预强化学习模型。代码已公开:https://github.com/AMAP-ML/mar-grpo。
原文摘要 · Abstract (English)
Reinforcement learning (RL) has been successfully applied to autoregressive (AR) and diffusion models. However, extending RL to hybrid AR-diffusion frameworks remains challenging due to interleaved inference and noisy log-probability estimation. In this work, we study masked autoregressive models (MAR) and show that the diffusion head plays a critical role in training dynamics, often introducing noisy gradients that lead to instability and early performance saturation. To address this issue, we propose a stabilized RL framework for MAR. We introduce multi-trajectory expectation (MTE), which estimates the optimization direction by averaging over multiple diffusion trajectories, thereby reducing diffusion-induced gradient noise. To avoid over-smoothing, we further estimate token-wise uncertainty from multiple trajectories and apply multi-trajectory optimization only to the top-k% uncertain tokens. In addition, we introduce a consistency-aware token selection strategy that filters out AR tokens that are less aligned with the final generated content. Extensive experiments across multiple benchmarks demonstrate that our method consistently improves visual quality, training stability, and spatial structure understanding over baseline GRPO and pre-RL models. Code is available at: https://github.com/AMAP-ML/mar-grpo.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。