Optimal Detection for Language Watermarks with Pseudorandom Collision
T. Tony Cai 1, Xiang Li 1, Qi Long 1, Weijie Su 1, Garrett G. Wen 2
Published on arXiv
2510.22007
Output Integrity Attack
OWASP ML Top 10 — ML09
Key Finding
Closed-form optimal detection rules improve detection power while maintaining rigorous Type I error control under pseudorandom collision-induced dependence in LLM outputs.
Minimal-unit minimax watermark detection framework
Novel technique introduced
Text watermarking plays a crucial role in ensuring the traceability and accountability of large language model (LLM) outputs and mitigating misuse. While promising, most existing methods assume perfect pseudorandomness. In practice, repetition in generated text induces collisions that create structured dependence, compromising Type I error control and invalidating standard analyses. We introduce a statistical framework that captures this structure through a hierarchical two-layer partition. At its core is the concept of minimal units -- the smallest groups treatable as independent across units while permitting dependence within. Using minimal units, we define a non-asymptotic efficiency measure and cast watermark detection as a minimax hypothesis testing problem. Applied to Gumbel-max and inverse-transform watermarks, our framework produces closed-form optimal rules. It explains why discarding repeated statistics often improves performance and shows that within-unit dependence must be addressed unless degenerate. Both theory and experiments confirm improved detection power with rigorous Type I error control. These results provide the first principled foundation for watermark detection under imperfect pseudorandomness, offering both theoretical insight and practical guidance for reliable tracing of model outputs.
Key Contributions
- Introduces a hierarchical two-layer partition framework with 'minimal units' to model structured dependence caused by pseudorandom collisions in repeated text
- Casts watermark detection as a minimax hypothesis testing problem and derives closed-form optimal detection rules for Gumbel-max and inverse-transform watermarks
- Provides the first principled statistical foundation for watermark detection under imperfect pseudorandomness, with both theoretical guarantees and improved empirical detection power
🛡️ Threat Analysis
Addresses detection of content watermarks embedded in LLM-generated text outputs for traceability and provenance — this is directly about verifying output integrity and authenticating AI-generated content. The paper improves watermark detection reliability rather than protecting model IP, so ML09 applies (not ML05).