arXiv:2506.04566cs.LGcs.AI2025-06被引 6

通过聚类和中位数聚合,提升私有化语言模型生成文本的质量与效率。

Clustering and Median Aggregation Improve Differentially Private Inference

  • 先聚类敏感数据再采样,避免不同主题混合影响生成质量。
  • 利用相似预测结果,改用中位数聚合降低隐私预算消耗。
  • 在保持隐私的前提下,生成文本更贴近真实分布,适合高敏感场景。

差分隐私(DP)语言模型推理是一种生成私有合成文本的方法。将敏感输入样本用于提示现成的大语言模型(LLM),生成类似样本。多个样本可聚合以满足严格的差分隐私保证。以往工作通过随机均匀采样构建推理批次,但我们发现该方法在敏感样本主题异质时会显著降低生成文本质量。为此,我们提出在选取推理批次前对输入数据进行聚类。进一步观察到聚类后各推理的下一个词预测趋于相似,据此提出一种新算法:通过私密计算中位数而非均值来聚合下一个词统计量。由于相似预测下中位数的局部敏感度更低,该方法可给出数据依赖且事后成立的差分隐私保证。实验表明,本方法在代表性指标(如MAUVE)及下游任务性能上均有提升,相比之前最先进的方法,在显著更低的隐私预算下生成高质量合成数据。

原文摘要 · Abstract (English)

Differentially private (DP) language model inference is an approach for generating private synthetic text. A sensitive input example is used to prompt an off-the-shelf large language model (LLM) to produce a similar example. Multiple examples can be aggregated together to formally satisfy the DP guarantee. Prior work creates inference batches by sampling sensitive inputs uniformly at random. We show that uniform sampling degrades the quality of privately generated text, especially when the sensitive examples concern heterogeneous topics. We remedy this problem by clustering the input data before selecting inference batches. Next, we observe that clustering also leads to more similar next-token predictions across inferences. We use this insight to introduce a new algorithm that aggregates next token statistics by privately computing medians instead of averages. This approach leverages the fact that the median has decreased local sensitivity when next token predictions are similar, allowing us to state a data-dependent and ex-post DP guarantee about the privacy properties of this algorithm. Finally, we demonstrate improvements in terms of representativeness metrics (e.g., MAUVE) as well as downstream task performance. We show that our method produces high-quality synthetic data at significantly lower privacy cost than a previous state-of-the-art method.

差分隐私语言模型聚类中位数聚合

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