LoRA微调模型存在成员推断漏洞,预训练模型反而加剧隐私泄露
LoRA-Leak: Membership Inference Attacks Against LoRA Fine-tuned Language Models
- 利用预训练模型作为参考,设计15种成员推断攻击方法
- 在保守设置下仍达0.775的AUC,证明LoRA模型易受成员推断攻击
- 仅丢弃特定层或使用丢弃法能有效防御,同时保持模型性能
语言模型通常采用“预训练+微调”范式,其中通用预训练模型可被微调以适应各类专业领域。低秩适配(LoRA)因其计算开销小、效果佳而被广泛使用。由于LoRA仅调整少量参数,常被认为对成员推断攻击(MIA)具有免疫力。然而我们发现,利用预训练模型会引发更严重的信息泄露,现有MIA方法忽视了这一点。为此,我们提出LoRA-Leak,一个全面评估基于LoRA微调语言模型训练数据成员推断风险的框架。该框架整合了15种攻击方法,包括10种现有攻击和5种改进攻击,均利用预训练模型作为参考。我们在三个先进语言模型上,针对三个主流NLP任务进行实验,结果表明:即使在保守微调设置下,基于LoRA的模型仍面临显著的成员推断风险(如AUC达0.775)。我们进一步测试不同微调设置,分析隐私风险变化。还评估了四种防御策略,发现只有在微调时丢弃特定层或引入随机丢弃(dropout)能有效降低攻击成功率,同时维持模型可用性。研究强调,在‘预训练+微调’范式下,预训练模型的存在使基于LoRA的语言模型面临更严重的成员推断威胁。
原文摘要 · Abstract (English)
Language Models (LMs) typically adhere to a "pre-training and fine-tuning" paradigm, where a universal pre-trained model can be fine-tuned to cater to various specialized domains. Low-Rank Adaptation (LoRA) has gained the most widespread use in LM fine-tuning due to its lightweight computational cost and remarkable performance. Because the proportion of parameters tuned by LoRA is relatively small, there might be a misleading impression that the LoRA fine-tuning data is invulnerable to Membership Inference Attacks (MIAs). However, we identify that utilizing the pre-trained model can induce more information leakage, which is neglected by existing MIAs. Therefore, we introduce LoRA-Leak, a holistic evaluation framework for MIAs against the fine-tuning datasets of LMs. LoRA-Leak incorporates fifteen membership inference attacks, including ten existing MIAs, and five improved MIAs that leverage the pre-trained model as a reference. In experiments, we apply LoRA-Leak to three advanced LMs across three popular natural language processing tasks, demonstrating that LoRA-based fine-tuned LMs are still vulnerable to MIAs (e.g., 0.775 AUC under conservative fine-tuning settings). We also applied LoRA-Leak to different fine-tuning settings to understand the resulting privacy risks. We further explore four defenses and find that only dropout and excluding specific LM layers during fine-tuning effectively mitigate MIA risks while maintaining utility. We highlight that under the "pre-training and fine-tuning" paradigm, the existence of the pre-trained model makes MIA a more severe risk for LoRA-based LMs. We hope that our findings can provide guidance on data privacy protection for specialized LM providers.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。