中文

谁写了这段代码?面向代码生成的水印技术

计算与语言 2024-07-04 v4

摘要

由于大语言模型卓越的生成性能引发了伦理与法律担忧,通过嵌入水印来检测机器生成文本的方法正在被开发。然而,我们发现现有工作在代码生成任务中因该任务低熵的特性而无法正常发挥作用。扩展一种修改logit的水印方法,我们提出基于熵阈值的选择性水印(Selective WatErmarking via Entropy Thresholding, SWEET),其通过在生成与检测水印时移除低熵片段来增强检测能力并缓解代码质量退化。我们的实验表明,SWEET在显著改善代码质量保持的同时,在检测机器生成代码文本方面优于所有基线,包括事后检测方法。我们的代码可在 https://github.com/hongcheki/sweet-watermark 获取。

关键词

引用

@article{arxiv.2305.15060,
  title  = {Who Wrote this Code? Watermarking for Code Generation},
  author = {Taehyun Lee and Seokhee Hong and Jaewoo Ahn and Ilgee Hong and Hwaran Lee and Sangdoo Yun and Jamin Shin and Gunhee Kim},
  journal= {arXiv preprint arXiv:2305.15060},
  year   = {2024}
}

备注

To be presented at ACL 2024