论文精选

PB-AMD 算法提出差分隐私预算最优分配方案

When to spend privacy: optimal allocation in private learning

精选理由

训练差分隐私模型时噪声怎么加最划算?这篇论文把预算分配变成控制问题,还给出平方根分配律,做隐私训练的可以看看。

arXiv 论文 2610.39954 将 DP-SGD 的高斯噪声方差调度建模为组合零集中差分隐私约束下的开环控制问题。作者证明所有局部最优解都会耗尽认证隐私预算,并据此提出 PB-AMD 算法,用伴随梯度与熵更新的方式保持方差为正且不超出预算。线性二次约化给出显式的平方根分配律,实验显示相对均匀分配在跟踪误差和预测性能上均有改进,调度优化耗时低于投影梯度下降。

原文 · arXiv cs.LG

When to spend privacy: optimal allocation in private learning

Privacy accounting bounds the cumulative privacy cost of Differentially Private Stochastic Gradient Descent but does not determine how to allocate a fixed budget across iterations. We formulate the complete Gaussian noise-variance schedule as an open-loop control under a composed zero-Concentrated Differential Privacy constraint, explicitly accounting for schedule selection. For expected quadratic terminal tracking, we prove that every local minimizer exhausts the certified budget, yielding an exact-budget simplex formulation. We introduce Privacy-Budget Adjoint Mirror Descent (PB-AMD), combining pathwise adjoint gradients with entropic updates that preserve positivity, charge floors, and the prescribed budget. Under regularity and coercivity assumptions, we establish existence and budget stability of optimal schedules, design-to-training utility stability, and deterministic, stochastic, and sample-average stationarity guarantees. A linear-quadratic reduction yields an explicit square-root allocation law. Experiments demonstrate tracking improvements over several allocation rules and predictive improvements over uniform allocation in a learning-rate restart setting, where learning-rate allocation performs similarly. Comparisons at a common stationarity tolerance show lower schedule-optimization times than projected gradient descent on the tested problems. The framework connects certified privacy accounting with optimization through nonlinear training dynamics.