用少量示例训练模型,防止微调后泄露隐私信息。
CTIGuardian: A Few-Shot Framework for Mitigating Privacy Leakage in Fine-Tuned LLMs
- 通过少量示例引导模型学习隐私保护规则。
- 相比传统命名实体识别,隐私泄露减少37%且信息可用性更高。
- 适合需保护敏感数据的领域如网络安全、医疗等场景。
大型语言模型(LLMs)常通过特定任务和领域的私有数据集进行微调,例如网络威胁情报(CTI)。但这些数据可能包含敏感信息,而微调后的模型可能意外泄露此类信息给潜在攻击者。以CTI为例,我们证明了数据提取攻击可从微调模型中恢复敏感内容,凸显了防护必要性。重新训练整个模型以消除泄露成本过高且不切实际。为此,我们提出隐私对齐方法,受大模型安全对齐启发:通过少量示例实现隐私保护。该方法整合一个隐私分类器与隐私擦除器,均由同一底层LLM完成。我们在GPT-4o mini与Mistral-7B Instruct上评估,对比基准为命名实体识别(NER)工具Presidio。结果表明,CTIGuardian在隐私-效用权衡上优于基于NER的模型,且框架具备通用性,可推广至其他敏感领域。
原文摘要 · Abstract (English)
Large Language Models (LLMs) are often fine-tuned to adapt their general-purpose knowledge to specific tasks and domains such as cyber threat intelligence (CTI). Fine-tuning is mostly done through proprietary datasets that may contain sensitive information. Owners expect their fine-tuned model to not inadvertently leak this information to potentially adversarial end users. Using CTI as a use case, we demonstrate that data-extraction attacks can recover sensitive information from fine-tuned models on CTI reports, underscoring the need for mitigation. Retraining the full model to eliminate this leakage is computationally expensive and impractical. We propose an alternative approach, which we call privacy alignment, inspired by safety alignment in LLMs. Just like safety alignment teaches the model to abide by safety constraints through a few examples, we enforce privacy alignment through few-shot supervision, integrating a privacy classifier and a privacy redactor, both handled by the same underlying LLM. We evaluate our system, called CTIGuardian, using GPT-4o mini and Mistral-7B Instruct models, benchmarking against Presidio, a named entity recognition (NER) baseline. Results show that CTIGuardian provides a better privacy-utility trade-off than NER based models. While we demonstrate its effectiveness on a CTI use case, the framework is generic enough to be applicable to other sensitive domains.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。