Skip to content

Back to the catalog

Semantic Behavioral Watermarking: Paraphrase-Robust and Forgery-Resistant Provenance for LLM Agents
arxiv.orgpaper

Semantic Behavioral Watermarking: Paraphrase-Robust and Forgery-Resistant Provenance for LLM Agents

Semantic Behavioral Watermarking (SBW) is a provenance scheme for LLM agents that embeds an owner identifier in semantic clusters of an agent's action choices rather than exact action symbols, making it robust to paraphrasing/tool-renaming that breaks three prior agent watermarks (e.g., AgentMark drops to 16.8% bit-recovery under observation paraphrasing). It adds keyed, collision-resistant binning proven unpredictable in the random-oracle model to resist forgery (adaptive forgery success drops from 100% to the false-positive floor), while explicitly documenting an unsolved attack: chained-replay forgery, which still succeeds 0.76-0.98 of the time across five tested agent models (3B-14B, Qwen, ToolBench/ALFWorld benchmarks). Code is released alongside the paper.

TL;DR A forgery-resistant, paraphrase-robust watermarking scheme for LLM agent action trajectories that also names and leaves open a chained-replay forgery attack.
Also seen from another angle🧠 AI →
Document

What it is

Semantic Behavioral Watermarking (SBW), from Ji, Wan, Chen, and Zhang (ICLR 2027 submission), is a training-free watermark for LLM agent trajectories: a frozen sentence encoder (bge-base) clusters tool/action embeddings via k-means, and a keyed PRF marks "green" steps over cluster symbols plus a history window (SeqWM-style), rather than exact action strings.

How it works

Binning is keyed: bin(a) = H(sk, quantize(R_sk · E(a))), where R_sk is a secret random projection to r hyperplanes — orthogonal to the public embedding geometry, so an adversary with only the public encoder can't recover which clusters are green. During generation, with green bias probability p=0.95, a step is resampled from green candidates (support-restricted, not distribution-preserving). Detection is a permutation test (M=300 wrong keys) scoring green windows over the observed cluster-symbol stream. History window length w is a tunable tradeoff: w=1 is used by default because it favors deletion robustness over clean detection (clean detection 0.83/0.91/0.98 for w=1/2/3, but deletion-robustness TPR at 30% deletion drops to 0.48/0.31/0.24 as w grows). Boundary actions near a cluster margin are routed to an abstain state rather than decoded, trading capacity for near-zero bit error.

Preconditions / Affected

Applies to agent architectures where ownership must be proven from discrete tool-call choices, not output tokens. Tested on 5 models (3B–14B: Qwen2.5-3B/7B, LLaMA-3.2-3B, Mistral-7B, phi-4) across ToolBench (600 trajectories) and ALFWorld (100 episodes). Requires the owner to control generation-time decoding and a fixed elicitation operator — the paper notes capacity is undefined without fixing this (two elicitation methods correlated at only 0.13 in per-step entropy).

What to do

Key quantitative thresholds from the paper for anyone evaluating or reproducing this:

  • Detection under rewriting (TPR @ 1% FPR): cluster-level 0.49–0.66 vs exact-symbol 0.05–0.17 on ToolBench; 0.92–0.97 vs 0.00–0.01 on ALFWorld.
  • Utility cost: choice agreement 72–83% (SBW) vs 22–27% (KGW-style logit biasing) against no-watermark argmax.
  • Capacity cost: clustering at k=100 retains only 41.9–50.1% of tool-level entropy — the explicit price of paraphrase robustness.
  • Forgery resistance: keyed binning drops adaptive forgery from 100% to ~21% at r=8, and to 1.2% (the FPR floor) at r=64 — raising projection dimension r is the concrete lever.
  • Payload sizing: recovering a 128-bit ID needs T≥422 steps at their calibrated capacity/error rates — relevant for deciding episode/corpus aggregation granularity.
  • Code: anonymous.4open.science/r/SBW-Agent-Watermark.

Caveats

The paper explicitly flags chained-replay forgery (adversary replays the victim's own steps in sequence) as unsolved: it succeeds 0.76–0.98 of the time across all five models, versus shuffled-splicing replay which SBW neutralizes (0.000 on Qwen2.5-3B). The fresh-bucket forgery guarantee (Thm. G.1) is provable only when the adversary's actions land in unobserved keyed buckets — not when buckets are already observed (~98% of fresh actions at r=8, per their own measurement). Mistral-7B was pre-registered as a fourth-vendor validation model and returned the highest score of the five, which the authors flag as a reason to weight that result carefully rather than cherry-pick it.

Organisations named here

  • chained-replay forgery
  • shuffled splicing

Technologies named here

  • Semantic Behavioral Watermarking
  • LLM agents
  • AgentMark
  • ToolBench
  • ALFWorld
  • Qwen
  • LLM

Techniques named here

  • Behavioral watermarking
  • Exact-symbol watermark binding
  • Cluster-level semantic watermarking
  • Keyed collision-resistant binning
  • Watermark forgery attack
  • Shuffled splicing attack
  • Chained replay attack
  • Logit biasing
  • Paraphrasing/rewriting attack