DREAM: Dynamic Red-teaming across Environments for AI Models
Liming Lu, Xiang Gu, Junyu Huang, Jiawei Du, Yunhuai Liu, Yongbin Zhou, Shuchao Pang
TL;DR
DREAM offers an automated, cross-environment red-teaming framework that unifies a Cross‑Environment Adversarial Knowledge Graph (CE‑AKG) with Contextualized Guided Policy Search (C‑GPS) to construct long, causally-linked attack chains against LLM-powered agents. The framework exposes systemic vulnerabilities—particularly contextual fragility and the inability to track malicious intent over multi-step interactions—undetected by static, single-environment evaluations. Through a large-scale evaluation of 12 agents across 349 environments and 1,986 atom attacks, DREAM demonstrates a domino effect where attack efficacy grows super-linearly with chain length and cross-environment pivots. Static defenses prove ineffective against such dynamic, stateful threats, underscoring the need for sophisticated, context-aware safety strategies and providing a reproducible benchmark for advancing agent safety research.
Abstract
Large Language Models (LLMs) are increasingly used in agentic systems, where their interactions with diverse tools and environments create complex, multi-stage safety challenges. However, existing benchmarks mostly rely on static, single-turn assessments that miss vulnerabilities from adaptive, long-chain attacks. To fill this gap, we introduce DREAM, a framework for systematic evaluation of LLM agents against dynamic, multi-stage attacks. At its core, DREAM uses a Cross-Environment Adversarial Knowledge Graph (CE-AKG) to maintain stateful, cross-domain understanding of vulnerabilities. This graph guides a Contextualized Guided Policy Search (C-GPS) algorithm that dynamically constructs attack chains from a knowledge base of 1,986 atomic actions across 349 distinct digital environments. Our evaluation of 12 leading LLM agents reveals a critical vulnerability: these attack chains succeed in over 70% of cases for most models, showing the power of stateful, cross-environment exploits. Through analysis of these failures, we identify two key weaknesses in current agents: contextual fragility, where safety behaviors fail to transfer across environments, and an inability to track long-term malicious intent. Our findings also show that traditional safety measures, such as initial defense prompts, are largely ineffective against attacks that build context over multiple interactions. To advance agent safety research, we release DREAM as a tool for evaluating vulnerabilities and developing more robust defenses.
