在图像编辑中利用基于验证器的强化学习
Leveraging Verifier-Based Reinforcement Learning in Image Editing
提出基于思维链的推理奖励模型(RRM)作为图像编辑的通用奖励,并用GRPO微调编辑模型,为视觉生成中的RLHF提供可解释、细粒度的奖励设计思路。
原文依据 We introduce Edit-R1, a framework to build a chain-of-thought (CoT) verifier-based reasoning reward model (RRM) and leverage it into the downstream editing task.
摘要译文
由模型翻译 · 术语统一 · 请对照原文核对关键结论尽管基于人类反馈的强化学习(RLHF)已成为文本到图像生成的关键范式,但其在图像编辑中的应用仍鲜有探索。一个关键瓶颈是缺乏适用于所有编辑任务的鲁棒通用奖励模型。现有的编辑奖励模型通常只给出整体分数而不进行详细检查,忽略了不同的指令要求,导致奖励有偏。为此,我们提出了基于验证器的推理奖励模型(RRM),它将指令分解为可验证的原则,针对每个原则评估编辑后的图像,并聚合细粒度分数以减少幻觉并提供更可解释的标准。为此,我们认为关键在于从简单的评分器转向推理验证器。我们引入了Edit-R1,一个构建基于思维链(CoT)验证器的推理奖励模型(RRM)并将其用于下游编辑任务的框架。Edit-RRM将指令分解为不同的原则,针对每个原则评估编辑后的图像,并聚合这些检查以提供可解释的细粒度奖励。为了构建这样的RRM,我们首先应用监督微调(SFT)作为“冷启动”来生成CoT奖励轨迹。然后,我们引入了组对比偏好优化(GCPO),一种利用人类成对偏好数据来强化我们的逐点RRM的强化学习算法。在构建RRM之后,我们使用GRPO来训练编辑模型,利用这个不可微分但强大的奖励模型。大量实验表明,作为编辑专用的奖励模型,我们的Edit-RRM超越了Seed-1.5-VL和Seed-1.6-VL等强大的VLM,并且我们观察到清晰的缩放趋势,性能从3B到7B参数持续提升。此外,Edit-R1为FLUX.1-kontext等编辑模型带来了增益,突显了其在增强图像编辑方面的有效性。
英文原文摘要 1932 字符
While Reinforcement Learning from Human Feedback (RLHF) has become a pivotal paradigm for text-to-image generation, its application to image editing remains largely unexplored. A key bottleneck is the lack of a robust general reward model for all editing tasks. Existing edit reward models usually give overall scores without detailed checks, ignoring different instruction requirements and causing biased rewards. To address this, we propose the Verifier-Based Reasoning Reward Model (RRM), which breaks instructions into verifiable principles, evaluates the edited images against each principle, and aggregates fine-grained scores to reduce hallucinations and provide more interpretable criteria. To address this, we argue the key is to move from a simple scorer to a reasoning verifier. We introduce Edit-R1, a framework to build a chain-of-thought (CoT) verifier-based reasoning reward model (RRM) and leverage it into the downstream editing task. The Edit-RRM breaks instructions into distinct principles, evaluates the edited image against each, and aggregates these checks to provide an interpretable, fine-grained reward. To build such an RRM, we first apply supervised fine-tuning (SFT) as a "cold-start" to generate CoT reward trajectories. Then, we introduce Group Contrastive Preference Optimization (GCPO), a reinforcement learning algorithm that leverages human pairwise preference data to reinforce our pointwise RRM. After building the RRM, we use GRPO to train editing models with this non-differentiable yet powerful reward model. Extensive experiments demonstrate that our Edit-RRM surpasses powerful VLMs such as Seed-1.5-VL and Seed-1.6-VL as an editing-specific reward model, and we observe a clear scaling trend, with performance consistently improving from 3B to 7B parameters. Moreover, Edit-R1 delivers gains to editing models like FLUX.1-kontext, highlighting its effectiveness in enhancing image editing.