Watermark Basins: Black-Box Detection of Distilled LLMs
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.
Authors
- Valeriu Lacatusu
- Alexandre Mourachko
- Tom Sander (ORCID: https://orcid.org/0009-0008-3051-5976)
- Sylvestre-Alvise Rebuffi (ORCID: https://orcid.org/0000-0003-2448-2078)
- Hady Elsahar (ORCID: https://orcid.org/0000-0001-9508-6721)
- Saeed Mahloujifar (ORCID: https://orcid.org/0000-0001-6586-8378)
- Rashel Moritz
- Pierre Fernandez
- Hongyan Chang (ORCID: https://orcid.org/0000-0002-0569-0173)
- Tomáš Souček (ORCID: https://orcid.org/0000-0001-6911-5517)
- Tuan Tran
- Vanessa Stark
- Surya Parimi
Publication Details
- Journal
- arXiv (Cornell University)
- Published
- 2026-10-27
- DOI
- https://doi.org/10.48550/arxiv.2610.37266
- Primary Topic
- Adversarial Robustness in Machine Learning
- Type
- preprint