水印盆地:蒸馏大语言模型的无白盒检测
Watermark Basins: Black-Box Detection of Distilled LLMs
Hongyan Chang · Hady Elsahar · Tom Sander · Pierre Fernandez · Saeed Mahloujifar · Sylvestre-Alvise Rebuffi · et al.
中文摘要
专有语言模型易受基于 API 的蒸馏攻击。现有无白盒检测器依赖 token 级统计,但在低温度采样下信号常不可检测。研究发现“水印盆地”,即微调时水印教师轨迹在参数空间嵌入的、与任务子空间正交的几何信号。基于此提出 WM-Basin 检测器,通过 API 查询可疑模型,利用固定探针模型将文本映射为梯度,投影去除任务方向后,检测残差梯度是否与水印盆地对齐。在 135M 至 8.2B 参数的数学推理蒸馏模型上评估,0.1% 假阳性率下,WM-Basin 在 n-gram=1 时检测率为 79%,n-gram=3 时为 45%,显著优于基线(29% 和 1.3%)。该信号在量化、剪枝及强化学习后训练中依然保留。
关键要点
- 01现有 token 级统计检测器在低温度采样下无法有效检测蒸馏痕迹,存在明显局限。
- 02发现水印盆地,即微调过程中嵌入参数空间且与任务子空间正交的持久几何信号。
- 03提出 WM-Basin 无白盒检测器,通过 API 查询并利用固定探针模型分析梯度对齐情况。
- 04在 0.1% 假阳性率下,n-gram=1 时检测率达 79%,显著优于现有 token 级基线方法。
- 05水印盆地信号具有鲁棒性,在量化、剪枝及强化学习后训练等修改后依然保留。
解读
尚无解读。
原始英文摘要
arXiv:2610.37266v1 Announce Type: cross Abstract: Proprietary language models and their reasoning traces are vulnerable to API-based distillation, where an adversary trains a student model on the teacher model's outputs. To obtain evidence of distillation, model providers can watermark the teacher's outputs, aiming to leave a radioactive signal in the student's outputs. Existing black-box detectors search for such radioactivity using token-level statistics of the suspect model's outputs. However, token-level signals are not always detectable, for instance when the distillation traces are sampled at a low temperature. In this work, we discover the watermark basin, a persistent signal that enables more reliable watermark detection: during fine-tuning, watermarked teacher traces embed a distinct, key-specific geometric signal in parameter space that lies orthogonal to the ordinary task-specific fine-tuning subspace. Building on this, we introduce Watermark Basin Alignment (WM-Basin), a black-box watermark detector that queries the suspect model through its API, without access to its weights or logits. WM-Basin maps generated text to gradient representations using a fixed clean probe model, projects out the ordinary fine-tuning task directions, and tests whether the residual gradient aligns with the watermark basin induced by the watermark key. We evaluate WM-Basin extensively across diverse teacher models and student architectures (from 135M to 8.2B parameters) distilled on mathematical reasoning. At a strict 0.1% false-positive rate, WM-Basin detects 79% of distilled models at n-gram=1 and 45% at n-gram=3, substantially improving over the state-of-the-art token-level baseline (29% and 1.3%, respectively). Because the watermark signal resides in a persistent basin, it survives post-distillation modifications including quantization, pruning, and reinforcement-learning post-training.