Thanks for your careful review (part 1/2)
Thank you for your valuable suggestions. We believe your concerns can be addressed through clarification and additional details in our responses below.
### About Symbol reuse.
> In Section 3 and Section 4, the symbol *P* represents both the model distribution in *P_M* and the prompt in *P₁, P₂, ..., P_N*, which can be confusing. Too many *P* across line 222 to line 227.
Thank you for your careful reading. We agree that the symbol reuse was confusing. In the new version, we use x to represent prompts throughout the paper, including section 4 and Appendix A and B, to address this concern.
### Line 289, Figure 2. ‘KTH’ does not refer to anything else in other contexts. Is it Exp-Edit or ITS-Edit?
Sorry, this refers to the exp-edit method. We mistakenly used KTH which appears in some papers. We have corrected this in the new version.
### In Table 1, why Water-Probe-V2 has low detection confidence for KGW series watermarks?
This issue mainly relates to the degree of distribution perturbation when KGW method is used with repeated sampling using the same watermark key. Compared to methods like EXP and EXP-Edit, KGW makes smaller modifications to the original distribution:
1. Methods like EXP and EXP-Edit use exponential sampling which significantly biases towards certain tokens, causing larger distribution perturbations.
2. In contrast, KGW only increases probability weights for green tokens moderately, resulting in milder perturbations.
To quantitatively demonstrate this, we designed the following comparative experiment:
Experimental Setup:
- Initial logits: 2000-dimensional vector sampled from normal distribution in range (-100, 100)
- Evaluation metric: JS divergence (measuring distribution difference before and after watermarking)
KGW Experiment:
- Parameters: gamma = 0.5, delta = 2.0
- Repeated 100 times:
1. Add watermark to logits
2. Convert to probability distribution using softmax
3. Calculate JS divergence with original distribution
- Record average JS divergence across 100 trials
AAR Experiment:
- Repeated 100 times:
1. Generate random 2000-dimensional key vector
2. Apply exponential transform and watermark to probability distribution
3. Calculate JS divergence after normalization
4. Average over 1000 repetitions per experiment
- Record mean of 100 average JS divergences
Results:
- KGW average JS divergence: 0.373102
- AAR average JS divergence: 5.067022
These results clearly demonstrate that KGW's distribution perturbation is significantly smaller than AAR method, which explains why Water-Probe-V2 has lower detection confidence for KGW series watermarks.
### Could you provide some experiments on watermark detection on closed-source models such as ChatGPT using your tool? Since the proposed method is a black-box setting, this should be achievable.
Please refer to the first point in the response to all reviewers.
### About the "inversions" in Water-Bag Strategy
Here, our "inversions" refers to certain keys that satisfy the constraints of equation (12). We apologize for not directly providing the calculation method for these keys in the paper. However, this calculation can be derived directly from equation (12) by examining how the Inversion Key modifies the distribution through P_M(y_i | x, y_{1:i-1}) and F(P_M(y_i | x, y_{1:i-1}), f(K_j, y_{i-n:i-1})).
Specifically, let $P_M(y_i | x, y_{1:i-1}) = p$ represent the original model distribution, and $F(p, f(K_j, y_{i-n:i-1})) = q$ represent the distribution after modification using key $K_j$.
According to equation (12), we have:
$\frac{1}{2}(q + F(p, f(\overline{K_j}, y_{i-n:i-1}))) = p$
Through simple algebraic manipulation, we can derive:
$F(p, f(\overline{K_j}, y_{i-n:i-1})) = 2p - q$
This formula provides the concrete method for calculating the Inversion Key $\overline{K_j}$ - for any input sequence $y_{i-n:i-1}$, the function $f(\overline{K_j}, y_{i-n:i-1})$ needs to map the original distribution $p$ to $2p - q$ to satisfy equation (12). This is what we mean by keys that satisfy the constraints.
It's worth noting that an inversion key doesn't necessarily have to be a number - as long as we can accurately calculate its modification to the distribution, it qualifies as a valid key.
We will provide a detailed explanation of how to calculate this inversion key in the appendix M.