给敏感词更多隐私保护,训练快90%还更安全
Adaptive Token-Weighted Differential Privacy for LLMs: Not All Tokens Require Equal Protection
- 按敏感程度动态分配噪声,重点保护敏感词汇
- 训练时间缩短约90%,准确率损失极小
- 可无缝接入现有流程,适合快速加固模型隐私
大型语言模型常记忆敏感信息,引发隐私担忧。现有差分隐私随机梯度下降(DP-SGD)在每一步梯度中注入均匀噪声,导致训练耗时长且精度下降。本文提出自适应令牌加权差分隐私(ATDP),通过为敏感与非敏感令牌分配不同梯度权重,将更多噪声集中在敏感令牌相关梯度上。初期采用较大噪声规模,快速破坏敏感内容的记忆。随后仅需少量轻量级后处理阶段,在敏感令牌对应参数上注入目标噪声,几乎不影响模型泛化能力。ATDP可无缝集成至任意基于DP的微调流程,或直接用于非私有模型作为快速隐私增强手段。结合初始删减微调阶段,形成高效流水线,实现与最先进DP-SGD相当的蜜罐保护效果,计算开销显著降低,训练时间缩短约90%,同时保持相当或更优的隐私保护水平及最小精度损失。
原文摘要 · Abstract (English)
Large language models (LLMs) frequently memorize sensitive or personal information, raising significant privacy concerns. Existing variants of differential privacy stochastic gradient descent (DPSGD) inject uniform noise into every gradient step, significantly extending training time and reducing model accuracy. We propose that concentrating noise primarily on gradients associated with sensitive tokens can substantially decrease DP training time, strengthen the protection of sensitive information, and simultaneously preserve the model's performance on non-sensitive data. We operationalize this insight through Adaptive Token-Weighted Differential Privacy (ATDP), a modification of vanilla DP-SGD that adaptively assigns different gradient weights to sensitive and non-sensitive tokens. By employing a larger noise scale at the early stage of training, ATDP rapidly disrupts memorization of sensitive content. As a result, ATDP only requires a few additional epochs of lightweight post-processing following standard fine-tuning, injecting targeted noise primarily on parameters corresponding to sensitive tokens, thus minimally affecting the model's general capabilities. ATDP can be seamlessly integrated into any existing DP-based fine-tuning pipeline or directly applied to non-private models as a fast privacy-enhancing measure. Additionally, combined with an initial redacted fine-tuning phase, ATDP forms a streamlined DP pipeline that achieves comparable canary protection to state-of-the-art DP-SGD methods, significantly reduces the computational overhead of DP fine-tuning, shortening training time by approximately 90 percent, while achieving comparable or superior privacy protection and minimal accuracy degradation.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。