arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2410.02016cs.LGcs.CR
收藏

大型语言模型的自适应私有下一词预测

Adaptively Private Next-Token Prediction of Large Language Models

University of Southern California(南加州大学)

James Flemings, Meisam Razaviyayn, Murali Annavaram

AI总结:

本文提出自适应私有解码框架AdaPMixED,通过带噪筛选机制和数据依赖分析,在保持模型实用性的同时将隐私损失降低16倍,解决了PMixED固定隐私水平难以扩展的问题。

摘要:

随着大型语言模型(LLMs)的迅速普及,为这些模型开发隐私保护机制至关重要。一种流行的保护措施是以差分隐私(DP)方式训练LLMs。然而,这类方案被证明计算开销高昂,并且会损害模型的实用性。由于LLMs通常部署在云端,只能通过API访问,机器学习即服务(MLaaS)提供商可以通过在解码过程中对预测进行私有化来保护其下游数据。然而,此类方案的实用性仍远落后于DP训练方法。最近一种有前景的方法——集成分布私有混合(PMixED)——通过从私有LLM的输出分布与公共模型的输出分布混合后的分布中采样来避免添加噪声。然而,PMixED必须在给定查询数量下满足固定的隐私水平,这使分析人员在推理前难以估计,因此无法扩展。为此,我们将要求放宽到更实际的设置,提出了自适应PMixED(AdaPMixED),这是一个基于PMixED的私有解码框架,能够根据给定输入查询上评估的私有和公共输出分布进行自适应调整。在该设置中,我们引入了一种带噪筛选机制,用于过滤掉可能带来高昂隐私损失的查询;同时提出了一种数据依赖的分析方法,在隐私损失计算中利用私有和公共输出分布之间的散度。实验评估表明,我们的机制和分析方法在保持原始PMixED实用性的同时,将隐私损失降低了16倍。此外,使用AdaPMixED执行10万次预测仍能实现较强的实用性,并达到5.25的合理数据依赖隐私损失。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