跳到正文
返回论文列表
cs.CR提交于 已译

从训练暴露到成员证据:OLMo 2 的受控研究

From Training Exposure to Membership Evidence: A Controlled Study of OLMo 2

Ali Banaei · Muhammad Shahab Shahzad · Gerald Q. Maguire Jr. · Marco Chiesa · Dejan Kosti\'c

中文摘要

语言模型是否在某个文档上训练过,关系到版权、隐私和基准污染。但现有语言模型成员推断(membership-inference, MI)评估大多事后收集成员与非成员,成功可能反映分布偏移,失败也无法定论。重新运行 OLMo 2 1B 官方 500 亿 token 中期训练阶段,用来自 7,814 对按领域和难度匹配的文档替换部分训练块;每对文档通过抛硬币决定哪篇插入、哪篇留出。Allen Institute for AI(AI2)以相同种子发布的运行在其他数据上完全一致,作为反事实参照。一次暴露平均会提高未见文档的似然,但很少能识别出该文档。以反事实为参照,参考比值的 ROC 曲线下面积(AUC)为 0.767,但跨领域使用单一阈值时,在 1% 假阳性率(FPR)下仅捕获 3.8% 的成员。相比之下,13 对匹配文档使配对检验在 5% 显著性水平下具有 80% 功效,可在给定同领域非成员和接近参照时检测集合级训练。用另一种子训练的参照恢复了反事实相对随机猜测的 AUC 优势的 92%,而无参照攻击跨领域合并时与随机猜测的差距在 0.07 以内。完整官方后训练配方在两个模型上重跑后,平均效应保留 89%。即使参照高度匹配,文档级变异性仍然很大,限制了单个文档的成员推断。将发布检查点、逐 token 对数概率和评估文档;代码见 https://github.com/ali-bana/OLMo-Twin。

关键要点

  1. 01事后收集成员与非成员会使成员推断结果受分布偏移干扰,失败也无法定论。
  2. 02重跑 OLMo 2 1B 中期训练,用 7,814 对匹配文档做受控插入/留出实验。
  3. 03单次暴露平均提高文档似然,但跨领域单阈值在 1% FPR 下仅捕获 3.8% 成员。
  4. 0413 对匹配文档即可在 5% 水平下以 80% 功效检测集合级训练。
  5. 05文档级变异性大,即使参照高度匹配,单个文档的成员推断仍受限。

解读

尚无解读。

原始英文摘要

arXiv:2610.41732v1 Announce Type: cross Abstract: Whether a language model was trained on a document matters for copyright, privacy, and benchmark contamination. Yet most membership-inference (MI) evaluations on language models collect members and non-members after the fact, so success may reflect distribution shift and failure is inconclusive. We re-ran OLMo 2 1B's official 50-billion-token mid-training stage, replacing selected training chunks with documents from 7,814 pairs matched on domain and difficulty. For each pair, a coin flip determined which document was inserted and which was held out. The run released by the Allen Institute for AI (AI2) from the same seed saw identical data everywhere else and serves as the counterfactual. One exposure raises a previously unseen document's likelihood on average but rarely identifies it. Using the counterfactual as reference, the reference ratio has an area under the ROC curve (AUC) of 0.767 but, with one threshold across domains, catches 3.8% of members at a 1% false-positive rate (FPR). In contrast, 13 matched pairs give a paired test 80% power at the 5% level to detect training on a collection, given same-domain non-members and a close reference. A reference trained with another seed recovers 92% of the counterfactual's AUC advantage over chance, whereas reference-free attacks stay within 0.07 of chance when pooled over domains. The full official post-training recipe, re-run on both models, leaves 89% of the average effect. Even with a closely matched reference, substantial document-level variability limits individual membership inference. We will release the checkpoints, per-token log-probabilities, and the evaluation documents; the code is available at https://github.com/ali-bana/OLMo-Twin.

同方向论文 · cs.CR

查看全部 →