English

AgentMark: Utility-Preserving Behavioral Watermarking for Agents

Cryptography and Security 2026-04-27 v2 Artificial Intelligence

Abstract

LLM-based agents are increasingly deployed to autonomously solve complex tasks, raising urgent needs for IP protection and regulatory provenance. While content watermarking effectively attributes LLM-generated outputs, it fails to directly identify the high-level planning behaviors (e.g., tool and subgoal choices) that govern multi-step execution. Critically, watermarking at the planning-behavior layer faces unique challenges: minor distributional deviations in decision-making can compound during long-term agent operation, degrading utility, and many agents operate as black boxes that are difficult to intervene in directly. To bridge this gap, we propose AgentMark, a behavioral watermarking framework that embeds multi-bit identifiers into planning decisions while preserving utility. It operates by eliciting an explicit behavior distribution from the agent and applying distribution-preserving conditional sampling, enabling deployment under black-box APIs while remaining compatible with action-layer content watermarking. Experiments across embodied, tool-use, and social environments demonstrate practical multi-bit capacity, robust recovery from partial logs, and utility preservation. The code is available at https://github.com/Tooooa/AgentMark.

Keywords

Cite

@article{arxiv.2601.03294,
  title  = {AgentMark: Utility-Preserving Behavioral Watermarking for Agents},
  author = {Kaibo Huang and Jin Tan and Yukun Wei and Wanling Li and Zipei Zhang and Hui Tian and Zhongliang Yang and Linna Zhou},
  journal= {arXiv preprint arXiv:2601.03294},
  year   = {2026}
}

Comments

Accepted to ACL 2026 (Main, Poster)

R2 v1 2026-07-01T08:53:11.866Z