跳转至

RePlan: Reasoning-Guided Region Planning for Complex Instruction-Based Image Editing

会议: ECCV2026
论文: ECCV 原文
官方编号: 3410
引用数: 12(任务提供,2026-09-17;未独立联网核验)
领域: 视觉语言推理 / 图像生成
关键词: 指令图像编辑、区域规划、思维链、注意力区域注入、GRPO

一句话总结

RePlan 让视觉语言模型先推理出“改哪个区域、具体怎样改”,再以无需训练编辑器的注意力区域注入并行执行修改,在 IV-Edit 上将 Flux.2-Klein-9B 的 Overall 从 3.45 提升至 3.92,主要改善区域一致性。

研究背景与动机

指令图像编辑不仅要生成好看的结果,还要在多个相似对象中找对目标。例如“把桌上用过并放下的杯子换成盆栽”,要求模型结合杯中是否有水、物体状态与语言含义判断目标,而不是把所有杯子都替换。本文把复杂画面、复杂指令及二者相互作用带来的困难称为指令视觉复杂性(Instruction-Visual Complexity,IV-Complexity)。

InstructPix2Pix 一类方法直接学习指令到编辑结果的映射;Qwen-Image、Bagel 等方法进一步引入视觉语言能力,但理解正确并不保证生成器改对位置。将整句需求编码成全局语义条件,仍可能让同类对象接收到相同编辑意图。逐区域图像修补可以加强定位,却需要多轮执行,前一轮的误差和画质损失会传给后一轮。

本文因此关注理解模型与编辑模型之间的接口:不仅传递“生成什么”,还要显式传递“这一条提示属于哪个区域”,同时保留全图视觉交互以维持连续性。核心 idea:把推理结果表示为可检查的区域框与编辑提示对,再用注意力可见性规则把各条提示送到对应区域,而不是让扩散模型自行从全局文本中重新猜目标。

方法详解

整体框架

输入是一张图像和自然语言编辑指令,输出是完成修改的图像。区域对齐规划器先生成思维链、全局提示以及区域框与局部提示;注意力区域注入把这些提示分别编码,并按区域限制它们与图像 token 的交互;规则组注意力则让这种受限注意力高效运行。所有区域在同一次编辑过程中共同生成,不是把每个区域依次修补。

默认规划器是 Qwen2.5-VL 7B,训练时的编辑骨干为 Flux.1-Kontext-dev。GRPO 训练的是规划器,注意力注入本身不要求重新训练扩散编辑器;因此不能把整个 RePlan 描述为“完全免训练”。论文也把同一框架接入 Qwen-Image-Edit 和 Flux.2-Klein-9B。

%%{init: {'flowchart': {'rankSpacing': 24, 'nodeSpacing': 28, 'padding': 6, 'wrappingWidth': 400}}}%%
flowchart TD
        Input["图像与编辑指令"] --> Planner["区域对齐规划器"]
        Planner --> Injection["注意力区域注入"]
        Injection --> Acceleration["规则组注意力"]
        Acceleration --> Output["联合去噪后的编辑图像"]

这里的“单次”指一次联合编辑过程,不是只有一个扩散去噪步。框架仍使用预训练编辑器的生成过程,只是不再随区域数增加而重复整套编辑流程。

关键设计

1. 区域对齐规划器:先消除指代歧义,再输出可执行计划

规划器先结合图像理解指令中的隐含条件,再把抽象要求改写为具体视觉变化。它输出由 <think>、<global> 和 <region> 分隔的结构化文本:前者放推理过程,中间是全图提示,最后是 JSON 列表,每项包含 bbox_2d 与 hint。这些字段不仅便于解释,也能让后续模块直接提取区域坐标和编辑动作。

全局提示覆盖整张图,可以要求修改背景或保持其余区域不变;局部提示只描述对应目标的变化。局部提示也可以明确要求“不改这个对象”。这不是传统意义上降低某些词权重的扩散负面提示,而是把保持要求绑定到一个具体区域,以保护容易与目标混淆的邻近对象。

显式计划使失败位置更容易判断:如果框选错了,应检查规划器;如果框和提示正确但结果不对,则应检查执行器。用户还可手动修改框或提示。GRPO 进一步训练这种计划的可解析性和实际编辑效果,具体奖励见下方训练策略。

2. 注意力区域注入:隔离局部语言条件,保留全图视觉联系

编辑器基于 MMDiT,将文本、原图特征和带噪目标潜变量放入统一注意力序列。RePlan 不把所有提示先混成一个文本表示,而是分别编码全局提示与各区域提示,再拼接其 token;原图经 VAE 编码成 patch 网格,区域框映射到网格,形成局部图像 token 组,所有未落入局部组的 patch 构成背景组。

