Search papers, labs, and topics across Lattice.
This paper introduces DARS, a novel reinforcement learning framework that enhances instruction-based image editing by implementing dual-level credit assignment across a planner-renderer pipeline. By utilizing multi-plan and multi-render rollouts, DARS effectively estimates reward variability, allowing for adaptive curriculum adjustments and improved supervision through structured reasoning outputs. Experimental results demonstrate that DARS significantly outperforms a Joint RL baseline, particularly in tasks requiring complex reasoning, highlighting its effectiveness in optimizing both planning and rendering stages.
DARS achieves superior performance in instruction-based image editing by transforming outcome-level feedback into actionable, localized supervision for both planning and rendering stages.
Instruction-based image editing uses a planner-renderer pipeline: a vision-language model (VLM) first converts the instruction into an edit plan, and a diffusion model then executes that plan. Training such systems with only final-image rewards is inefficient because a poor edit does not reveal whether additional optimization should place more emphasis on the planner or the renderer, and even planner-dominant cases remain difficult to localize within a free-form reasoning trace. We present DARS, a reinforcement learning framework for dual-level credit assignment in this two-stage setting. Across modules, multi-plan multi-render rollouts estimate between-plan and within-plan reward variability for soft module routing, while rollout mean rewards provide hardness estimates for an adaptive curriculum. Within the planner, a four-field structured reasoning output enables a prefix-gated reward and token-level advantage reweighting, turning outcome-level feedback into localized supervision. Experiments on five benchmarks show that DARS outperforms a Joint~RL baseline with the same backbone, data, reward model, and rollout budget, with the largest gains on reasoning-intensive edits.
