让拉普拉斯机制在高维模型中实用,提升隐私训练效果
Lap2: Revisiting Laplace DP-SGD for High Dimensions via Majorization Theory
- 用坐标级矩界和凸性理论,实现拉普拉斯机制下的L2裁剪
- 在ε=0.54下微调RoBERTa-base达87.88%准确率,优于高斯机制
- 适合追求强隐私保障的高维模型训练场景
差分隐私随机梯度下降(DP-SGD)是深度学习中保障隐私的核心技术,广泛应用于从头训练和大语言模型微调。尽管主流采用高斯机制,拉普拉斯机制因依赖L1范数裁剪而长期被忽视。高维模型中,n维梯度的L1范数可能高达L2范数的√n倍,导致噪声尺度随模型规模急剧上升,严重损害模型性能或导致不可训练。本文提出Lap2,通过计算坐标级矩界并运用主序理论,构建不依赖数据的全模型紧上界,利用矩会计函数的Schur-凸性,设计满足L2裁剪约束的主序集,实现多变量隐私会计。该方法可支持数千阶矩,随模型维度增长仍保持良好性能。实验证明,其显著提升拉普拉斯DP-SGD表现,在ε=0.54下微调RoBERTa-base(125M参数)于SST-2任务达87.88%准确率,优于高斯机制(87.16%)与标准拉普拉斯(48.97%)。
原文摘要 · Abstract (English)
Differentially Private Stochastic Gradient Descent (DP-SGD) is a cornerstone technique for ensuring privacy in deep learning, widely used in both training from scratch and fine-tuning large-scale language models. While DP-SGD predominantly relies on the Gaussian mechanism, the Laplace mechanism remains underutilized due to its reliance on L1 norm clipping. This constraint severely limits its practicality in high-dimensional models because the L1 norm of an n-dimensional gradient can be up to sqrt(n) times larger than its L2 norm. As a result, the required noise scale grows significantly with model size, leading to poor utility or untrainable models. In this work, we introduce Lap2, a new solution that enables L2 clipping for Laplace DP-SGD while preserving strong privacy guarantees. We overcome the dimensionality-driven clipping barrier by computing coordinate-wise moment bounds and applying majorization theory to construct a tight, data-independent upper bound over the full model. By exploiting the Schur-convexity of the moment accountant function, we aggregate these bounds using a carefully designed majorization set that respects the L2 clipping constraint. This yields a multivariate privacy accountant that scales gracefully with model dimension and enables the use of thousands of moments. Empirical evaluations demonstrate that our approach significantly improves the performance of Laplace DP-SGD, achieving results comparable to or better than Gaussian DP-SGD under strong privacy constraints. For instance, fine-tuning RoBERTa-base (125M parameters) on SST-2 achieves 87.88% accuracy at epsilon=0.54, outperforming Gaussian (87.16%) and standard Laplace (48.97%) under the same budget.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。