QuantileMark: A Message-Symmetric Multi-bit Watermark for LLMs

Junlin Zhu, Baizhou Huang, Xiaojun Wan
4/15/2026
cs.CL

Official code: zzzjunlin/QuantileMark (0 stars, Apache-2.0, last commit 2026-04-13).

Abstract

As large language models become standard backends for content generation, practical provenance increasingly requires multi-bit watermarking. In provider-internal deployments, a key requirement is message symmetry: the message itself should not systematically affect either text quality or verification outcomes. Vocabulary-partition watermarks can break message symmetry in low-entropy decoding: some messages are assigned most of the probability mass, while others are forced to use tail tokens. This makes embedding quality and message decoding accuracy message-dependent. We propose QuantileMark, a white-box multi-bit watermark that embeds messages within the continuous cumulative probability interval $[0, 1)$. At each step, QuantileMark partitions this interval into $M$ equal-mass bins and samples strictly from the bin assigned to the target symbol, ensuring a fixed $1/M$ probability budget regardless of context entropy. For detection, the verifier reconstructs the same partition under teacher forcing, computes posteriors over latent bins, and aggregates evidence for verification. We prove message-unbiasedness, a property ensuring that the base distribution is recovered when averaging over messages. This provides a theoretical foundation for generation-side symmetry, while the equal-mass design additionally promotes uniform evidence strength across messages on the detection side. Empirical results on C4 continuation and LFQA show improved multi-bit recovery and detection robustness over strong baselines, with negligible impact on generation quality. Our code is available at GitHub (https://github.com/zzzjunlin/QuantileMark).

View on arXivView PDF

Code Implementations(1)

Apache-2.0

This is the official code release for the paper: "QuantileMark: A Message-Symmetric Multi-bit Watermark for LLMs"

00Shell, PythonApr 13, 20266 months agoApache-2.0

Similar Papers

Cite this paper

@article{zhu2026quantilemark,
  title  = {QuantileMark: A Message-Symmetric Multi-bit Watermark for LLMs},
  author = {Junlin Zhu and Baizhou Huang and Xiaojun Wan},
  year   = {2026},
  eprint = {2604.13786},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2604.13786v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2604.13786.svg)](https://codeofpaper.com/paper/2604.13786)

Discussion