注意力可见性遵循五条规则:组内完全交互;不同文本提示组互不可见;图像与潜变量 token 之间保留全局交互;每个局部图像组只能接收本区域提示和全局提示;背景组只能接收全局提示。关键不是把画面切成彼此独立的小图,而是切断错误的语言条件连接,同时保留视觉信息流。

这一区别解释了它为何能够兼顾局部控制和整体风格:一个区域不会直接接收另一区域的改色提示,但各区域仍能通过图像与潜变量交互协调光照、边界和风格。框也不是“框外像素绝不变化”的硬约束,原文明确允许必要的影响延伸到框外。对重叠框和互相冲突的提示,当前正文没有给出完整的冲突消解细则。

3. 规则组注意力:把稠密遮罩换成即时可见性判断

直接构造 token 两两之间的稠密遮罩虽然容易实现,却会妨碍高效注意力内核并增加显存访问。这里的可见性只取决于少数规则组,通常少于 10 个,而不是每一对 token 都有独特规则,因此没有必要存储完整的两两关系矩阵。

实现中,每个 token 保存两个位集合:它属于哪些组,以及它允许关注哪些组。计算注意力时,查询 token 的允许集合与键 token 的所属集合做按位与,交集非空即可交互;FlexAttention 将这个判定融合进内核。这样保留了区域注入的语义约束,却不必物化稠密遮罩。

原文给出的额外存储复杂度从 \(O(n^2)\) 降为 \(O(n\lceil k/W\rceil)\) 个机器字,其中 \(n\) 是 token 数、\(k\) 是规则组数、\(W\) 是机器字的位宽。这是遮罩相关存储的改进,不能据此声称整个注意力计算变成线性复杂度。收益在多区域时尤其明显,因为联合编辑只增加提示长度,不需要为每个区域重新生成整张图。

一个完整示例

原文图 3 的桌面示例同时出现红色咖啡杯和装水的玻璃杯。面对“把用过并放下的杯子换成小盆栽”,规划器根据红杯看起来空置、玻璃杯仍装水,选择红杯作为修改对象;这属于模型对场景的推断,不是来自已知物体使用历史。

计划给出红杯区域 [224, 372, 263, 431],提示将其替换为盆栽;另给玻璃杯区域 [175, 329, 220, 388],提示保持不变;全局提示要求其余部分不变。编辑器分别编码这些提示后,红杯 patch 接收替换要求,玻璃杯 patch 接收保护要求,背景接收全局保持要求。

联合去噪时,不同提示彼此隔离,但视觉 token 仍全局通信,因此盆栽可以与桌面保持视觉协调。若结果误改玻璃杯,可先检查规划器是否正确生成保护区域,再检查区域提示的可见性规则,而不必只通过反复重写整句指令来试错。

损失函数 / 训练策略

论文使用约 1k 条复杂编辑的图像与指令样本,不需要配对的目标编辑图像;正文还提到这些样本先用于监督对齐。GRPO 对同一指令采样得到的多个候选计划及其结果进行相对比较,更新 Qwen2.5-VL 7B 规划器,而不是用这些样本训练新的扩散生成骨干。

第一阶段先让计划可被执行:标签格式奖励检查标记结构,区域格式奖励检查 JSON 列表及字典能否正确解析,所谓“推理质量奖励”则按 <think> 中的文本长度增长并设置上限。长度奖励只是推理行为的代理信号,并不直接验证推理是否正确,这一点不能被名称掩盖。

第二阶段把计划交给编辑器生成图像,由 Qwen2.5-VL 72B 评价目标是否改对(Target)、变化是否符合指令(Effect),以及无关区域和全局风格是否保留(Consistency)。为防止“完全不编辑也能获得高一致性”的取巧,原文将一致性奖励乘以效果奖励:

