DARS Improves Instruction-Based Image Editing with Dual-Level RL.

Haoxiang Cao, Jiajiong Cao, Xuanpu Zhang, Changqian Yu, Chaoqun Wang· August 21, 2026 View original

Key takeaways

  • Training instruction-based image editing systems with only final-image rewards is inefficient.
  • DARS introduces dual-level credit assignment for optimizing both the planner and renderer modules.
  • Structured reasoning outputs enable localized feedback and token-level advantage reweighting within the planner.
  • DARS significantly improves performance on reasoning-intensive image edits compared to existing methods.

Who benefits

Creative ArtsMarketingMedia & EntertainmentE-commerce

Summary

Researchers introduce DARS, a reinforcement learning framework for instruction-based image editing that uses dual-level credit assignment. DARS optimizes both the planner and renderer in a two-stage pipeline, significantly outperforming baselines by localizing feedback and improving reasoning-intensive edits.

Instruction-based image editing typically relies on a two-stage pipeline: a vision-language model (VLM) generates an edit plan, which a diffusion model then executes. A major challenge in training such systems is the inefficiency of using only final-image rewards. When an edit is poor, it's difficult to determine whether the planner or the renderer needs more optimization, and even within the planner's free-form reasoning trace, localizing the exact cause of failure is hard. To address this, the DARS (Dual-Level Credit Assignment RL with Structured Reasoning) framework has been developed. DARS introduces a reinforcement learning approach that performs credit assignment at two levels. Across the modules, multi-plan multi-render rollouts estimate reward variability to enable soft module routing and provide hardness estimates for an adaptive curriculum. Within the planner, DARS utilizes a four-field structured reasoning output, allowing for prefix-gated rewards and token-level advantage reweighting. This transforms outcome-level feedback into precise, localized supervision. Experiments across five benchmarks demonstrate that DARS significantly outperforms a Joint RL baseline, especially on complex, reasoning-intensive edits, by effectively localizing and applying feedback.

Why it matters

This research provides a more efficient and effective way to train AI models for complex image editing tasks, leading to more precise and reliable results for creative professionals and content creators.

How to implement this in your domain

  1. 1Adopt dual-level credit assignment and structured reasoning techniques for training multi-stage generative AI pipelines.
  2. 2Implement adaptive curriculum learning based on task hardness estimates to optimize training efficiency for complex tasks.
  3. 3Explore prefix-gated rewards and token-level advantage reweighting for more localized feedback in VLM-based planning.
  4. 4Integrate DARS principles into image editing tools to enhance their ability to follow complex natural language instructions.

Original post by Haoxiang Cao, Jiajiong Cao, Xuanpu Zhang, Changqian Yu, Chaoqun Wang

"arXiv:2608.20161v1 Announce Type: new Abstract: 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-i…"

View on X

Originally posted by Haoxiang Cao, Jiajiong Cao, Xuanpu Zhang, Changqian Yu, Chaoqun Wang on X · view source

Want to go deeper?

Turn these trends into skills with Learnijoy's hands-on AI & tech courses.

Explore courses