Trust Your Critic: Robust Reward Modeling and Reinforcement Learning for Faithful Image Editing and Generation
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
- 2025Uniworld-V2: Reinforce Image Editing with Diffusion Negative-aware Finetuning and MLLM Implicit Feedback19 Oct
- 2025EditReward: A Human-Aligned Reward Model for Instruction-Guided Image Editing30 Sep
- 2025EditScore: Unlocking Online RL for Image Editing via High-Fidelity Reward Modeling28 Sep
Other references · 48
- 2020Denoising Diffusion Probabilistic Modelscited by 20,851
- 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,694
- 2022Photorealistic Text-to-Image Diffusion Models with Deep Language Understandingcited by 6,322
- 2023Adding Conditional Control to Text-to-Image Diffusion Modelscited by 5,466
- 2024GPT-4o System Cardcited by 4,467
- 2023SDXL: Improving Latent Diffusion Models for High-Resolution Image Synthesiscited by 4,290
- 2024Scaling Rectified Flow Transformers for High-Resolution Image Synthesiscited by 4,094
- 2021GLIDE: Towards Photorealistic Image Generation and Editing with Text-Guided Diffusion Modelscited by 3,595
- 2022InstructPix2Pix: Learning to Follow Image Editing Instructionscited by 2,591
- 2025Qwen3-VL Technical Reportcited by 2,534
- 2022Prompt-to-Prompt Image Editing with Cross Attention Controlcited by 2,409
- 2024LlamaFactory: Unified Efficient Fine-Tuning of 100+ Language Modelscited by 1,442
36 more
- 2025InternVL3.5: Advancing Open-Source Multimodal Models in Versatility, Reasoning, and Efficiencycited by 1,419
- 2022Scaling Autoregressive Models for Content-Rich Text-to-Image Generationcited by 1,287
- 2024Visual Autoregressive Modeling: Scalable Image Generation via Next-Scale Predictioncited by 1,005
- 2023GenEval: An Object-Focused Framework for Evaluating Text-to-Image Alignmentcited by 946
- 2023Training Diffusion Models with Reinforcement Learningcited by 939
- 2022eDiff-I: Text-to-Image Diffusion Models with an Ensemble of Expert Denoiserscited by 921
- 2025OpenAI GPT-5 System Cardcited by 910
- 2025Emerging Properties in Unified Multimodal Pretrainingcited by 855
- 2025Janus-Pro: Unified Multimodal Understanding and Generation with Data and Model Scalingcited by 787
- 2024Emu3: Next-Token Prediction is All You Needcited by 761
- 2024Show-o: One Single Transformer to Unify Multimodal Understanding and Generationcited by 734
- 2023MagicBrush: A Manually Annotated Dataset for Instruction-Guided Image Editingcited by 615
- 2024ELLA: Equip Diffusion Models with LLM for Enhanced Semantic Alignmentcited by 455
- 2026Kimi K2.5: Visual Agentic Intelligencecited by 449
- 2023DPOK: Reinforcement Learning for Fine-tuning Text-to-Image Diffusion Modelscited by 431
- 2025Step1X-Edit: A Practical Framework for General Image Editingcited by 417
- 2025BLIP3-o: A Family of Fully Open Unified Multimodal Models-Architecture, Training and Datasetcited by 363
- 2023Emu Edit: Precise Image Editing via Recognition and Generation Taskscited by 321
- 2022VQGAN-CLIP: Open Domain Image Generation and Editing with Natural Language Guidancecited by 285
- 2022EDICT: Exact Diffusion Inversion via Coupled Transformationscited by 256
- 2025Z-Image: An Efficient Image Generation Foundation Model with Single-Stream Diffusion Transformercited by 247
- 2025Show-o2: Improved Native Unified Multimodal Modelscited by 242
- 2025OmniGen2: Towards Instruction-Aligned Multimodal Generationcited by 241
- 2024AnyEdit: Mastering Unified High-Quality Image Editing for Any Ideacited by 235
- 2025ImgEdit: A Unified Image Editing Dataset and Benchmarkcited by 225
- 2025DiffusionNFT: Online Diffusion Reinforcement with Forward Processcited by 204
- 2025In-Context Edit: Enabling Instructional Image Editing with In-Context Generation in Large Scale Diffusion Transformercited by 199
- 2024JanusFlow: Harmonizing Autoregression and Rectified Flow for Unified Multimodal Understanding and Generationcited by 162
- 2025T2I-R1: Reinforcing Image Generation with Collaborative Semantic-level and Token-level CoTcited by 162
- 2025Envisioning Beyond the Pixels: Benchmarking Reasoning-Informed Visual Editingcited by 97
- 2025ShareGPT-4o-Image: Aligning Multimodal Models with GPT-4o-Level Image Generationcited by 94
- 2025GPT-IMAGE-EDIT-1.5M: A Million-Scale, GPT-Generated Image Datasetcited by 68
- 2025TIIF-Bench: How Does Your T2I Model Follow Your Instructions?cited by 64
- 2025OpenGPT-4o-Image: A Comprehensive Dataset for Advanced Image Generation and Editingcited by 27
- 2025UniGenBench++: A Unified Semantic Evaluation Benchmark for Text-to-Image Generationcited by 23
- 2025Ovis-Image Technical Reportcited by 8
Built on by
Citing papers per year: 2026: 9.
9 papers.
- 2026JoyAI-Image: Awaking Spatial Intelligence in Unified Multimodal Understanding and Generationcited by 10
- 2026Auto-Rubric as Reward: From Implicit Preferences to Explicit Multimodal Generative Criteriacited by 8
- 2026GMO-E2DIT: Grounded Multi-Operation Editing for E-Commerce Imagescited by 3
- 2026FIRM-Video: Check Before You Score for Reliable Text-to-Video Reward Modelingcited by 2
- 2026Can We Perform Online RL for Image Editing without Editing Rewards?cited by 1
- 2026Beyond the Current Observation: Evaluating Multimodal Large Language Models in Controllable Non-Markov Gamescited by 0
- 2026Think Before You Score: Thinking Reward Model for Visual Generationcited by 0
- 2026Do MLLM Judges Judge the Edit? Auditing Bias in Image Editing Evaluation with Verified Quality Preservationcited by 0
- 2026Reasoning-Informed Visual Editingcited by 0
- 2026Reasoning-Informed Visual Editing8 Oct 2026
- 2026Do MLLM Judges Judge the Edit? Auditing Bias in Image Editing Evaluation with Verified Quality Preservation1 Oct 2026
- 2026Think Before You Score: Thinking Reward Model for Visual Generation29 Sep 2026
- 2026Can We Perform Online RL for Image Editing without Editing Rewards?24 Aug 2026
- 2026FIRM-Video: Check Before You Score for Reliable Text-to-Video Reward Modeling22 Aug 2026
- 2026GMO-E2DIT: Grounded Multi-Operation Editing for E-Commerce Images1 Jul 2026
- 2026Beyond the Current Observation: Evaluating Multimodal Large Language Models in Controllable Non-Markov Games17 Jun 2026
- 2026Auto-Rubric as Reward: From Implicit Preferences to Explicit Multimodal Generative Criteria8 May 2026
- 2026JoyAI-Image: Awaking Spatial Intelligence in Unified Multimodal Understanding and Generation5 May 2026
Similar, not cited
Close in wording, no citation either way
- 2026Understanding Reward Hacking in Text-to-Image Reinforcement Learning
- 2026Leveraging Verifier-Based Reinforcement Learning in Image Editing
- 2026Evaluation-Verification Reward for Consistent Multi-Reference Image Editing
- 2025UniGen-1.5: Enhancing Image Generation and Editing through Reward Unification in Reinforcement Learning
- 2024RL for Consistency Models: Faster Reward Guided Text-to-Image Generation
- 2026Qwen-Image-2.0-RL Technical Report
- 2025The Image as Its Own Reward: Reinforcement Learning with Adversarial Reward for Image Generation
- 2026MT-EditFlow: Reinforcement Learning for Multi-Turn Image Editing with Flow Matching