\[ R'_C = R_C \cdot R_E. \]

第二阶段综合目标奖励、效果奖励、加权一致性奖励,并用一个较小权重保留第一阶段格式奖励。缓存中的部分排版公式失真,正文未明确给出该权重和各奖励归一化的数值,故不补写不可核验的超参数。训练评价器 Qwen2.5-VL 72B 与测试评价器 Gemini-2.5-Pro 也应区分。

实验关键数据

主实验

IV-Edit 含约 800 对图像与指令,覆盖 7 类指代、16 类任务,平均指令长度为 21 个词,其中 182 个样本涉及多个目标区域。Gemini-2.5-Pro 按 1 至 5 分评价 Quality、Target、Effect 和 Consistency,Overall 为四项的简单平均;Target 和 Consistency 是评分维度,不是标准检测召回率与精确率。

下表摘录原文表 1,所有指标越高越好。Weighted 旨在以 Effect 调节 Consistency,但正文给出的乘积公式未清楚说明尺度处理,不能直接用表中四列均值重算;这里只保留原表报告值,不自行补公式。

模型 Quality Target Effect Consistency Overall Weighted
Gemini-Flash-Image 3.89 4.11 3.93 2.89 3.71 3.44
Flux.1-Kontext-dev 3.93 3.34 2.73 2.88 3.22 2.49
Flux.1-Kontext-dev + RePlan 4.16 3.47 2.59 3.64 3.46 2.55
Qwen-Image-Edit 3.47 3.72 3.24 1.79 3.05 2.62
Qwen-Image-Edit + RePlan 3.86 3.77 3.16 3.24 3.51 2.91
Flux.2-Klein-9B 4.05 3.81 3.31 2.61 3.45 2.95
Flux.2-Klein-9B + RePlan 4.28 4.04 3.41 3.96 3.92 3.33

消融实验

原文表 2 使用 Flux.1-Kontext-dev 执行编辑,完整方法的 Overall/Weighted 为 3.46/2.55。以下都是原表绝对值,不将零样本规划器的结果误写成不同编辑骨干的性能。

配置 Overall Weighted 说明
Gemini2.5-Pro 零样本规划器 2.95 1.93 作者观察到区域框错误
Qwen2.5-VL 7B 零样本规划器 2.60 1.63 提示分解与格式遵从较弱
去掉推理 3.31 2.49 直接输出区域计划
单阶段 RL 3.42 2.51 跳过第一阶段格式学习
RePlan 完整模型 3.46 2.55 推理与两阶段 RL

关键发现

  • 一致性收益最稳定:三个骨干分别从 2.88、1.79、2.61 提升至 3.64、3.24、3.96;但前两个骨干的 Effect 分别从 2.73、3.24 降至 2.59、3.16,不能声称所有编辑能力全面提升。
  • Flux.2-Klein-9B + RePlan 的 Overall 为 3.92,高于 Gemini-Flash-Image 的 3.71,但 Weighted 为 3.33,低于后者的 3.44;“超过闭源模型”的说法必须限定指标。
  • 原文表 3 的五区域端到端延迟为:多轮 Kontext 221.0 秒、朴素遮罩 RePlan 113.8 秒、规则组注意力 RePlan 54.1 秒;单区域加速版与基线均为 44.2 秒。这些是原文设置下的结果,不代表任意设备的固定耗时。
  • 100 个样本、12 位专家的盲评中,RePlan 胜出 35%,Kontext 胜出 15%,平局 50%;排除平局后才是 70% 的胜率,不能把它描述为全部样本的胜率。

亮点与洞察

  • 显式计划是可调试的中间表示。框与提示把“理解错目标”和“生成器执行错”分开,也允许用户只修正局部计划。
  • 区域控制不等于完全隔离图像区域。隔离文本条件、保留视觉通信,使局部约束与全图风格协调可以同时存在。
  • 算法约束与内核实现相互配合。只有将组规则直接融入注意力计算,免训练遮罩才不会带来过大的执行开销。

局限与展望

  • 本文正文没有独立的局限小节。以下是基于方法与结果的阅读判断,不冒充作者明确承认的结论。
  • 规划器会犯框定位、隐含语义判断和格式错误;长度奖励不能验证思维链真实性。可研究区域验证或结果反馈,但本文没有展示完整的自动纠错闭环。
  • 框外像素并不保证严格不变,部分骨干的 Effect 还下降。对密集重叠对象、冲突局部提示和严格像素保真任务,仍需专门评测。
  • Weighted 的尺度与聚合细节不清楚,自动评分又依赖外部 VLM;100 个样本的人类比较只覆盖 Kontext 组合,不能验证所有骨干及所有指标。
  • 当前缓存提供主文与参考文献,未包含主文提及的附录;完整数据来源、奖励超参数及更详细训练配置无法据此核验。约 1k 样本仅指规划器适配数据,不能忽略基础模型的预训练成本。

相关工作与启发

  • vs Qwen-Image / Bagel:重点不是再训练一个理解生成统一模型,而是把已有 VLM 的推理变成区域级执行条件。区域接口可插拔,但最终画质与编辑能力仍依赖骨干。
  • vs 多轮图像修补:多轮方法按区域顺序执行,RePlan 在一次编辑过程中联合处理多个区域;其速度优势随区域增多更明显,但不是少步扩散算法。
  • vs KrisBench / RISEBench:IV-Edit 更强调复杂场景与复杂指代的交互,并覆盖图中文字、表格和海报等任务;不能只看推理难度而忽略目标定位难度。
  • 资源:ECCV 官方论文页、论文所列项目页。项目链接来自原文,本次未联网核验,未推测代码仓库或 arXiv ID。

评分

  • 新颖性:4/5。区域计划与受限注意力的接口清晰,规则组加速也具有实际价值。
  • 实验充分度:4/5。包含多骨干、规划器和训练消融、延迟与人类比较,但指标口径及复现细节仍有限。
  • 写作质量:4/5。任务动机与执行规则易理解,部分评价尺度和实现边界需要更明确说明。
  • 价值:4/5。对复杂指令和多区域编辑有参考意义,但不应解读为全面替代通用图像编辑模型。