mlsanityData to 14 Oct 2026

GEM: Geometric Erasure by Contrastive Velocity Matching in Rectified Flows

arXiv 2606.00140 ↗First posted 29 May 2026cs.LG cs.AIcited by 2

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

also in 1 more
Abstract (arXiv)

While the rapid adoption of multimodal generative models offers immense potential, it has also increased the risks of harmful content synthesis, deepfakes, and copyright infringements. To address these challenges, concept erasure has emerged as a prospective safeguard. However, as the field gradually transitions from U-Net-based diffusion models to Rectified Flow Transformers, erasure research has struggled to keep pace. In this work, we introduce GEM, a simple but highly effective erasure framework for Rectified Flow models. As part of our contribution, we establish a principled bridge between trajectory-based unlearning grounded in Generative Flow Networks and classic teacher-guided erasure: we translate trajectory-based signals into a teacher-guided flow-matching setup that unifies the strengths of both paradigms. Concretely, a teacher provides complementary attraction and repulsion signals that we combine into a single geometric guidance objective, yielding targeted suppression of unwanted concepts while preserving benign generation. Our code is available at https://github.com/multimodal-ai-lab/GEM.

Key facts. Cited by 2 papers in the map (data through 14 Oct 2026). Placed in Robustness, privacy and adversarial aspects, one of 14 directions in Diffusion models. Builds on 40 of its 53 references that are in the map. First posted on arXiv 29 May 2026.

Builds on

40 of 53 references.

Closest ancestors · 18

Same direction, newest first

  1. 2025Erased but Not Forgotten: How Backdoors Compromise Concept Erasure29 Apr
  2. 2024EraseAnything: Enabling Concept Erasure in Rectified Flow Transformers29 Dec
  3. 2024TraSCE: Trajectory Steering for Concept Erasure10 Dec
  4. 2024STEREO: A Two-Stage Framework for Adversarially Robust Concept Erasing from Text-to-Image Diffusion Models29 Aug
  5. 2024Reliable and Efficient Concept Erasure of Text-to-Image Diffusion Models17 Jul
  6. 2024R.A.C.E.: Robust Adversarial Concept Erasure for Secure Text-to-Image Diffusion Model25 May
  7. 2024Defensive Unlearning with Adversarial Training for Robust Concept Erasure in Diffusion Models24 May
  8. 2024MACE: Mass Concept Erasure in Diffusion Models10 Mar
  9. 2023One-Dimensional Adapter to Rule Them All: Concepts, Diffusion Models and Erasing Applications26 Dec
  10. 2023MMA-Diffusion: MultiModal Attack on Diffusion Models29 Nov
  11. 2023Ring-A-Bell! How Reliable are Concept Removal Methods for Diffusion Models?16 Oct
  12. 2023Prompting4Debugging: Red-Teaming Text-to-Image Diffusion Models by Finding Problematic Prompts12 Sep
6 more
  1. 2023Unified Concept Editing in Diffusion Models25 Aug
  2. 2023Forget-Me-Not: Learning to Forget in Text-to-Image Diffusion Models30 Mar
  3. 2023Ablating Concepts in Text-to-Image Diffusion Models23 Mar
  4. 2023Erasing Concepts from Diffusion Models13 Mar
  5. 2022Safe Latent Diffusion: Mitigating Inappropriate Degeneration in Diffusion Models9 Nov
  6. 2022Red-Teaming the Stable Diffusion Safety Filter3 Oct

Other references · 22

  1. 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,744
  2. 2020Denoising Diffusion Probabilistic Modelscited by 21,076
  3. 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,713
  4. 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,863
  5. 2014Microsoft COCO: Common Objects in Contextcited by 17,608
  6. 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,599
  7. 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 9,122
  8. 2022Scalable Diffusion Models with Transformerscited by 6,054
  9. 2022Classifier-Free Diffusion Guidancecited by 5,847
  10. 2022Flow Matching for Generative Modelingcited by 5,702
  11. 2024Scaling Rectified Flow Transformers for High-Resolution Image Synthesiscited by 4,181
  12. 2025Gemini 2.5: Pushing the Frontier with Advanced Reasoning, Multimodality, Long Context, and Next Generation Agentic Capabilitiescited by 3,796
10 more
  1. 2022Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flowcited by 3,700
  2. 2025FLUX.1 Kontext: Flow Matching for In-Context Image Generation and Editing in Latent Spacecited by 1,006
  3. 2025Flow-GRPO: Training Flow Matching Models via Online RLcited by 661
  4. 2021GFlowNet Foundationscited by 290
  5. 2022Trajectory balance: Improved credit assignment in GFlowNetscited by 251
  6. 2022Can Machines Help Us Answering Question 16 in Datasheets, and In Turn Reflecting on Inappropriate Content?cited by 155
  7. 2023Circumventing Concept Erasure Methods For Text-to-Image Generative Modelscited by 85
  8. 2025Set You Straight: Auto-Steering Denoising Trajectories to Sidestep Unwanted Conceptscited by 62
  9. 2025T2I-RiskyPrompt: A Benchmark for Safety Evaluation, Attack, and Defense on Text-to-Image Modelcited by 17
  10. 2025EraseFlow: Learning Concept Erasure Policies via GFlowNet-Driven Alignmentcited by 9

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 2.

2 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2026EraseAnything++: Enabling Concept Erasure in Rectified Flow Transformers Leveraging Multi-Object Optimization
  2. 2026Differential Vector Erasure: Unified Training-Free Concept Erasure for Flow Matching Models
  3. 2026FlowErase-RL: Rethinking Concept Erasure as Reward Optimization in Flow Matching Models
  4. 2026FlowErase-OPD: Multi-Concept Erasure via Anchored On-Policy Distillation in Flow Matching Models
  5. 2026Closed-Form Concept Erasure via Double Projections
  6. 2025Erased, But Not Forgotten: Erased Rectified Flow Transformers Still Remain Unsafe Under Concept Attack
  7. 2026VisualErase: Dual-Branch Visual Trajectory Redirection for Robust Concept Erasure in Text-to-Image Diffusion Models
  8. 2025RecTok: Reconstruction Distillation along Rectified Flow