arXiv:2410.19482cs.LG2024-10NAACL被引 55

提出概率化方法评估大模型数据记忆风险,更真实反映生成不确定性。

Measuring memorization in language models via probabilistic extraction

  • 用多轮采样评估记忆提取概率,突破传统单次贪婪采样的局限
  • 实验显示不同采样策略下记忆风险差异显著,传统方法易误判
  • 适合关注模型隐私安全的研究者与开发人员参考

大语言模型容易记忆训练数据,可能在生成时泄露敏感信息。现有可发现提取法将训练样本拆分为前缀和后缀,用前缀提示模型,若模型以贪婪采样生成匹配后缀,则判定为可提取。该方法仅给出单一查询的二值结果,计算高效但不可靠,因未考虑实际非贪婪采样中的随机性。本文提出概率化可发现提取,无需额外开销,通过多轮查询量化目标序列被提取的概率。我们在不同模型、采样方式及训练数据重复情况下评估该方法,发现其能提供比传统方法更细致的提取风险信息。

原文摘要 · Abstract (English)

Large language models (LLMs) are susceptible to memorizing training data, raising concerns about the potential extraction of sensitive information at generation time. Discoverable extraction is the most common method for measuring this issue: split a training example into a prefix and suffix, then prompt the LLM with the prefix, and deem the example extractable if the LLM generates the matching suffix using greedy sampling. This definition yields a yes-or-no determination of whether extraction was successful with respect to a single query. Though efficient to compute, we show that this definition is unreliable because it does not account for non-determinism present in more realistic (non-greedy) sampling schemes, for which LLMs produce a range of outputs for the same prompt. We introduce probabilistic discoverable extraction, which, without additional cost, relaxes discoverable extraction by considering multiple queries to quantify the probability of extracting a target sequence. We evaluate our probabilistic measure across different models, sampling schemes, and training-data repetitions, and find that this measure provides more nuanced information about extraction risk compared to traditional discoverable extraction.

模型记忆隐私安全采样策略

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。