仅通过生成文本推断数据是否在大模型训练集中,突破了以往需访问输出概率的限制。
Towards Label-Only Membership Inference Attack against Pre-trained Large Language Models
- 基于分词级语义相似度估算输出概率,构建标签仅限的成员推断攻击。
- 在WikiMIA和MIMIR基准上性能超越现有标签仅限攻击,媲美需完整输出的先进方法。
- 适用于研究大模型数据隐私漏洞,尤其关注训练数据泄露风险的从业者。
成员推断攻击(MIA)旨在判断某数据样本是否属于模型训练集。尽管已有大量研究探讨大型语言模型(LLMs)的成员推断攻击,但多数方法依赖完整的输出逻辑值(即“基于逻辑值的攻击”),而这一信息在实际中通常不可用。本文研究预训练大语言模型在“仅标签设置”下的脆弱性,即攻击者只能获取生成的文本标记。我们发现,现有标签仅限的成员推断攻击对预训练模型效果甚微,尽管它们在个性化模型的微调数据集上表现优异。其失败原因主要有二:更强的泛化能力与扰动过粗。由于预训练语料库广泛且每条样本仅出现几次,成员与非成员间的鲁棒性差异极小,导致逐标记扰动无法捕捉该差异。为此,我们提出PETAL:一种基于“逐标记语义相似性”的标签仅限成员推断攻击。PETAL利用标记级语义相似度近似输出概率,并计算困惑度,最终依据成员更可能被“更好记忆”、困惑度更低的假设判断成员身份。我们在WikiMIA基准和更具挑战性的MIMIR基准上进行了广泛实验。实证结果表明,PETAL在五种主流开源大模型上的表现优于现有标签仅限攻击的扩展版本,甚至在所有指标上与先进的基于逻辑值的攻击相当。
原文摘要 · Abstract (English)
Membership Inference Attacks (MIAs) aim to predict whether a data sample belongs to the model's training set or not. Although prior research has extensively explored MIAs in Large Language Models (LLMs), they typically require accessing to complete output logits (\ie, \textit{logits-based attacks}), which are usually not available in practice. In this paper, we study the vulnerability of pre-trained LLMs to MIAs in the \textit{label-only setting}, where the adversary can only access generated tokens (text). We first reveal that existing label-only MIAs have minor effects in attacking pre-trained LLMs, although they are highly effective in inferring fine-tuning datasets used for personalized LLMs. We find that their failure stems from two main reasons, including better generalization and overly coarse perturbation. Specifically, due to the extensive pre-training corpora and exposing each sample only a few times, LLMs exhibit minimal robustness differences between members and non-members. This makes token-level perturbations too coarse to capture such differences. To alleviate these problems, we propose \textbf{PETAL}: a label-only membership inference attack based on \textbf{PE}r-\textbf{T}oken sem\textbf{A}ntic simi\textbf{L}arity. Specifically, PETAL leverages token-level semantic similarity to approximate output probabilities and subsequently calculate the perplexity. It finally exposes membership based on the common assumption that members are `better' memorized and have smaller perplexity. We conduct extensive experiments on the WikiMIA benchmark and the more challenging MIMIR benchmark. Empirically, our PETAL performs better than the extensions of existing label-only attacks against personalized LLMs and even on par with other advanced logit-based attacks across all metrics on five prevalent open-source LLMs.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。