Skip to results
MLSift
← Feed
routineAI Safety, Security & AlignmentWatermarking2607.00325

Watermarking for Proprietary Dataset Protection

John Kirchenbauer, Brian R. Bartoldson, Bhavya Kailkhura, Tom Goldstein

cs.LG cs.CL

Abstract

A growing body of literature suggests that training data membership inference problems are fundamentally hard tasks in modern language modeling settings. We argue that output watermarking techniques are the right gadget to make training membership tests for generative models more tractable, based on prior results showing that language models exhibit residual watermark "radioactivity" under partially watermarked training datasets. We pit a watermark-based dataset inference approach head-to-head against traditional loss-based membership inference methods and show that watermarking can achieve comparable membership detection performance when subset exposure is high enough, under an alternate set of assumptions.

Topics

Classified with taxonomy v2 on Wed, 2 Sept 2026.

The PDF is 1–3 MB. Open it in your browser's viewer, or load it here.

Open PDF