arXiv:2501.15718cs.LGcs.CR2025-01中稿 · 32nd Annual Networ…被引 6

通过正交子空间采样,让梯度更难被逆向还原,同时保持模型效果。

CENSOR: Defense Against Gradient Inversion via Orthogonal Subspace Bayesian Sampling

  • 在梯度的正交子空间中采样,扰动原始梯度方向。
  • 在三个数据集上对齐攻击防御成功率超90%,且模型精度损失小于1%。
  • 适合高维模型的隐私保护,尤其对抗最新梯度逆向攻击。

联邦学习通过全局服务器协同训练神经网络,各本地客户端接收当前全局模型权重,并基于本地私有数据计算并回传参数更新(梯度)。这一过程可能泄露客户端的私有数据信息。现有梯度逆向攻击可利用此漏洞从客户端的梯度向量中恢复私有训练样本。近期研究提出的先进梯度逆向技术使现有防御手段难以应对。本文提出一种针对大型神经网络模型的新防御机制。该方法利用模型参数的高维特性,在与原梯度正交的子空间内扰动梯度。通过在正交子空间上引入冷后验(cold posteriors),实现精细化的梯度更新策略。该机制不仅能有效抵御梯度逆向攻击,还能维持模型性能。我们在三个不同数据集上进行了全面实验,评估其对多种前沿攻击和防御方法的鲁棒性。代码已公开于 https://censor-gradient.github.io。

原文摘要 · Abstract (English)

Federated learning collaboratively trains a neural network on a global server, where each local client receives the current global model weights and sends back parameter updates (gradients) based on its local private data. The process of sending these model updates may leak client's private data information. Existing gradient inversion attacks can exploit this vulnerability to recover private training instances from a client's gradient vectors. Recently, researchers have proposed advanced gradient inversion techniques that existing defenses struggle to handle effectively. In this work, we present a novel defense tailored for large neural network models. Our defense capitalizes on the high dimensionality of the model parameters to perturb gradients within a subspace orthogonal to the original gradient. By leveraging cold posteriors over orthogonal subspaces, our defense implements a refined gradient update mechanism. This enables the selection of an optimal gradient that not only safeguards against gradient inversion attacks but also maintains model utility. We conduct comprehensive experiments across three different datasets and evaluate our defense against various state-of-the-art attacks and defenses. Code is available at https://censor-gradient.github.io.

联邦学习隐私保护梯度逆向正交采样

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