Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Unbiased Watermark for Large Language Models

About

The recent advancements in large language models (LLMs) have sparked a growing apprehension regarding the potential misuse. One approach to mitigating this risk is to incorporate watermarking techniques into LLMs, allowing for the tracking and attribution of model outputs. This study examines a crucial aspect of watermarking: how significantly watermarks impact the quality of model-generated outputs. Previous studies have suggested a trade-off between watermark strength and output quality. However, our research demonstrates that it is possible to integrate watermarks without affecting the output probability distribution with appropriate implementation. We refer to this type of watermark as an unbiased watermark. This has significant implications for the use of LLMs, as it becomes impossible for users to discern whether a service provider has incorporated watermarks or not. Furthermore, the presence of watermarks does not compromise the performance of the model in downstream tasks, ensuring that the overall utility of the language model is preserved. Our findings contribute to the ongoing discussion around responsible AI development, suggesting that unbiased watermarks can serve as an effective means of tracking and attributing model outputs without sacrificing output quality.

Zhengmian Hu, Lichang Chen, Xidong Wu, Yihan Wu, Hongyang Zhang, Heng Huang• 2023

Related benchmarks

TaskDatasetResultRank
Language ModelingC4
Perplexity15.62
1182
Mathematical ReasoningGSM8K (test)
Accuracy17.89
751
Question AnsweringTruthfulQA
Truthful*Inf Score62.54
42
Text WatermarkingC4
PPL10.3701
27
Watermark DetectionC4 subset--
24
Spoofing attack traceabilityRealToxicityPrompts (test)
AUC54.62
20
Spoofing Attack RobustnessBookSum
AUC0.4914
20
Spoofing Attack RobustnessC4 RealNewsLike
AUC0.5021
20
Spoofing attack traceabilityRTP-LX (test)
AUC49.97
20
Paraphrase Attack RobustnessC4 RealNewsLike
AUC0.5011
20
Showing 10 of 27 rows

Other info

Follow for update