Watermarking has emerged as a promising technique to track AI-generated content and differentiate it from authentic human creations. While prior work extensively studies watermarking for autoregressive large language models (LLMs) and image diffusion models, it remains comparatively underexplored for discrete diffusion language models (DDLMs), which are becoming popular due to their high inference throughput. In this paper, we introduce one of the first watermarking methods for DDLMs. Our approach applies a distribution-preserving Gumbel-max sampling trick at every diffusion step and seeds the randomness by sequence position to enable reliable detection. We empirically demonstrate reliable detectability on LLaDA, a state-of-the-art DDLM. We also analytically prove that the watermark is distortion-free, with a false detection probability that decays exponentially in the sequence length. A key practical advantage is that our method realizes desired watermarking properties with no expensive hyperparameter tuning, making it straightforward to deploy and scale across models and benchmarks.
翻译:水印技术已成为追踪人工智能生成内容并区分其与真实人类创作的有前景的方法。尽管先前研究已广泛探讨了自回归大语言模型(LLMs)和图像扩散模型的水印嵌入,但对于因高推理吞吐量而日益流行的离散扩散语言模型(DDLMs),相关研究仍相对不足。本文提出了针对DDLMs的首批水印方法之一。我们的方法在每一扩散步骤中应用保分布的Gumbel-max采样技巧,并通过序列位置设定随机种子以实现可靠检测。我们在当前最先进的DDLM模型LLaDA上通过实验验证了该方法的可靠检测性。同时,我们通过理论分析证明该水印具有无失真特性,其误检概率随序列长度呈指数级衰减。本方法的关键实践优势在于:无需昂贵的超参数调优即可实现理想的水印特性,使其能够直接部署并轻松扩展到不同模型与基准测试中。