mlsanityData to 7 Oct 2026

Trust Your Critic: Robust Reward Modeling and Reinforcement Learning for Faithful Image Editing and Generation

arXiv 2603.12247 ↗First posted 12 Mar 2026cs.CVcited by 9

On the map: LLM alignment / RLHF / preference optimisation → Diffusion and multimodal model alignment

Abstract (arXiv)

Reinforcement learning (RL) has emerged as a promising paradigm for enhancing image editing and text-to-image (T2I) generation. However, current reward models, which act as critics during RL, often suffer from hallucinations and assign noisy scores, inherently misguiding the optimization process. In this paper, we present FIRM (Faithful Image Reward Modeling), a comprehensive framework that develops robust reward models to provide accurate and reliable guidance for faithful image generation and editing. First, we design tailored data curation pipelines to construct high-quality scoring datasets. Specifically, we evaluate editing using both execution and consistency, while generation is primarily assessed via instruction following. Using these pipelines, we collect the FIRM-Edit-370K and FIRM-Gen-293K datasets, and train specialized reward models (FIRM-Edit-8B and FIRM-Gen-8B) that accurately reflect these criteria. Second, we introduce FIRM-Bench, a comprehensive benchmark specifically designed for editing and generation critics. Evaluations demonstrate that our models achieve superior alignment with human judgment compared to existing metrics. Furthermore, to seamlessly integrate these critics into the RL pipeline, we formulate a novel "Base-and-Bonus" reward strategy that balances competing objectives: Consistency-Modulated Execution (CME) for editing and Quality-Modulated Alignment (QMA) for generation. Empowered by this framework, our resulting models FIRM-Qwen-Edit and FIRM-SD3.5 achieve substantial performance breakthroughs. Comprehensive experiments demonstrate that FIRM mitigates hallucinations, establishing a new standard for fidelity and instruction adherence over existing general models. All of our datasets, models, and code have been publicly available at https://firm-reward.github.io.

Key facts. Cited by 9 papers in the map (data through 7 Oct 2026). Placed in Diffusion and multimodal model alignment, one of 10 directions in LLM alignment / RLHF / preference optimisation. Builds on 51 of its 55 references that are in the map. First posted on arXiv 12 Mar 2026.

Builds on

51 of 55 references.

Closest ancestors · 3

Same direction, newest first

  1. 2025Uniworld-V2: Reinforce Image Editing with Diffusion Negative-aware Finetuning and MLLM Implicit Feedback19 Oct
  2. 2025EditReward: A Human-Aligned Reward Model for Instruction-Guided Image Editing30 Sep
  3. 2025EditScore: Unlocking Online RL for Image Editing via High-Fidelity Reward Modeling28 Sep

Other references · 48

  1. 2020Denoising Diffusion Probabilistic Modelscited by 20,851
  2. 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,694
  3. 2022Photorealistic Text-to-Image Diffusion Models with Deep Language Understandingcited by 6,322
  4. 2023Adding Conditional Control to Text-to-Image Diffusion Modelscited by 5,466
  5. 2024GPT-4o System Cardcited by 4,467
  6. 2023SDXL: Improving Latent Diffusion Models for High-Resolution Image Synthesiscited by 4,290
  7. 2024Scaling Rectified Flow Transformers for High-Resolution Image Synthesiscited by 4,094
  8. 2021GLIDE: Towards Photorealistic Image Generation and Editing with Text-Guided Diffusion Modelscited by 3,595
  9. 2022InstructPix2Pix: Learning to Follow Image Editing Instructionscited by 2,591
  10. 2025Qwen3-VL Technical Reportcited by 2,534
  11. 2022Prompt-to-Prompt Image Editing with Cross Attention Controlcited by 2,409
  12. 2024LlamaFactory: Unified Efficient Fine-Tuning of 100+ Language Modelscited by 1,442
