arXiv:2504.05618cs.LGcs.AI2025-04

提出几何优化的差分隐私方法,显著提升训练效率

Technical Report: Full Version of Analyzing and Optimizing Perturbation of DP-SGD Geometrically

  • 分离梯度方向与大小的噪声扰动,针对性优化
  • 相同隐私保护下,模型准确率提升5%-12%
  • 适用于多种模型和数据集,通用性强

差分隐私(DP)已成为机器学习中的主流隐私保护范式,尤其在DP-SGD提出后广泛应用。然而,直接在训练中对梯度加噪的DP-SGD无法缓解噪声对梯度方向的负面影响,导致效率低下。尽管已有方法如梯度裁剪和隐私预算放大等用于权衡隐私与效率,但其根本原因尚未揭示。本文首次泛化DP-SGD并理论分析其对训练过程的影响,发现:在受扰梯度中,仅方向噪声显著影响模型效率,而大小噪声可通过调整梯度裁剪和学习率缓解。此外,传统方法在梯度本身加无偏噪声时,会引入方向上的偏差。因此,从几何角度看,现有扰动策略次优。为此,我们设计了基于几何视角的扰动策略GeoDP,分别对梯度的方向和大小进行扰动。通过直接减少方向噪声,GeoDP在保持相同隐私保障的前提下,显著降低噪声对效率的负面影响。在两个公开数据集(MNIST、CIFAR-10)、一个合成数据集及三种主流模型(逻辑回归、CNN、ResNet)上的大量实验验证了该策略的有效性与普适性。

原文摘要 · Abstract (English)

Differential privacy (DP) has become a prevalent privacy model in a wide range of machine learning tasks, especially after the debut of DP-SGD. However, DP-SGD, which directly perturbs gradients in the training iterations, fails to mitigate the negative impacts of noise on gradient direction. As a result, DP-SGD is often inefficient. Although various solutions (e.g., clipping to reduce the sensitivity of gradients and amplifying privacy bounds to save privacy budgets) are proposed to trade privacy for model efficiency, the root cause of its inefficiency is yet unveiled. In this work, we first generalize DP-SGD and theoretically derive the impact of DP noise on the training process. Our analysis reveals that, in terms of a perturbed gradient, only the noise on direction has eminent impact on the model efficiency while that on magnitude can be mitigated by optimization techniques, i.e., fine-tuning gradient clipping and learning rate. Besides, we confirm that traditional DP introduces biased noise on the direction when adding unbiased noise to the gradient itself. Overall, the perturbation of DP-SGD is actually sub-optimal from a geometric perspective. Motivated by this, we design a geometric perturbation strategy GeoDP within the DP framework, which perturbs the direction and the magnitude of a gradient, respectively. By directly reducing the noise on the direction, GeoDP mitigates the negative impact of DP noise on model efficiency with the same DP guarantee. Extensive experiments on two public datasets (i.e., MNIST and CIFAR-10), one synthetic dataset and three prevalent models (i.e., Logistic Regression, CNN and ResNet) confirm the effectiveness and generality of our strategy.

差分隐私梯度扰动高效训练几何优化

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