BadWAM: When World-Action Models Dream Right but Act Wrong
arXiv · HuggingFace · ▲47
Abstract (verbatim)
World-action models (WAMs) are emerging as a promising foundation for embodied control: rather than predicting actions alone, they learn representations that couple action generation with future world prediction. This coupling is often viewed as a source of robustness, interpretability, and safety, as a robot's action can in principle be checked against its imagined future. In this paper, we show that this assumption is fragile. We introduce BadWAM, a unified framework for modeling and evaluating World-Action Drift Attacks: a new class of WAM-specific adversarial attacks that use small visual perturbations to break the alignment between what a WAM imagines and what it executes. BadWAM characterizes this attack surface along two natural criteria: attack strength and stealthiness. When the adversary prioritizes disruption, BadWAM instantiates an action-only adversarial attack, which directly drives the model toward task-failing actions. When the adversary additionally prioritizes stealth, BadWAM instantiates an imagination-preserving adversarial attack, which seeks to induce harmful action shifts while keeping the model's predicted future close to its clean imagination. Together, these two attacks capture a spectrum of WAM-specific failures: from overt action hijacking to stealthier cases where the model appears to imagine a plausible future but executes a desynchronized action. We evaluate BadWAM across different variants of WAMs. Results show that our attacks substantially reduce task success rates under closed-loop execution. For example, our action-only attack reduces the model performance from 96.5% to 43.1% success. The results of our imagination-preserving attack further exposes a WAM-specific vulnerability: moderate future-preserving regularization can maintain strong attack performance while reducing future imagination drift.
Background
Background Analysis
1. Technical Context
Robots need to interact intelligently with the environment—for example, organizing items at home, moving goods in warehouses, or performing tasks in hazardous scenarios. Traditional methods let robots "only predict actions" (e.g., directly outputting the next move), but this approach is error-prone because robots cannot foresee long-term consequences. World-Action Models (WAMs) address this by learning not just "how to act" but also "what happens next." For instance, a robot can imagine, "If I move this box, will the shelf collapse?" This ability makes robots safer and more reliable by allowing them to detect dangers early.
2. Previous Limitations
While WAMs sound ideal, they have a critical flaw: actions and imagination can desynchronize. For example, a robot might imagine successfully completing a task (e.g., "placing a cup on the table") but execute an action that fails (e.g., "the cup drops"). Worse, this "disconnect" is hard to detect—because the robot’s imagination still looks plausible. Prior attack research focused on image classification or simple robot policies but did not consider WAMs’ unique "action-imagination coupling" problem.
3. Our Solution
This paper introduces the BadWAM framework to test WAM vulnerabilities. It evaluates two attack methods:
- Direct action sabotage: Small visual perturbations (e.g., modifying a pixel in an image) trick the robot into making mistakes, like "missing the grasp" or "knocking over objects."
- Stealthy attacks: The robot’s imagination appears normal (e.g., "the cup is still on the table"), but the actual action fails (e.g., "the arm suddenly releases the object").
Experiments show these attacks significantly reduce task success rates (e.g., from 96.5% to 43.1%), proving WAMs’ "safety signal" (imagined future) is unreliable.
4. Unique Angle
Unlike prior work, BadWAM does not focus on "making models more accurate" but asks: "When do models deceive themselves?" It reveals a key issue: WAMs’ actions and imagination are coupled but can desynchronize. Attackers can exploit this, breaking actions without damaging imagination. This perspective guides future safety research—optimizing prediction alone is insufficient; actions and imagination must truly align.
















