用单次噪声注入的超网络,降低差分隐私训练的噪声干扰。
Escaping Iterative Parameter-Space Noise: Differentially Private Learning with a Hypernetwork
- 用超网络将私有数据映射为模型参数,仅一次注入低维噪声。
- 在固定隐私预算下,模型性能优于传统DP-SGD方法。
- 适合对隐私敏感且需高精度微调的扩散模型场景。
差分隐私(DP)神经网络训练常因梯度方法如DP-SGD需反复向参数空间注入高维噪声而受限。本文提出新框架,避免在参数空间中迭代优化。不使用私有梯度更新目标模型,而是通过在公开数据上训练的超网络,将私有数据映射为模型参数。具体地,每个样本被嵌入到低维表示,嵌入结果聚合并扰动以获得差分隐私的数据集嵌入,再由超网络生成目标模型参数。由于隐私噪声仅注入一次低维表示,显著减轻了噪声的负面影响。我们在合成设置中理论证明,在固定隐私预算下,本方法生成的模型性能优于DP-SGD。此外,将该方法应用于扩散模型的LoRA微调,结果显示其FID低于使用DP-SGD及其他基于公共数据引导的方法的模型。
原文摘要 · Abstract (English)
Differentially private (DP) training of neural networks is often hindered by the large amount of noise required by gradient-based methods such as DP-SGD, which repeatedly inject high-dimensional noise in parameter space throughout training. In this paper, we propose a new framework for DP learning that avoids iterative optimization in parameter space. Instead of updating the target model using privatized gradients, we employ a hypernetwork trained on public datasets to map a private dataset to the parameters of the target model. Specifically, each example is embedded into a low-dimensional representation, the embeddings are aggregated and perturbed to obtain a DP dataset embedding, and the hypernetwork generates the target model parameters from this noisy embedding. Because privacy noise is injected only once into a low-dimensional dataset representation, our approach can significantly reduce the adverse effect of noise. We theoretically show in a synthetic setting that, under a fixed privacy budget, models produced by our approach achieve higher utility than those trained with DP-SGD. Moreover, we apply our approach to LoRA fine-tuning of diffusion models and show that it achieves lower FID than LoRA models trained with DP-SGD and other public-data-guided methods.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。