mlsanityData to 7 Oct 2026

MT-EditFlow: Reinforcement Learning for Multi-Turn Image Editing with Flow Matching

arXiv 2606.01985 ↗First posted 1 Jun 2026cs.CVcited by 1

On the map: Diffusion models → Image generation, text-to-image and editing

Abstract (arXiv)

Recent breakthroughs in instruction-based image editing have captured significant attention, as models are now capable of handling real-world editing demands with the practicality required by everyday users. However, editing models trained primarily for single-turn edits often break down in multi-turn editing--the natural interactive setting where a user iteratively refines an image based on the model's own previous outputs. This failure stems from the all-or-nothing requirement, where a single failed turn compromises the entire sequence, and error propagation, where exposure bias leads to compounding editing errors. To address these challenges, we introduce MT-EditFlow, a flow-matching reinforcement learning framework designed to optimize reward signals for sequential image editing. MT-EditFlow integrates a multi-turn perspective with a multi-reward formulation to provide a unified structure applicable to both GRPO and NFT-based reinforcement learning methods. We systematically analyze and optimize the reward signal by investigating effective scoring strategies for turn-level aggregation, VLM reasoning modes to trade off reward bias and variance, and advantage fusion levels to prevent reward hacking. Our findings reveal that broadcasting the aggregated advantage across the entire editing trajectory effectively bridges the gap between local planning and global multi-turn task success. Extensive experiments demonstrate that MT-EditFlow significantly improves performance across diverse base models. Notably, it boosts FLUX.1-Kontext-dev by 6.85 points in turn-3 overall performance, surpassing state-of-the-art open-source models such as Qwen-Image-Edit. By maintaining high marginal success rates and reducing exposure bias, MT-EditFlow provides a foundation for more reliable and natural human-AI collaboration in visual content creation.

Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Placed in Image generation, text-to-image and editing, one of 14 directions in Diffusion models. Builds on 54 of its 69 references that are in the map. First posted on arXiv 1 Jun 2026.

Builds on

54 of 69 references.

Closest ancestors · 5

Same direction, newest first

  1. 2025FLUX.1 Kontext: Flow Matching for In-Context Image Generation and Editing in Latent Space17 Jun
  2. 2024OmniGen: Unified Image Generation17 Sep
  3. 2024Object-Conditioned Energy-Based Attention Map Alignment in Text-to-Image Diffusion Models10 Apr
  4. 2023Rich Human Feedback for Text-to-Image Generation15 Dec
  5. 2022InstructPix2Pix: Learning to Follow Image Editing Instructions17 Nov

Other references · 49

  1. 2020Language Models are Few-Shot Learnerscited by 32,199
  2. 2020Denoising Diffusion Probabilistic Modelscited by 20,851
  3. 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,694
  4. 2023GPT-4 Technical Reportcited by 18,132
  5. 2017Proximal Policy Optimization Algorithmscited by 14,441
  6. 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,084
  7. 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 8,964
  8. 2025Qwen3 Technical Reportcited by 8,195
  9. 2022Classifier-Free Diffusion Guidancecited by 5,717
  10. 2022Flow Matching for Generative Modelingcited by 5,426
  11. 2025Gemini 2.5: Pushing the Frontier with Advanced Reasoning, Multimodality, Long Context, and Next Generation Agentic Capabilitiescited by 3,655
  12. 2022Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flowcited by 3,550
