REMARK-LLM: A Robust and Efficient Watermarking Framework for Generative Large Language Models

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Zhang, Ruisi, Hussain, Shehzeen Samarah, Neekhara, Paarth, Koushanfar, Farinaz
Format: Preprint
Published: 2023
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866917632309985280
author Zhang, Ruisi
Hussain, Shehzeen Samarah
Neekhara, Paarth
Koushanfar, Farinaz
author_facet Zhang, Ruisi
Hussain, Shehzeen Samarah
Neekhara, Paarth
Koushanfar, Farinaz
contents We present REMARK-LLM, a novel efficient, and robust watermarking framework designed for texts generated by large language models (LLMs). Synthesizing human-like content using LLMs necessitates vast computational resources and extensive datasets, encapsulating critical intellectual property (IP). However, the generated content is prone to malicious exploitation, including spamming and plagiarism. To address the challenges, REMARK-LLM proposes three new components: (i) a learning-based message encoding module to infuse binary signatures into LLM-generated texts; (ii) a reparameterization module to transform the dense distributions from the message encoding to the sparse distribution of the watermarked textual tokens; (iii) a decoding module dedicated for signature extraction; Furthermore, we introduce an optimized beam search algorithm to guarantee the coherence and consistency of the generated content. REMARK-LLM is rigorously trained to encourage the preservation of semantic integrity in watermarked content, while ensuring effective watermark retrieval. Extensive evaluations on multiple unseen datasets highlight REMARK-LLM proficiency and transferability in inserting 2 times more signature bits into the same texts when compared to prior art, all while maintaining semantic integrity. Furthermore, REMARK-LLM exhibits better resilience against a spectrum of watermark detection and removal attacks.
format Preprint
id arxiv_https___arxiv_org_abs_2310_12362
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle REMARK-LLM: A Robust and Efficient Watermarking Framework for Generative Large Language Models
Zhang, Ruisi
Hussain, Shehzeen Samarah
Neekhara, Paarth
Koushanfar, Farinaz
Cryptography and Security
Computation and Language
We present REMARK-LLM, a novel efficient, and robust watermarking framework designed for texts generated by large language models (LLMs). Synthesizing human-like content using LLMs necessitates vast computational resources and extensive datasets, encapsulating critical intellectual property (IP). However, the generated content is prone to malicious exploitation, including spamming and plagiarism. To address the challenges, REMARK-LLM proposes three new components: (i) a learning-based message encoding module to infuse binary signatures into LLM-generated texts; (ii) a reparameterization module to transform the dense distributions from the message encoding to the sparse distribution of the watermarked textual tokens; (iii) a decoding module dedicated for signature extraction; Furthermore, we introduce an optimized beam search algorithm to guarantee the coherence and consistency of the generated content. REMARK-LLM is rigorously trained to encourage the preservation of semantic integrity in watermarked content, while ensuring effective watermark retrieval. Extensive evaluations on multiple unseen datasets highlight REMARK-LLM proficiency and transferability in inserting 2 times more signature bits into the same texts when compared to prior art, all while maintaining semantic integrity. Furthermore, REMARK-LLM exhibits better resilience against a spectrum of watermark detection and removal attacks.
title REMARK-LLM: A Robust and Efficient Watermarking Framework for Generative Large Language Models
topic Cryptography and Security
Computation and Language
url https://arxiv.org/abs/2310.12362