通过训练顺序记忆性,可证明他人是否在使用你的语言模型。
Blackbox Model Provenance via Palimpsestic Membership Inference
- 利用模型对训练数据顺序的记忆特性,检测衍生模型是否源自原模型。
- 在查询场景下,120个不同规模模型的检验均达到极低p值(≤1e-8)。
- 无需重训练即可从少量文本中识别模型来源,适合版权保护场景。
假设爱丽丝训练了一个开源权重的语言模型,鲍勃使用该模型的黑箱衍生版本生成文本。爱丽丝能否证明鲍勃在使用她的模型?我们将其建模为独立性检验问题:零假设是鲍勃的模型或输出与爱丽丝随机打乱的训练过程无关。基于语言模型中的回溯记忆现象——模型更可能记住训练后期的数据——我们设计统计量来捕捉鲍勃的模型或文本与爱丽丝训练数据顺序之间的相关性。若爱丽丝随机打乱了训练数据,则任何显著相关性都构成可量化的统计证据。在查询设置中,通过提示估计鲍勃模型对爱丽丝训练样本及其顺序的似然度,对40多个参数从1B到12B的Pythia和OLMo基模型微调版本进行分析,除6个外所有情况p值均不超过1e-8。在观测设置中,采用两种方法:一是估算鲍勃文本与爱丽丝训练片段重叠的似然度;二是通过重新运行爱丽丝训练最后1%阶段并打乱数据,构建多个模型版本,比较鲍勃文本对其的似然度。第二种方法可在仅几百词文本中可靠区分;第一种无需重训练但需数百万词才能获得高检验力。
原文摘要 · Abstract (English)
Suppose Alice trains an open-weight language model and Bob uses a blackbox derivative of Alice's model to produce text. Can Alice prove that Bob is using her model, either by querying Bob's derivative model (query setting) or from the text alone (observational setting)? We formulate this question as an independence testing problem--in which the null hypothesis is that Bob's model or text is independent of Alice's randomized training run--and investigate it through the lens of palimpsestic memorization in language models: models are more likely to memorize data seen later in training, so we can test whether Bob is using Alice's model using test statistics that capture correlation between Bob's model or text and the ordering of training examples in Alice's training run. If Alice has randomly shuffled her training data, then any significant correlation amounts to exactly quantifiable statistical evidence against the null hypothesis, regardless of the composition of Alice's training data. In the query setting, we directly estimate (via prompting) the likelihood Bob's model gives to Alice's training examples and order; we correlate the likelihoods of over 40 fine-tunes of various Pythia and OLMo base models ranging from 1B to 12B parameters with the base model's training data order, achieving a p-value on the order of at most 1e-8 in all but six cases. In the observational setting, we try two approaches based on estimating 1) the likelihood of Bob's text overlapping with spans of Alice's training examples and 2) the likelihood of Bob's text with respect to different versions of Alice's model we obtain by repeating the last phase (e.g., 1%) of her training run on reshuffled data. The second approach can reliably distinguish Bob's text from as little as a few hundred tokens; the first does not involve any retraining but requires many more tokens (several hundred thousand) to achieve high power.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。