arXiv:2410.02016cs.LGcs.CR2024-10被引 3

让大模型预测更私密:动态调整保护强度,效率提升16倍

Adaptively Private Next-Token Prediction of Large Language Models

  • 根据输入动态调整隐私保护强度,避免固定成本
  • 实验显示隐私损失降低16倍,10万次预测后仍保持良好效果
  • 适合云端部署的大模型服务,兼顾隐私与实用性能

随着大语言模型广泛应用,其隐私保护愈发重要。现有差分隐私训练方法计算开销大且损害模型性能。针对云服务中仅通过API访问模型的场景,本文提出自适应隐私预测框架AdaPMixED,基于PMixED思想但不再固定隐私预算。引入噪声筛选机制,剔除可能高隐私损耗的查询;采用依赖数据的分析方式,利用私有与公开模型输出分布的差异来计算隐私损失。实验表明,该方法在保持模型性能的同时,将隐私损失降低16倍。即使执行10万次预测,仍能实现5.25的合理数据依赖隐私损失,显著优于原始PMixED。

原文摘要 · Abstract (English)

As Large Language Models (LLMs) proliferate, developing privacy safeguards for these models is crucial. One popular safeguard involves training LLMs in a differentially private manner. However, such solutions are shown to be computationally expensive and detrimental to the utility of these models. Since LLMs are deployed on the cloud and thus only accessible via an API, a Machine Learning as a Service (MLaaS) provider can protect its downstream data by privatizing the predictions during the decoding process. However, the practicality of such solutions still largely lags behind DP training methods. One recent promising approach, Private Mixing of Ensemble Distributions (PMixED), avoids additive noise by sampling from the output distributions of private LLMs mixed with the output distribution of a public model. Yet, PMixED must satisfy a fixed privacy level for a given number of queries, which is difficult for an analyst to estimate before inference and, hence, does not scale. To this end, we relax the requirements to a more practical setting by introducing Adaptive PMixED (AdaPMixED), a private decoding framework based on PMixED that is adaptive to the private and public output distributions evaluated on a given input query. In this setting, we introduce a noisy screening mechanism that filters out queries with potentially expensive privacy loss, and a data-dependent analysis that exploits the divergence of the private and public output distributions in its privacy loss calculation. Our experimental evaluations demonstrate that our mechanism and analysis can reduce the privacy loss by 16x while preserving the utility over the original PMixED. Furthermore, performing 100K predictions with AdaPMixED still achieves strong utility and a reasonable data-dependent privacy loss of 5.25.

大模型隐私保护差分隐私自适应

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