REVIEW 3 major objections 5 minor 1 cited by
Watermarking for diffusion language models can be embedded by steering the order in which tokens are unmasked, not by biasing token probabilities.
Reviewed by Pith at T0; open to challenge. T0 means a machine referee read the full paper against a public rubric. the ladder, T0–T4 →
T0 review · deepseek-v4-flash
2026-08-03 06:17 UTC pith:H2BKSILD
load-bearing objection Decoding-order watermarking for dLLMs is a real new idea, and the experiments are broad; but the reported FPR=0.000 rests on a token-ID parity hash that does not justify the theoretical null. the 3 major comments →
dgMARK: Decoding-Guided Watermarking for Diffusion Language Models
The pith
A machine-rendered reading of the paper's core claim, the machinery that carries it, and where it could break.
Core claim
The paper claims that the decoding order of a dLLM can serve as a watermark channel. In dgMARK, at each step the decoder restricts attention to unrevealed positions whose sampled candidate token falls in the parity-matching set G_j = {v : f(v, ξ) ≡ j mod 2}, where f is a balanced binary hash keyed by a secret ξ, and selects the highest-reward position among those, falling back to the full set when no match exists. This simple rule systematically elevates the parity-matching rate of generated tokens, so detection reduces to a z-test on whether the observed rate significantly exceeds 0.5. The paper also introduces a one-step lookahead beam variant that preserves future parity-alignment opportu
What carries the argument
The central mechanism is the binary parity hash f: V × Ξ → {0,1} that partitions the vocabulary into parity-matching sets G_i, combined with a decoding algorithm that prioritizes positions whose candidate token lies in G_i. The hash is the carrier of the watermark signal: its balance gives the binomial null (Parity matches ~ Binomial(n, 1/2)), and its secrecy prevents easy spoofing. The one-step lookahead beam search computes, for each top candidate, how many future positions would remain parity-matching after committing, making the embedding stronger while adding computation.
Load-bearing premise
The scheme bets that real dLLMs are sensitive enough to the unmasking order that steering which position is revealed next can shift token statistics without visibly harming text quality, and that the parity hash is balanced and secret enough for the binomial null to hold.
What would settle it
Run dgMARK on a dLLM trained to be fully order-invariant (e.g., by averaging over all unmasking permutations during training); if the parity-matching rate no longer rises above 0.5, the watermark channel disappears. Alternatively, generate text with the token-ID modulo 2 hash but without a secret key and compute the empirical z-score distribution on many sequences: if it deviates from a standard normal, the detection null is mis-specified and the false-positive claims do not hold as stated.
If this is right
- Decoding order is a practical watermark channel for dLLMs, complementary to probability-biasing methods, and can be combined with common decoding strategies such as confidence, entropy, and margin-based ordering.
- Watermark detection only needs the generated text and the secret key, not the model's internals or the prompt, making it suitable for provenance verification in deployed systems.
- The one-step lookahead variant provides a tunable trade-off between detectability, text quality, and computational cost, with beam sizes of 3-8 yielding near-perfect detection at low false-positive rates.
- The sliding-window detector extends the watermark's robustness to post-editing, including random insertions, deletions, substitutions, and paraphrasing, so it can survive common text transformations.
- Longer generations (around 200 tokens or more) are particularly amenable to dgMARK, suggesting practical use in summaries, reports, and stories, while very short outputs would need aggregation or a less stringent detection threshold.
Where Pith is reading between the lines
- Because dgMARK's signal relies on the gap between ideal order-invariance and practical order-sensitivity, its reliability may weaken as dLLMs become better trained and more order-invariant; the method's long-term viability depends on whether that sensitivity persists in future models.
- The experiments instantiate the hash as token-ID modulo 2, which is not secret and may correlate with token frequencies, so the paper's detection null (parity matches ~ Binomial(n, 1/2)) is a theoretical ideal rather than a verified property of the actual test; using a true cryptographic PRF would close this gap but was not empirically tested.
- dgMARK could likely be extended to carry multi-bit payloads (e.g., by using more than one parity class or by keying the hash to the prompt), enabling richer provenance metadata than a single binary watermark bit.
- An attacker who reorders or rephrases text to restore parity balance might evade the sliding-window detector; since the watermark lives in token positions, adversarial re-sorting attacks are a natural threat model worth testing.
Editorial analysis
A structured set of objections, weighed in public.
Referee Report
Summary. dgMARK is a watermarking method for discrete diffusion language models that embeds a signal by steering the order in which tokens are unmasked, rather than by reweighting token probabilities. A binary hash f(v, ξ) assigns a parity to each vocabulary token; at each decoding step the decoder restricts attention to positions whose sampled candidate token has parity matching the position index, selects the highest-reward such position (falling back to any position if none matches), and reveals that token. Detection counts parity matches and applies a z-test (Eq. 1), with a sliding-window variant for robustness to post-editing. The paper evaluates on LLaDA-8B/1.5/2.0 and Dream-7B over C4 and Writing Prompts, compares to KGW and PATTERN-MARK, and reports high detectability, modest perplexity increase, robustness to token edits and paraphrasing, and computational overhead. The experiments instantiate the hash as token-ID modulo 2 (Appendix D.1), although the design allows keyed PRFs.
Significance. The paper identifies a genuinely new watermark channel for dLLMs — the unmasking order — and provides broad empirical evidence: multiple model families, datasets, sampling strategies, quality benchmarks, robustness attacks, and overhead measurements. The beam-search lookahead variant and the attempt to avoid probability reweighting are useful contributions. The significance is conditional on resolving a calibration gap in the detection statistic: the experiments use a public, non-secret parity map, whereas the detection null and the security argument require a balanced, secret keyed hash. If the authors re-run with a keyed PRF and validate human-text false-positive rates, the contribution would be solid and timely.
major comments (3)
- [§3.5 (Eq. 1), Appendix D.1] Eq. (1) derives the Binomial(n,1/2) null from a balanced and secret f(·,ξ). Appendix D.1 states that all experiments use f(v)=token_ID mod 2, which is public and not necessarily balanced over the natural-token distribution. Any correlation between token-ID parity and token frequency shifts the null mean, so the FPR=0.000/TNR=1.000 values in Tables 1 and 2 are calibrated only on non-watermarked model outputs under this non-secret hash. They do not establish the behavior on human-authored text, and they do not transfer to the keyed-PRF scheme the paper says it supports. This is load-bearing for the provenance claim; please report parity balance of the tokenizer, human-text FPR, and preferably re-run detection with a keyed PRF.
- [§3.5 (robust detection)] The sliding-window statistic zwin=(1/S)Σ z_s^2 is introduced without a null distribution or a threshold calibration. Figures 2–3 report ROC/AUC, but no fixed-FPR operating points are given under insertion, deletion, substitution, or paraphrasing. Because absolute-position parity is disrupted by insertions/deletions, the two-sided aggregation is heuristic, and it is not clear what false-positive rate corresponds to the advertised robustness. Please provide the null/calibration for zwin or report TPR/FPR at the chosen operating point for each attack budget.
- [§3.3, Appendix D.1] The security claim is unverified: the paper argues that a keyed PRF would make the watermark resilient to adversarial reverse-engineering, but no experiments or analysis consider an adversary who knows the token-ID parity map used in the evaluation. Since embedding and detection use exactly the same public statistic, an adversary can forge or strip the watermark. Please state the threat model explicitly (secret vs. public f) and provide at least one adversarial evaluation for the PRF version, e.g., best-effort parity flipping or reordering.
minor comments (5)
- [Abstract] First sentence in the main text reads "We proposedgMARK"; should be "We propose dgMARK".
- [Eq. (1)] The denominator appears as "p n/4"; it should be sqrt(n/4).
- [Tables 1–2] Please clarify in the captions that FPR/TNR are computed on non-watermarked dLLM outputs at z=4.0, not on human-authored text.
- [Tables 9–10] Some cells appear internally inconsistent; e.g., Table 9 with +5-beam at z=4.0 lists TPR=0.987 and FNR=0.127, which do not sum to 1. Please double-check all error-rate columns.
- [Figure 15] The stacked-bar labels are difficult to read; adding a legend or text annotations would improve clarity.
Circularity Check
No significant circularity: the watermark embedder and detector are a designed matched pair, and the paper's empirical evaluations are independent.
full rationale
dgMARK is a watermarking scheme, so its embedder and detector are intentionally matched: Algorithm 2 prioritizes positions whose sampled candidate satisfies a parity condition, and Section 3.5 detects watermarks by counting parity matches. This is a designed signal channel, not a derivation of a prediction from the same input. The paper's actual claims—detectability on LLaDA/Dream, PPL and benchmark quality, and robustness to editing/paraphrasing—are tested against non-watermarked model outputs and external metrics, so the evidence is independent of the construction. The Eq. (1) null relies on a balanced secret hash; the Appendix's use of public token-ID parity is an external-validity limitation (the null may be miscalibrated for human text or adversarially knowable), not a circular reduction. There are no fitted parameters masquerading as predictions, and no load-bearing self-citations. Score 0.
Axiom & Free-Parameter Ledger
free parameters (4)
- Hash mapping f =
token_id mod 2
- Detection threshold z =
4.0
- Sliding-window size w =
8
- Beam size k =
1,3,5,8
axioms (3)
- domain assumption Practical dLLMs are sufficiently order-sensitive that changing the unmasking order changes generated-text statistics.
- domain assumption A balanced hash f implies E[m_i] ≈ 1/2 for non-watermarked text.
- domain assumption The two-sided sliding-window statistic z_win has a stable null distribution that needs no calibration.
read the original abstract
We propose dgMARK, a decoding-guided watermarking method for discrete diffusion language models (dLLMs). Unlike autoregressive models, dLLMs can generate tokens in arbitrary order. While an ideal conditional predictor would be invariant to this order, practical dLLMs exhibit strong sensitivity to the unmasking order, creating a new channel for watermarking. dgMARK steers the unmasking order toward positions whose high-reward candidate tokens satisfy a simple parity constraint induced by a binary hash, without explicitly reweighting the model's learned probabilities. The method is plug-and-play with common decoding strategies (e.g., confidence, entropy, and margin-based ordering) and can be strengthened with a one-step lookahead variant. Watermarks are detected via elevated parity-matching statistics, and a sliding-window detector ensures robustness under post-editing operations including insertion, deletion, substitution, and paraphrasing. Project website: https://dgmark-watermarking.github.io
Figures
Forward citations
Cited by 1 Pith paper
-
Chainwash: Multi-Step Rewriting Attacks on Diffusion Language Model Watermarks
Chained rewrites by open-weight LLMs reduce watermark detection on diffusion LM outputs from 87.9% to 4.86% after five steps across multiple styles and models.
Reference graph
Works this paper leans on
-
[3]
Bie, T., Cao, M., Chen, K., Du, L., Gong, M., Gong, Z., Gu, Y ., Hu, J., Huang, Z., Lan, Z., et al. Llada2. 0: Scaling up diffusion language models to 100b.arXiv preprint arXiv:2512.15745,
-
[6]
Accessed: 2026-01-18
URL https://deepmind.google/models/ gemini-diffusion/. Accessed: 2026-01-18. Fan, A., Lewis, M., and Dauphin, Y . Hierarchical neural story generation. InACL,
2026
-
[7]
Gloaguen, T., Staab, R., Jovanovi ´c, N., and Vechev, M
URL https://zenodo.org/records/ 10256836. Gloaguen, T., Staab, R., Jovanovi ´c, N., and Vechev, M. Watermarking diffusion language models.arXiv preprint arXiv:2509.24368,
-
[8]
The llama 3 herd of models.arXiv preprint arXiv:2407.21783,
Grattafiori, A., Dubey, A., Jauhri, A., Pandey, A., Kadian, A., Al-Dahle, A., Letman, A., Mathur, A., Schelten, A., Vaughan, A., et al. The llama 3 herd of models.arXiv preprint arXiv:2407.21783,
-
[10]
V ., Singh, S., and Ragupathy, S
Kumar, A., Murthy, S. V ., Singh, S., and Ragupathy, S. The ethics of interaction: Mitigating security threats in llms. arXiv preprint arXiv:2401.12273,
-
[11]
Mercury: Ultra-fast language models based on diffusion
Labs, I., Khanna, S., Kharbanda, S., Li, S., Varma, H., Wang, E., Birnbaum, S., Luo, Z., Miraoui, Y ., Palrecha, A., et al. Mercury: Ultra-fast language models based on diffusion. arXiv preprint arXiv:2506.17298,
-
[12]
A survey on diffu- sion language models.arXiv preprint arXiv:2508.10875,
Li, T., Chen, M., Guo, B., and Shen, Z. A survey on diffu- sion language models.arXiv preprint arXiv:2508.10875,
-
[14]
Lr-dwm: Effi- cient watermarking for diffusion language models.arXiv preprint arXiv:2601.12376,
Raban, O., Fetaya, E., and Chechik, G. Lr-dwm: Effi- cient watermarking for diffusion language models.arXiv preprint arXiv:2601.12376,
-
[15]
Raffel, C., Shazeer, N., Roberts, A., Lee, K., Narang, S., Matena, M., Zhou, Y ., Li, W., and Liu, P. J. Exploring the limits of transfer learning with a unified text-to-text transformer.arXiv preprint arXiv:1910.10683,
Pith/arXiv arXiv 1910
-
[16]
Denoising diffusion implicit models
Song, J., Meng, C., and Ermon, S. Denoising diffusion implicit models. InICLR, 2021a. Song, Y ., Sohl-Dickstein, J., Kingma, D. P., Kumar, A., Er- mon, S., and Poole, B. Score-based generative modeling through stochastic differential equations. InICLR, 2021b. Team, G., Kamath, A., Ferret, J., Pathak, S., Vieillard, N., Merhej, R., Perrin, S., Matejovicova...
-
[17]
10 dgMARK: Decoding-Guided Watermarking for Diffusion Language Models Wei, Q., Zhang, Y ., Liu, Z., Liu, D., and Zhang, L. Acceler- ating diffusion large language models with slowfast: The three golden principles.arXiv preprint arXiv:2506.10848,
-
[18]
Wu, J., Yang, S., Zhan, R., Yuan, Y ., Chao, L. S., and Wong, D. F. A survey on LLM-generated text detection: Necessity, methods, and future directions.Computational Linguistics, 2025a. Wu, L., Zhong, L., Qu, W., Li, Y ., Liu, Y ., Zhai, S., Shen, C., and Zhang, J. Dmark: Order-agnostic watermark- ing for diffusion large language models.arXiv preprint arX...
-
[19]
Yu, R., Li, Q., and Wang, X. Discrete diffusion in large lan- guage and multimodal models: A survey.arXiv preprint arXiv:2506.13759,
-
[20]
A survey of large language model agents for ques- tion answering.arXiv preprint arXiv:2503.19213,
Yue, M. A survey of large language model agents for ques- tion answering.arXiv preprint arXiv:2503.19213,
-
[21]
Zhu, F., Wang, R., Nie, S., Zhang, X., Wu, C., Hu, J., Zhou, J., Chen, J., Lin, Y ., Wen, J.-R., and Li, C. Llada 1.5: Variance-reduced preference optimization for large language diffusion models.arXiv preprint arXiv:2505.19223,
-
[24]
Writing Prompts
assumes greedy token selection, we report results using greedy selection. Sampling PPL↓ z= 4.0z= 5.0 FPR TNR TPR FNR FPR TNR TPR FNR dgMARK (Greedy) 4.40 0.0 1.0 0.601 0.399 0.0 1.0 0.282 0.718 + 3-beam 9.17 0.0 1.0 1.000 0.000 0.0 1.0 1.000 0.000 + 5-beam 14.77 0.0 1.0 1.000 0.000 0.0 1.0 1.000 0.000 + 8-beam 17.94 0.0 1.0 1.000 0.000 0.0 1.0 1.000 0.000...
2000
-
[256]
The prompt is adapted from prior work (Kirchenbauer et al., 2024). Llama 3 Prompt As an expert copy-editor, please rewrite the following text in your own voice while ensuring that the final output contains the same information as the original text and has roughly the same length. Please paraphrase all sentences and do not omit any crucial details. Additio...
2024
-
[1999]
Diffusion beats autoregressive in data- constrained settings.arXiv preprint arXiv:2507.15857,
Prabhudesai, M., Wu, M., Zadeh, A., Fragkiadaki, K., and Pathak, D. Diffusion beats autoregressive in data- constrained settings.arXiv preprint arXiv:2507.15857,
-
[2018]
LLM Usage This manuscript made limited use of Large Language Models (LLMs) for language editing only
11 dgMARK: Decoding-Guided Watermarking for Diffusion Language Models A. LLM Usage This manuscript made limited use of Large Language Models (LLMs) for language editing only. Their role was restricted to improving readability—such as grammar, style, and flow—without contributing to the conception of ideas, analyses, or results. All scientific content rema...
2025
-
[2021]
Bagchi, A., Bhimaraju, A., Choraria, M., Alabi, D., and Varshney, L. R. Watermarking discrete diffusion language models.arXiv preprint arXiv:2511.02083,
-
[2023]
Accessed: 2026-01-18
URL https://www.scottaaronson.com/ talks/watermark.ppt. Accessed: 2026-01-18. Arriola, M., Sahoo, S. S., Gokaslan, A., Yang, Z., Qi, Z., Han, J., Chiu, J. T., and Kuleshov, V . Block diffusion: Interpolating between autoregressive and diffusion lan- guage models. InICLR,
2026
-
[2024]
Training verifiers to solve math word problems
Cobbe, K., Kosaraju, V ., Bavarian, M., Chen, M., Jun, H., Kaiser, L., Plappert, M., Tworek, J., Hilton, J., Nakano, R., et al. Training verifiers to solve math word problems. arXiv preprint arXiv:2110.14168,
-
[2025]
Chen, M., Tworek, J., Jun, H., Yuan, Q., Pinto, H. P. D. O., Kaplan, J., Edwards, H., Burda, Y ., Joseph, N., Brockman, G., et al. Evaluating large language models trained on code.arXiv preprint arXiv:2107.03374,
-
[2026]
Targeted phishing campaigns using large scale language models.arXiv preprint arXiv:2301.00665,
Karanjai, R. Targeted phishing campaigns using large scale language models.arXiv preprint arXiv:2301.00665,
discussion (0)
Sign in with ORCID, Apple, or X to comment. Anyone can read and Pith papers without signing in.