改进文本生成图像的奖励机制,让模型更准地学习早期去噪步骤。
Multi-GRPO: Multi-Group Advantage Estimation for Text-to-Image Generation with Tree-Based Trajectories and Multiple Rewards
- 用树状轨迹分组,精准估算早期去噪步骤的贡献。
- 按奖励独立计算优势,解决多目标冲突导致的训练不稳。
- 适用于需要精确文字渲染的图像生成任务,如带颜色约束的图文合成。
近期基于组相对策略优化(GRPO)的方法在对齐文本到图像(T2I)模型方面展现出潜力,但现有方法存在两大缺陷:(1) 共享信用分配——基于组归一化稀疏终端奖励计算的轨迹级优势在各时间步均匀分配,无法准确估计具有广阔探索空间的早期去噪步骤;(2) 奖励混杂——预设权重融合多目标奖励(如文本准确性、视觉质量、文本颜色),因量纲与方差不匹配导致梯度不稳定和更新冲突。为此,本文提出多组优势估计框架 Multi-GRPO,采用两种正交分组机制:为改善信用分配,引入受蒙特卡洛树搜索启发的树状轨迹,在选定的早期去噪步骤进行分支,形成时间分组,通过后代节点精准估计早期步骤的优势,同时共享前缀以降低计算开销;为实现多目标优化,引入基于奖励的分组,先独立计算每项奖励函数的优势再聚合,解耦冲突信号。为便于多目标对齐评估,构建了含显式颜色约束的视觉文字渲染数据集 OCR-Color-10。在单奖励的 PickScore-25k 和多目标的 OCR-Color-10 基准上,Multi-GRPO 均表现出更优的稳定性与对齐性能,有效平衡冲突目标。代码将公开于 https://github.com/fikry102/Multi-GRPO。
原文摘要 · Abstract (English)
Recently, Group Relative Policy Optimization (GRPO) has shown promising potential for aligning text-to-image (T2I) models, yet existing GRPO-based methods suffer from two critical limitations. (1) \textit{Shared credit assignment}: trajectory-level advantages derived from group-normalized sparse terminal rewards are uniformly applied across timesteps, failing to accurately estimate the potential of early denoising steps with vast exploration spaces. (2) \textit{Reward-mixing}: predefined weights for combining multi-objective rewards (e.g., text accuracy, visual quality, text color)--which have mismatched scales and variances--lead to unstable gradients and conflicting updates. To address these issues, we propose \textbf{Multi-GRPO}, a multi-group advantage estimation framework with two orthogonal grouping mechanisms. For better credit assignment, we introduce tree-based trajectories inspired by Monte Carlo Tree Search: branching trajectories at selected early denoising steps naturally forms \emph{temporal groups}, enabling accurate advantage estimation for early steps via descendant leaves while amortizing computation through shared prefixes. For multi-objective optimization, we introduce \emph{reward-based grouping} to compute advantages for each reward function \textit{independently} before aggregation, disentangling conflicting signals. To facilitate evaluation of multiple objective alignment, we curate \textit{OCR-Color-10}, a visual text rendering dataset with explicit color constraints. Across the single-reward \textit{PickScore-25k} and multi-objective \textit{OCR-Color-10} benchmarks, Multi-GRPO achieves superior stability and alignment performance, effectively balancing conflicting objectives. Code will be publicly available at \href{https://github.com/fikry102/Multi-GRPO}{https://github.com/fikry102/Multi-GRPO}.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。