arXiv:2607.05866cs.LGcs.AI2026-07

用自然梯度提升私有训练效率,实现更快收敛且更优性能。

Differentially Private Natural Gradient Descent

论文配图:Differentially Private Natural Gradient Descent
图 1 · 摘自论文原文
  • 通过解耦曲率估计与隐私数据,避免额外泄露
  • 在相同隐私预算下,收敛速度最快可达10倍提升
  • 适合对隐私与效率均有要求的模型训练场景

在固定隐私预算下,差分隐私(DP)训练的效用最终由优化效率决定。标准的一阶DP优化器如DP-SGD仅依赖局部梯度,忽略损失曲率,导致在病态优化面中严重振荡,浪费宝贵的隐私预算。实践中面临两难:提前停止训练或大幅增加每步噪声,均严重损害模型性能。自然梯度下降(NGD)通过用曲率预处理梯度,使更新方向契合损失几何,从每一步噪声中提取更高效信号,为突破隐私-效用瓶颈提供理论路径。然而直接结合NGD与DP存在根本挑战:曲率估计本身消耗巨大隐私预算,各向同性DP操作与各向异性缩放冲突,且逆曲率会放大平坦方向的参数更新,引发训练不稳定性。本文提出DP-NGD,通过解耦曲率估计、引入白化空间机制协调各向异性优化与各向同性隐私约束,并动态截断曲率以稳定训练。大量实验表明,DP-NGD在标准基准上达到当前最优准确率,在相同隐私预算下实现最高达10倍的收敛加速。

原文摘要 · Abstract (English)

Under a fixed privacy budget, the utility of differentially private (DP) training is ultimately determined by its optimization efficiency. Standard first-order DP optimizers such as DP-SGD rely solely on local gradients and ignore the underlying loss curvature. This geometric blindness causes severe zigzagging in ill-conditioned landscapes, squandering precious privacy budgets on inefficient iterations. Practitioners are thus trapped in a bind: either stop training prematurely or inject massive per-step noise, both of which critically compromise final model utility. Natural Gradient Descent (NGD) resolves this by preconditioning gradients with curvature, aligning updates with the loss geometry and extracting more efficient signal from every noisy step, offering a principled pathway to break the privacy-utility bottleneck. Despite its theoretical appeal, directly integrating NGD with DP introduces fundamental challenges: curvature estimation itself consumes prohibitive privacy budgets, isotropic DP operations conflict with the anisotropic scaling of NGD, and the inverse curvature catastrophically amplify parameter updates in flat directions, causing training instability. We propose DP-NGD, a practical framework that systematically addresses these obstacles by decoupling curvature estimation from private data, reconciling isotropic DP constraints with anisotropic second-order optimization via a whitened-space mechanism, and dynamically clamping the curvature to stabilize training. Extensive experiments on standard benchmarks demonstrate that DP-NGD achieves state-of-the-art accuracy, breaking through the utility ceilings of first-order baselines while delivering up to a $10\times$ convergence speedup under the same privacy budget.

差分隐私自然梯度优化效率

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