arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2407.04411cs.CRcs.AIcs.CL
收藏

Waterfall:面向大语言模型的鲁棒可扩展文本水印与溯源框架

Waterfall: Framework for Robust and Scalable Text Watermarking and Provenance for LLMs

National University of Singapore(新加坡国立大学);Centre for Frontier AI Research(前沿人工智能研究中心);Institute for Infocomm Research, A*STAR(新加坡资讯通信研究院(新加坡科技研究局下属))

Gregory Kang Ruey Lau, Xinyuan Niu, Hieu Dao, Jiangwei Chen, Chuan-Sheng Foo, Bryan Kian Hsiang Low

AI总结:

针对现有文本水印方法鲁棒性不足、扩展性差的问题,本文提出无需训练的Waterfall框架,利用LLM改写及多技术组合实现多类型多语言文本水印与LLM数据溯源,性能优于现有最优方法。

摘要:

保护文章、代码等文本的知识产权(IP)正变得愈发重要,尤其是当前复杂攻击手段不断涌现,比如大语言模型(LLM)生成的改写攻击,甚至是利用受版权保护的文本未经授权训练LLM以侵犯知识产权的行为。然而,现有文本水印方法既无法足够鲁棒地抵御这类攻击,也不具备面向数百万用户实际落地的可扩展性。本文提出Waterfall,这是首个无需训练的鲁棒可扩展文本水印框架,适用于LLM支持的多种文本类型(如文章、代码)和语言,可用于通用文本及LLM数据溯源。Waterfall包含多项关键创新:它是首个将LLM用作水印生成阶段改写器的框架,还创新性地结合了多种技术,这些技术在实现鲁棒可验证性与可扩展性方面效果出人意料地出色。我们通过实验证明,与当前最优(SOTA)的文章文本水印方法相比,Waterfall在可扩展性、鲁棒可验证性和计算效率上均表现显著更优,同时展示了其可直接应用于代码水印的能力。我们还证明Waterfall可用于LLM数据溯源:LLM训练数据的水印可在LLM输出中被检测到,从而能够检测数据被未经授权用于LLM训练的行为,还有望实现开源LLM的以模型为中心的水印,这一点一直是现有LLM水印研究的局限所在。我们的代码可在https://github.com/aoi3142/Waterfall获取。

英文摘要:

Protecting intellectual property (IP) of text such as articles and code is increasingly important, especially as sophisticated attacks become possible, such as paraphrasing by large language models (LLMs) or even unauthorized training of LLMs on copyrighted text to infringe such IP. However, existing text watermarking methods are not robust enough against such attacks nor scalable to millions of users for practical implementation. In this paper, we propose Waterfall, the first training-free framework for robust and scalable text watermarking applicable across multiple text types (e.g., articles, code) and languages supportable by LLMs, for general text and LLM data provenance. Waterfall comprises several key innovations, such as being the first to use LLM as paraphrasers for watermarking along with a novel combination of techniques that are surprisingly effective in achieving robust verifiability and scalability. We empirically demonstrate that Waterfall achieves significantly better scalability, robust verifiability, and computational efficiency compared to SOTA article-text watermarking methods, and showed how it could be directly applied to the watermarking of code. We also demonstrated that Waterfall can be used for LLM data provenance, where the watermarks of LLM training data can be detected in LLM output, allowing for detection of unauthorized use of data for LLM training and potentially enabling model-centric watermarking of open-sourced LLMs which has been a limitation of existing LLM watermarking works. Our code is available at https://github.com/aoi3142/Waterfall.

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

↑