arXiv · 2310.07710
A Resilient and Accessible Distribution-Preserving Watermark for Large Language Models
Abstract
Watermarking techniques offer a promising way to identify machine-generated content via embedding covert information into the contents generated from language models. A challenge in the domain lies in preserving the distribution of original generated content after watermarking. Our research extends and improves upon existing watermarking framework, placing emphasis on the importance of a \textbf{Di}stribution-\textbf{P}reserving (DiP) watermark. Contrary to the current strategies, our proposed DiPmark simultaneously preserves the original token distribution during watermarking (distribution-preserving), is detectable without access to the language model API and prompts (accessible), and is provably robust to moderate changes of tokens (resilient). DiPmark operates by selecting a random set of tokens prior to the generation of a word, then modifying the token distribution through a distribution-preserving reweight function to enhance the probability of these selected tokens during the sampling process. Extensive empirical evaluation on various language models and tasks demonstrates our approach's distribution-preserving property, accessibility, and resilience, making it a effective solution for watermarking tasks that demand impeccable quality preservation.
Explore related subjects
Keep this discovery
Yihan Wu, Zhengmian Hu, Junfeng Guo, Hongyang Zhang, Heng Huang. 2023-10-11. A Resilient and Accessible Distribution-Preserving Watermark for Large Language Models. https://arxiv.org/abs/2310.07710
Cite the original work for its findings. Save a collection to share your selection of sources.