Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

SAEMark: Steering Personalized Multilingual LLM Watermarks with Sparse Autoencoders

About

Watermarking LLM-generated text is critical for content attribution and misinformation prevention. However, existing methods compromise text quality, require white-box model access and logit manipulation. These limitations exclude API-based models and multilingual scenarios. We propose SAEMark, a general framework for post-hoc multi-bit watermarking that embeds personalized messages solely via inference-time, feature-based rejection sampling without altering model logits or requiring training. Our approach operates on deterministic features extracted from generated text, selecting outputs whose feature statistics align with key-derived targets. This framework naturally generalizes across languages and domains while preserving text quality through sampling LLM outputs instead of modifying. We provide theoretical guarantees relating watermark success probability and compute budget that hold for any suitable feature extractor. Empirically, we demonstrate the framework's effectiveness using Sparse Autoencoders (SAEs), achieving superior detection accuracy and text quality. Experiments across 4 datasets show SAEMark's consistent performance, with 99.7% F1 on English and strong multi-bit detection accuracy. SAEMark establishes a new paradigm for scalable watermarking that works out-of-the-box with closed-source LLMs while enabling content attribution.

Zhuohao Yu, Xingru Jiang, Weizheng Gu, Yidong Wang, Qingsong Wen, Shikun Zhang, Wei Ye• 2025

Related benchmarks

TaskDatasetResultRank
Watermarking Attack RobustnessGemma 9B v2 (test)
TPR11
49
Watermark Robustness AnalysisGemma-2-2B
Post-attack TPR27
49
Distribution-distance evaluationPrompts 100 (evaluation)
Distinct-N (WM)68
14
Semantic similarity analysisGemma-2 within-prompt completions 2B
Cosine Distance0.378
8
Semantic similarity analysisGemma-2 9B within-prompt completions
Cosine Distance0.404
8
Watermark Detection RobustnessGemma-2 2B Pre-trained (PT) (test)
TPR (None)99
7
Watermark Detection RobustnessGemma-2 9B Pre-trained (PT) (test)
TPR (Baseline)13
7
Watermarked text generation and detectionGemma-2 2B Pre-trained
TPR99
7
Watermarked text generation and detectionGemma-2 9B Pre-trained
TPR13
7
Showing 9 of 9 rows

Other info

Follow for update