36 more
  1. 2025InternVL3.5: Advancing Open-Source Multimodal Models in Versatility, Reasoning, and Efficiencycited by 1,419
  2. 2022Scaling Autoregressive Models for Content-Rich Text-to-Image Generationcited by 1,287
  3. 2024Visual Autoregressive Modeling: Scalable Image Generation via Next-Scale Predictioncited by 1,005
  4. 2023GenEval: An Object-Focused Framework for Evaluating Text-to-Image Alignmentcited by 946
  5. 2023Training Diffusion Models with Reinforcement Learningcited by 939
  6. 2022eDiff-I: Text-to-Image Diffusion Models with an Ensemble of Expert Denoiserscited by 921
  7. 2025OpenAI GPT-5 System Cardcited by 910
  8. 2025Emerging Properties in Unified Multimodal Pretrainingcited by 855
  9. 2025Janus-Pro: Unified Multimodal Understanding and Generation with Data and Model Scalingcited by 787
  10. 2024Emu3: Next-Token Prediction is All You Needcited by 761
  11. 2024Show-o: One Single Transformer to Unify Multimodal Understanding and Generationcited by 734
  12. 2023MagicBrush: A Manually Annotated Dataset for Instruction-Guided Image Editingcited by 615
  13. 2024ELLA: Equip Diffusion Models with LLM for Enhanced Semantic Alignmentcited by 455
  14. 2026Kimi K2.5: Visual Agentic Intelligencecited by 449
  15. 2023DPOK: Reinforcement Learning for Fine-tuning Text-to-Image Diffusion Modelscited by 431
  16. 2025Step1X-Edit: A Practical Framework for General Image Editingcited by 417
  17. 2025BLIP3-o: A Family of Fully Open Unified Multimodal Models-Architecture, Training and Datasetcited by 363
  18. 2023Emu Edit: Precise Image Editing via Recognition and Generation Taskscited by 321
  19. 2022VQGAN-CLIP: Open Domain Image Generation and Editing with Natural Language Guidancecited by 285
  20. 2022EDICT: Exact Diffusion Inversion via Coupled Transformationscited by 256
  21. 2025Z-Image: An Efficient Image Generation Foundation Model with Single-Stream Diffusion Transformercited by 247
  22. 2025Show-o2: Improved Native Unified Multimodal Modelscited by 242
  23. 2025OmniGen2: Towards Instruction-Aligned Multimodal Generationcited by 241
  24. 2024AnyEdit: Mastering Unified High-Quality Image Editing for Any Ideacited by 235
  25. 2025ImgEdit: A Unified Image Editing Dataset and Benchmarkcited by 225
  26. 2025DiffusionNFT: Online Diffusion Reinforcement with Forward Processcited by 204
  27. 2025In-Context Edit: Enabling Instructional Image Editing with In-Context Generation in Large Scale Diffusion Transformercited by 199
  28. 2024JanusFlow: Harmonizing Autoregression and Rectified Flow for Unified Multimodal Understanding and Generationcited by 162
  29. 2025T2I-R1: Reinforcing Image Generation with Collaborative Semantic-level and Token-level CoTcited by 162
  30. 2025Envisioning Beyond the Pixels: Benchmarking Reasoning-Informed Visual Editingcited by 97
  31. 2025ShareGPT-4o-Image: Aligning Multimodal Models with GPT-4o-Level Image Generationcited by 94
  32. 2025GPT-IMAGE-EDIT-1.5M: A Million-Scale, GPT-Generated Image Datasetcited by 68
  33. 2025TIIF-Bench: How Does Your T2I Model Follow Your Instructions?cited by 64
  34. 2025OpenGPT-4o-Image: A Comprehensive Dataset for Advanced Image Generation and Editingcited by 27
  35. 2025UniGenBench++: A Unified Semantic Evaluation Benchmark for Text-to-Image Generationcited by 23
  36. 2025Ovis-Image Technical Reportcited by 8

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 9.

9 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2026Understanding Reward Hacking in Text-to-Image Reinforcement Learning
  2. 2026Leveraging Verifier-Based Reinforcement Learning in Image Editing
  3. 2026Evaluation-Verification Reward for Consistent Multi-Reference Image Editing
  4. 2025UniGen-1.5: Enhancing Image Generation and Editing through Reward Unification in Reinforcement Learning
  5. 2024RL for Consistency Models: Faster Reward Guided Text-to-Image Generation
  6. 2026Qwen-Image-2.0-RL Technical Report
  7. 2025The Image as Its Own Reward: Reinforcement Learning with Adversarial Reward for Image Generation
  8. 2026MT-EditFlow: Reinforcement Learning for Multi-Turn Image Editing with Flow Matching