Gradient Mirage: Trainable yet Label-Unidentifiable Gradients in Large Language Model Split Learning
作者: Shiyu Miao, Yunlong Mao, Zirui Huang, Liang Yao, Tianshuo Zheng, Yanhui Gu, Fan Liu, Sheng Zhong
分类: cs.CL, cs.LG
发布日期: 2026-08-19
💡 一句话要点
提出Gradient Mirage以解决大语言模型分割学习中的梯度匹配攻击问题
🎯 匹配领域: 支柱九:具身大模型 (Embodied Foundation Models)
关键词: 梯度匹配攻击 分割学习 隐私保护 大语言模型 自回归监督 方向性差分隐私 优化信号 随机化技术
📋 核心要点
- 现有的分割学习方法在面对梯度匹配攻击时,假设暴露的梯度与训练目标一致,导致隐私泄露风险。
- 论文提出Gradient Mirage,通过诱导攻击者解决一个错误指定的逆问题,打破梯度与目标的一致性。
- 实验结果显示,Gradient Mirage在保护隐私的同时,保持了良好的微调性能,显著优于现有防御方法。
📝 摘要(中文)
梯度匹配攻击(GMA)在大语言模型分割学习中依赖于一个关键但未被充分探讨的假设:在分割接口暴露的梯度是客户端全标签训练目标的真实导数。我们提出Gradient Mirage,一种防御机制,打破这一一致性,同时保留反向信号的优化效用。其核心思想是诱使攻击者解决一个错误指定的逆问题,使得在序列空间中没有合理的标签序列能够解释观察到的梯度。Gradient Mirage通过在目标、方向和规模三个维度上诱导不一致性来实现这一点。实验表明,Gradient Mirage在可比微调性能下提供了显著更强的保护,达到了更好的隐私-效用权衡。
🔬 方法详解
问题定义:论文要解决的问题是现有分割学习方法在面对梯度匹配攻击时,攻击者能够通过暴露的梯度恢复客户端的私有标签,导致隐私泄露。现有方法未能有效打破梯度与训练目标之间的假设一致性,存在安全隐患。
核心思路:论文的核心思路是通过诱导攻击者解决一个错误指定的逆问题,使得没有合理的标签序列能够解释观察到的梯度。这一设计旨在打破梯度与目标之间的直接关联,从而增强隐私保护。
技术框架:整体架构包括三个主要模块:选择性自回归监督(Selective Autoregressive Supervision)、规模盲化(Scale Blinding)和方向隐私化(Directional Privatization)。选择性自回归监督从掩蔽的替代损失中导出暴露的梯度,规模盲化通过随机乘法重缩放来模糊梯度的自然幅度,方向隐私化则在保持幅度的同时随机化梯度方向。
关键创新:最重要的技术创新点在于通过三维诱导不一致性(目标、方向和规模)来打破攻击者的假设,从而有效防止梯度匹配攻击。这与现有方法的本质区别在于,现有方法通常依赖于保持梯度与目标的一致性。
关键设计:关键设计包括使用掩蔽的替代损失函数来计算梯度,随机化乘法重缩放以模糊梯度幅度,以及通过von Mises-Fisher机制在方向上进行随机化,同时确保满足方向性差分隐私的保障。
🖼️ 关键图片
📊 实验亮点
实验结果表明,Gradient Mirage在与现有防御方法的比较中,提供了显著更强的隐私保护,同时保持了微调性能。具体而言,在相似的微调效果下,Gradient Mirage的隐私保护能力提升了XX%,显示出更优的隐私-效用权衡。
🎯 应用场景
该研究的潜在应用领域包括保护用户隐私的分割学习系统,尤其是在医疗、金融等敏感数据处理场景中。通过增强隐私保护,Gradient Mirage可以促进更广泛的分割学习技术的应用,减少数据泄露风险,提升用户信任。
📄 摘要(原文)
Gradient matching attacks (GMAs) in LLM split learning (SL) rely on a critical yet underexplored assumption: the gradient exposed at the split interface is a faithful derivative of the client's full-label training objective. This gradient-objective consistency allows a curious server to recover private labels by searching for a sequence whose induced gradient explains the observation. We propose Gradient Mirage, a defense that breaks this consistency without discarding the optimization utility of the backward signal. Our key idea is to induce the adversary to solve a misspecified inverse problem, in which no plausible label sequence in the sequence space can explain the observed gradients. Concretely, Gradient Mirage achieves this by inducing inconsistency across three dimensions: objective, direction, and scale. Selective Autoregressive Supervision derives the exposed gradient from a masked surrogate loss rather than the full-label objective assumed by the attacker; Scale Blinding then applies randomized multiplicative rescaling, obscuring the gradient's natural magnitude; and Directional Privatization further randomizes the gradient direction while preserving its magnitude through the von Mises-Fisher (vMF) mechanism under a directional metric differential privacy guarantee. Crucially, utility is preserved: the Top segment still learns from all target tokens via Dual-Track Backpropagation, the exposed gradient remains informative since each supervised token retains its complete autoregressive context, and Bottom-Gradient Recovery restores the effective gradient for Bottom-segment optimization. Extensive experiments show that Gradient Mirage provides substantially stronger protection than existing defenses under comparable fine-tuning performance, achieving a better privacy-utility trade-off.