arXiv:2505.11837cs.LGcs.AI2025-05被引 2

知识蒸馏未必更安全,反而可能放大隐私泄露风险

On Membership Inference Attacks in Knowledge Distillation

  • 用混合监督机制解释蒸馏如何让模型更易被成员推理攻击
  • 部分学生模型的成员攻击成功率比教师模型高出47%
  • 提出三种实用方法,兼顾隐私与模型性能

大型语言模型(LLMs)训练数据可能包含敏感信息,面临成员推理攻击(MIA)风险。知识蒸馏常用于压缩大模型为小模型,但其隐私影响尚不明确。我们系统评估了六对教师-学生模型及六种攻击方法下蒸馏对MIA脆弱性的影响。结果发现,学生模型并未一致降低MIA成功率,某些情况下成员特定攻击成功率达47%以上,挑战了蒸馏能提升隐私的假设。根源在于蒸馏中的混合监督:对易受攻击的数据点,教师预测与真实标签高度一致,导致学生模型学习到过于自信的预测,加剧成员与非成员间的可区分性;而对不易受攻击点,教师预测与真实标签常不一致,提供矛盾学习信号。为此,我们提出三种干预措施——仅对非易受攻击点进行蒸馏、引入低维瓶颈投影(Bottleneck Projection)、以及一种归一化变体(NoNorm)。实验表明,这些方法在保持模型性能的同时,显著降低了聚合和成员特定的MIA成功率,改善了蒸馏模型的隐私-效用权衡。

原文摘要 · Abstract (English)

Large language models (LLMs) are trained on massive corpora that may contain sensitive information, creating privacy risks under membership inference attacks (MIAs). Knowledge distillation is widely used to compress LLMs into smaller student models, but its privacy implications are poorly understood. We systematically evaluate how distillation affects MIA vulnerability across six teacher-student model pairs and six attack methods. We find that distilled student models do not consistently exhibit lower MIA success than their teacher models, and in some cases demonstrate substantially higher member-specific attack success, challenging the assumption that knowledge distillation inherently improves privacy. We attribute this to mixed supervision in distillation: for vulnerable training data points, teacher predictions often align with ground-truth labels, causing student models to learn overly confident predictions that amplify the separability between members and non-members; conversely, for non-vulnerable points, teacher predictions and ground truth frequently diverge, providing inconsistent learning signals. To mitigate this, we propose three practical interventions -- restricting distillation to non-vulnerable points, adding a low-dimensional Bottleneck Projection, and a normalization variant (NoNorm). Experiments show these methods reduce both aggregate and member-specific MIA success while preserving model utility, improving privacy-utility trade-offs for distilled LLMs.

知识蒸馏隐私安全成员推理攻击

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