mlsanityData to 7 Oct 2026

FlowErase-RL: Rethinking Concept Erasure as Reward Optimization in Flow Matching Models

arXiv 2605.19739 ↗First posted 19 May 2026cs.CVcited by 1

On the map: Diffusion models → Robustness, privacy and adversarial aspects

Abstract (arXiv)

Recent advances in flow matching models have significantly improved text-to-image generation quality, but also introduce growing safety risks due to the generation of harmful or undesirable content. Existing concept erasure methods are either inference-time interventions with limited effectiveness or rely on supervised fine-tuning (SFT), which requires precisely aligned data and struggles with scalability and multi-concept settings. In this paper, we propose FlowErase-RL, the first GRPO-based framework for concept erasure in flow matching models. We reformulate concept erasure as a reward optimization problem and introduce a dynamic dual-path reward mechanism that jointly optimizes (i) a Concept Erasure (CE) reward to suppress target concepts and (ii) a Non-target Space (NS) reward to preserve generative fidelity. The two reward paths are adaptively balanced during training via a performance-driven switching strategy, enabling stable optimization without explicit supervision. Extensive experiments on nudity, object, and artistic style erasure demonstrate that our method achieves state-of-the-art erasure performance while maintaining strong image quality and semantic alignment. Moreover, it exhibits robust resistance to adversarial attacks and scales effectively to multi-concept scenarios. Our results establish a new paradigm for safe and controllable generation in flow matching models.

Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Placed in Robustness, privacy and adversarial aspects, one of 14 directions in Diffusion models. Builds on 37 of its 49 references that are in the map. First posted on arXiv 19 May 2026.

Builds on

37 of 49 references.

Closest ancestors · 19

Same direction, newest first

  1. 2026Differential Vector Erasure: Unified Training-Free Concept Erasure for Flow Matching Models1 Feb
  2. 2026ActErase: A Training-Free Paradigm for Precise Concept Erasure via Activation Redirection1 Jan
  3. 2025TRCE: Towards Reliable Malicious Concept Erasure in Text-to-Image Diffusion Models10 Mar
  4. 2025SAeUron: Interpretable Concept Unlearning in Diffusion Models with Sparse Autoencoders29 Jan
  5. 2024EraseAnything: Enabling Concept Erasure in Rectified Flow Transformers29 Dec
  6. 2024Precise, Fast, and Low-cost Concept Erasure in Value Space: Orthogonal Complement Matters9 Dec
  7. 2024ConceptPrune: Concept Editing in Diffusion Models via Skilled Neuron Pruning29 May
  8. 2024R.A.C.E.: Robust Adversarial Concept Erasure for Secure Text-to-Image Diffusion Model25 May
  9. 2024SafeGen: Mitigating Sexually Explicit Content Generation in Text-to-Image Models10 Apr
  10. 2024MACE: Mass Concept Erasure in Diffusion Models10 Mar
  11. 2023MMA-Diffusion: MultiModal Attack on Diffusion Models29 Nov
  12. 2023To Generate or Not? Safety-Driven Unlearned Diffusion Models Are Still Easy To Generate Unsafe Images ... For Now18 Oct
7 more
  1. 2023Ring-A-Bell! How Reliable are Concept Removal Methods for Diffusion Models?16 Oct
  2. 2023Prompting4Debugging: Red-Teaming Text-to-Image Diffusion Models by Finding Problematic Prompts12 Sep
  3. 2023Unified Concept Editing in Diffusion Models25 Aug
  4. 2023Forget-Me-Not: Learning to Forget in Text-to-Image Diffusion Models30 Mar
  5. 2023Erasing Concepts from Diffusion Models13 Mar
  6. 2022Safe Latent Diffusion: Mitigating Inappropriate Degeneration in Diffusion Models9 Nov
  7. 2022Red-Teaming the Stable Diffusion Safety Filter3 Oct

Other references · 18

  1. 2015Deep Residual Learning for Image Recognitioncited by 51,809
  2. 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,456
  3. 2020Denoising Diffusion Probabilistic Modelscited by 20,851
  4. 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,694
  5. 2014Microsoft COCO: Common Objects in Contextcited by 17,498
  6. 2020Denoising Diffusion Implicit Modelscited by 9,655
  7. 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
  8. 2021Diffusion Models Beat GANs on Image Synthesiscited by 8,185
  9. 2022Photorealistic Text-to-Image Diffusion Models with Deep Language Understandingcited by 6,322
  10. 2022Classifier-Free Diffusion Guidancecited by 5,717
  11. 2022Flow Matching for Generative Modelingcited by 5,426
  12. 2024Scaling Rectified Flow Transformers for High-Resolution Image Synthesiscited by 4,094
6 more
  1. 2021GLIDE: Towards Photorealistic Image Generation and Editing with Text-Guided Diffusion Modelscited by 3,595
  2. 2022Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flowcited by 3,550
  3. 2025Flow-GRPO: Training Flow Matching Models via Online RLcited by 605
  4. 2017GANs Trained by a Two Time-Scale Update Rule Converge to a Local Nash Equilibriumcited by 208
  5. 2024Latent Guard: a Safety Framework for Text-to-image Generationcited by 68
  6. 2025Closing the Safety Gap: Surgical Concept Erasure in Visual Autoregressive Modelscited by 7

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. 2026EraseAnything++: Enabling Concept Erasure in Rectified Flow Transformers Leveraging Multi-Object Optimization
  2. 2025EraseFlow: Learning Concept Erasure Policies via GFlowNet-Driven Alignment
  3. 2025A Comprehensive Survey on Concept Erasure in Text-to-Image Diffusion Models
  4. 2026Erasing Without Collateral Damage: Precise Concept Removal in Diffusion Models
  5. 2025Semantic Surgery: Zero-Shot Concept Erasure in Diffusion Models
  6. 2026Closed-Form Concept Erasure via Double Projections
  7. 2025Minimalist Concept Erasure in Generative Models
  8. 2026GEM: Geometric Erasure by Contrastive Velocity Matching in Rectified Flows