差分隐私让神经网络训练更难预测,但稳定边界规律仍存在
Optimizer Dynamics at the Edge of Stability with Differential Privacy
- 对比经典与隐私保护优化器,研究梯度裁剪和噪声对训练的影响
- 隐私保护训练降低模型尖锐度,但仍能逼近甚至超过经典稳定阈值
- 适合关注隐私训练中优化行为的模型开发者与安全研究人员
深度学习模型可能泄露个体训练样本的敏感信息,尽管差分隐私(DP)能限制此类泄露,但其对优化动态的影响尚不明确。本文通过比较梯度下降(GD)与自适应优化器Adam及其隐私保护版本,研究神经网络在DP训练下的动态变化。已有研究表明,全批量方法在边缘稳定性(EoS)区域训练,而小批量和自适应方法也表现出类似的边缘稳定性行为。在此类区域,训练损失与尖锐度——即训练损失海森矩阵的最大特征值——呈现特定规律。在DP训练中,逐样本梯度裁剪和高斯噪声改变了更新规则,目前尚不清楚这些稳定性模式是否依然存在。本文分析裁剪与噪声如何影响尖锐度与损失演化,发现尽管DP通常降低尖锐度并可能阻止优化器完全达到经典稳定性阈值,但来自EoS及类似自适应方法的稳定性模式仍持续存在,最大学习率与最大隐私预算下,甚至可逼近或超越这些阈值。结果揭示了差分隐私为神经网络优化引入的不可预测性。
原文摘要 · Abstract (English)
Deep learning models can reveal sensitive information about individual training examples, and while differential privacy (DP) provides guarantees restricting such leakage, it also alters optimization dynamics in poorly understood ways. We study the training dynamics of neural networks under DP by comparing Gradient Descent (GD), and Adam to their privacy-preserving variants. Prior work shows that these optimizers exhibit distinct stability dynamics: full-batch methods train at the Edge of Stability (EoS), while mini-batch and adaptive methods exhibit analogous edge-of-stability behavior. At these regimes, the training loss and the sharpness--the maximum eigenvalue of the training loss Hessian--exhibit certain characteristic behavior. In DP training, per-example gradient clipping and Gaussian noise modify the update rule, and it is unclear whether these stability patterns persist. We analyze how clipping and noise change sharpness and loss evolution and show that while DP generally reduces the sharpness and can prevent optimizers from fully reaching the classical stability thresholds, patterns from EoS and analogous adaptive methods stability regimes persist, with the largest learning rates and largest privacy budgets approaching, and sometimes exceeding, these thresholds. These findings highlight the unpredictability introduced by DP in neural network optimization.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。