Stress Testing Concept Erasure with Large Language Model Agents
On the map: LLM agents and tool use → General LLM agent frameworks and benchmarks
Abstract (arXiv)
Concept erasure aims to remove semantic concepts from a trained generative model and is increasingly important for responsible AI deployment. However, verifying whether a model has robustly removed targeted concepts remains a critical challenge. Existing evaluation methods are typically pre-defined and static, failing to expose vulnerabilities under diverse natural-language probes and challenging conditions. Moreover, manually designed evaluation strategies can be biased and difficult to scale. We posit that concept erasure evaluation is best formulated as an adaptive hypothesis search, operationalised by agents that iteratively propose, critique, and verify tests to systematically expand coverage of failure modes. To this end, we propose Stress Testing Agents for Concept Erasure (STACE), a framework that autonomously stress-tests concept-erased models using multiple Large Language Model (LLM) agents, by iteratively generating and verifying stress-testing hypotheses grounded by external knowledge. We also introduce a suite of metrics for assessing the performance and efficiency of LLM-agent-powered stress-testing frameworks. Our extensive experiments show that STACE outperforms five LLM-based evaluation baselines on four concept categories. Further analysis across two T2I models, six concept erasure approaches, and various erasure strengths show that STACE is robust for different settings. We also show that STACE can be adapted beyond concept erasure evaluation to other problem domains, such as LLM jailbreaking. Our code is available anonymously.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in General LLM agent frameworks and benchmarks, one of 16 directions in LLM agents and tool use. Builds on 31 of its 77 references that are in the map. First posted on arXiv 20 Jul 2026.
Builds on
31 of 77 references.
Closest ancestors · 1
Same direction, newest first
- 2024Exploring Large Language Model based Intelligent Agents: Definitions, Methods, and Prospects7 Jan
Other references · 30
- 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,380
- 2022ReAct: Synergizing Reasoning and Acting in Language Modelscited by 8,695
- 2023Toolformer: Language Models Can Teach Themselves to Use Toolscited by 4,182
- 2022Locating and Editing Factual Associations in GPTcited by 2,691
- 2022Mass-Editing Memory in a Transformercited by 1,001
- 2022Safe Latent Diffusion: Mitigating Inappropriate Degeneration in Diffusion Modelscited by 513
- 2023Unified Concept Editing in Diffusion Modelscited by 388
- 2023SalUn: Empowering Machine Unlearning via Gradient-based Weight Saliency in Both Image Classification and Generationcited by 386
- 2024MACE: Mass Concept Erasure in Diffusion Modelscited by 307
- 2023Auto-GPT for Online Decision Making: Benchmarks and Additional Opinionscited by 255
- 2023Ring-A-Bell! How Reliable are Concept Removal Methods for Diffusion Models?cited by 250
- 2023To Generate or Not? Safety-Driven Unlearned Diffusion Models Are Still Easy To Generate Unsafe Images ... For Nowcited by 210
18 more
- 2023MMA-Diffusion: MultiModal Attack on Diffusion Modelscited by 194
- 2023Prompting4Debugging: Red-Teaming Text-to-Image Diffusion Models by Finding Problematic Promptscited by 171
- 2023One-Dimensional Adapter to Rule Them All: Concepts, Diffusion Models and Erasing Applicationscited by 169
- 2024Defensive Unlearning with Adversarial Training for Robust Concept Erasure in Diffusion Modelscited by 168
- 2024EraseAnything: Enabling Concept Erasure in Rectified Flow Transformerscited by 95
- 2024T2VSafetyBench: Evaluating the Safety of Text-to-Video Generative Modelscited by 66
- 2024Erasing Undesirable Concepts in Diffusion Models with Adversarial Preservationcited by 50
- 2025AutoRedTeamer: Autonomous Red Teaming with Lifelong Attack Integrationcited by 43
- 2024Jailbreak Attacks and Defenses against Multimodal Generative Models: A Surveycited by 32
- 2024Fuzz-Testing Meets LLM-Based Agents: An Automated and Efficient Framework for Jailbreaking Text-To-Image Generation Modelscited by 27
- 2025Fine-Grained Erasure in Text-to-Image Diffusion-based Foundation Modelscited by 25
- 2025When Are Concepts Erased From Diffusion Models?cited by 16
- 2025Reason2Attack: Jailbreaking Text-to-Image Models via LLM Reasoningcited by 15
- 2024Chain-of-Jailbreak Attack for Image Generation Models via Editing Step by Stepcited by 11
- 2025Unlearning Isn't Invisible: Detecting Unlearning Traces in LLMs from Model Outputscited by 11
- 2025CRCE: Coreference-Retention Concept Erasure in Text-to-Image Diffusion Modelscited by 6
- 2025DyME: Dynamic Multi-Concept Erasure in Diffusion Models with Bi-Level Orthogonal LoRA Adaptationcited by 6
- 2025M-ErasureBench: A Comprehensive Multimodal Evaluation Benchmark for Concept Erasure in Diffusion Modelscited by 2
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2025The Erasure Illusion: Stress-Testing the Generalization of LLM Forgetting Evaluation
- 2026ResearchGym: Evaluating Language Model Agents on Real-World AI Research
- 2026When Stress Becomes Signal: Detecting Antifragility-Compatible Regimes in Multi-Agent LLM Systems
- 2026STALE: Can LLM Agents Know When Their Memories Are No Longer Valid?
- 2026Measuring Stability and Failure Behavior in Language Models Under Structured Perturbations
- 2024MMLU-SR: A Benchmark for Stress-Testing Reasoning Capability of Large Language Models
- 2026Whispers in the Noise: Surrogate-Guided Concept Awakening via a Multi-Agent Framework
- 2026Reproducing and Stress-Testing Two Approaches to LLM Reasoning Reliability: Test-Time Probability Aggregation and Logic-Representation Editing