37 more
  1. 2022Prompt-to-Prompt Image Editing with Cross Attention Controlcited by 2,409
  2. 2025DINOv3cited by 1,297
  3. 2019ZeRO: Memory Optimizations Toward Training Trillion Parameter Modelscited by 1,278
  4. 2023Training Diffusion Models with Reinforcement Learningcited by 939
  5. 2024SpatialVLM: Endowing Vision-Language Models with Spatial Reasoning Capabilitiescited by 679
  6. 2021Maximum Likelihood Training of Score-Based Diffusion Modelscited by 670
  7. 2023PyTorch FSDP: Experiences on Scaling Fully Sharded Data Parallelcited by 641
  8. 2023MagicBrush: A Manually Annotated Dataset for Instruction-Guided Image Editingcited by 615
  9. 2025Flow-GRPO: Training Flow Matching Models via Online RLcited by 605
  10. 2023DPOK: Reinforcement Learning for Fine-tuning Text-to-Image Diffusion Modelscited by 431
  11. 2025Step1X-Edit: A Practical Framework for General Image Editingcited by 417
  12. 2025DanceGRPO: Unleashing GRPO on Visual Generationcited by 388
  13. 2024Hallucination of Multimodal Large Language Models: A Surveycited by 380
  14. 2023Emu Edit: Precise Image Editing via Recognition and Generation Taskscited by 321
  15. 2024SpatialRGPT: Grounded Spatial Reasoning in Vision Language Modelscited by 297
  16. 2024UltraEdit: Instruction-based Fine-Grained Image Editing at Scalecited by 266
  17. 2024Reinforcement learningcited by 236
  18. 2024AnyEdit: Mastering Unified High-Quality Image Editing for Any Ideacited by 235
  19. 2025ImgEdit: A Unified Image Editing Dataset and Benchmarkcited by 225
  20. 2025DiffusionNFT: Online Diffusion Reinforcement with Forward Processcited by 204
  21. 2023VIEScore: Towards Explainable Metrics for Conditional Image Synthesis Evaluationcited by 196
  22. 2025HPSv3: Towards Wide-Spectrum Human Preference Scorecited by 196
  23. 2024HQ-Edit: A High-Quality Dataset for Instruction-based Image Editingcited by 191
  24. 2025Seedream 3.0 Technical Reportcited by 146
  25. 2019Generalization in Generation: A closer look at Exposure Biascited by 123
  26. 2025Uniworld-V2: Reinforce Image Editing with Diffusion Negative-aware Finetuning and MLLM Implicit Feedbackcited by 78
  27. 2025Vision Language Models are Biasedcited by 63
  28. 2024I2EBench: A Comprehensive Benchmark for Instruction-based Image Editingcited by 61
  29. 2025EditScore: Unlocking Online RL for Image Editing via High-Fidelity Reward Modelingcited by 61
  30. 2025EditReward: A Human-Aligned Reward Model for Instruction-Guided Image Editingcited by 59
  31. 2024Do Vision-Language Models Represent Space and How? Evaluating Spatial Frame of Reference Under Ambiguitiescited by 58
  32. 2025Pico-Banana-400K: A Large-Scale Dataset for Text-Guided Image Editingcited by 58
  33. 2024Flow Priors for Linear Inverse Problems via Iterative Corrupted Trajectory Matchingcited by 50
  34. 2025textttComplex-Edit: CoT-Like Instruction Generation for Complexity-Controllable Image Editing Benchmarkcited by 25
  35. 2024LVLM-COUNT: Enhancing the Counting Ability of Large Vision-Language Modelscited by 20
  36. 2025GIE-Bench: Towards Grounded Evaluation for Text-Guided Image Editingcited by 15
  37. 2025EdiVal-Agent: An Object-Centric Framework for Automated, Fine-Grained Evaluation of Multi-Turn Editingcited by 10

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 1.

1 paper.

Similar, not cited

Close in wording, no citation either way

  1. 2026Leveraging Verifier-Based Reinforcement Learning in Image Editing
  2. 2026Edit-R2: Context-Aware Reinforcement Learning for Multi-Turn Image Editing
  3. 2026Region-Constrained Group Relative Policy Optimization for Flow-Based Image Editing
  4. 2026SpatialFlow-GRPO: Where Spatial Credit Drives Image Editing
  5. 2025Training-Free Reward-Guided Image Editing via Trajectory Optimal Control
  6. 2026Scalable Minimal-Change Learning for Controllable Image Editing
  7. 2026ImageEdit-R1: Boosting Multi-Agent Image Editing via Reinforcement Learning
  8. 2026Can We Perform Online RL for Image Editing without Editing Rewards?