arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2602.01428cs.LGcs.CR
收藏

提升语言模型中水印强度与推测采样效率之间的权衡

Improving the Trade-off Between Watermark Strength and Speculative Sampling Efficiency for Language Models

University of Pennsylvania(宾夕法尼亚大学)

Weiqing He, Xiang Li, Li Shen, Weijie Su, Qi Long

AI总结:

本文提出了一种提升语言模型中水印强度与推测采样效率之间权衡的方法,通过定量指标和系统机制实现两者的平衡。

摘要:

水印技术是一种系统性的方法,用于追踪大型语言模型(LLM)输出的来源,但其在实际部署中受到推断效率低的阻碍。推测采样可以加速推断,其效率随着草稿模型与目标模型之间的接受率提高而提高。然而,近期的研究揭示了一个根本性的权衡:较高的水印强度会降低接受率,从而防止同时实现两者。我们重新审视这一权衡,并证明它并非绝对。我们引入了一个衡量水印强度的定量指标,该指标控制统计可检测性,并在令牌是伪随机数的确定性函数时最大化。使用这一指标,我们将权衡完全表征为一个约束优化问题,并为两种现有的水印技术方案推导出显式的帕累托曲线。最后,我们引入了一个系统性的机制,将伪随机性注入到草稿令牌接受中,确保在保持推测采样效率的同时实现最大水印强度。实验进一步表明,这种方法在提高可检测性的同时不牺牲效率。我们的发现揭示了一个将推测采样和水印技术联系起来的原则,为它们的高效和实际部署铺平了道路。

英文摘要:

Watermarking is a principled approach for tracing the provenance of large language model (LLM) outputs, but its deployment in practice is hindered by inference inefficiency. Speculative sampling accelerates inference, with efficiency improving as the acceptance rate between draft and target models increases. Yet recent work reveals a fundamental trade-off: higher watermark strength reduces acceptance, preventing their simultaneous achievement. We revisit this trade-off and show it is not absolute. We introduce a quantitative measure of watermark strength that governs statistical detectability and is maximized when tokens are deterministic functions of pseudorandom numbers. Using this measure, we fully characterize the trade-off as a constrained optimization problem and derive explicit Pareto curves for two existing watermarking schemes. Finally, we introduce a principled mechanism that injects pseudorandomness into draft-token acceptance, ensuring maximal watermark strength while maintaining speculative sampling efficiency. Experiments further show that this approach improves detectability without sacrificing efficiency. Our findings uncover a principle that unites speculative sampling and watermarking, paving the way for their efficient and practical deployment.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