arXiv:2608.17415cs.CV2026-08中稿 · ECCV

通过谱正交化提升大规模私有训练的精度与稳定性。

Spectral Gradient Orthogonalization Improves Differentially Private Training at Scale

  • 用极分解对梯度进行谱正交化,从噪声中恢复信号方向。
  • 大批次下准确率提升20.9%(WRN-28-10)和14.9%(ResNet-18),方差减半至三分之二。
  • 适合高容量模型的大批量训练,可与DP-Adam媲美但内存更低。

差分隐私训练在裁剪后的梯度上添加各向同性高斯噪声,等量破坏所有奇异方向。在视觉模型中,空间相关性使梯度能量集中在低秩子空间,多数噪声落在信号较弱的方向。本文引入基于极分解的谱梯度正交化作为后处理步骤,在不增加隐私成本的前提下,从噪声梯度的低秩结构中恢复方向信号。该方法的效用由相变决定:仅当每方向的谱信噪比(SNR)足够高以恢复奇异向量时,正交化才提升精度;在低SNR情况下,梯度方向偏差被近似随机的正交更新取代,反而有害。恢复阈值由梯度的谱间隙决定,大批次下可满足。实验表明,该方法在大批次下显著提升性能:相比DP-SGD,WRN-28-10(B=4096)提升20.9%,ResNet-18提升14.9%,且运行间方差降低2~3倍。微调场景下,其稳定性媲美DP-Adam,但内存开销仅为一阶。结合时间去噪后,在CIFAR-10上达到50.3%准确率(epsilon=4),为测试中最高。这些增益仅在中高SNR场景(如大批次、高容量模型)有效,小批次或低SNR场景仍推荐使用DP-SGD或时间去噪。

原文摘要 · Abstract (English)

Differentially private training adds isotropic Gaussian noise to clipped gradients, corrupting every singular direction equally. In vision models, where spatial correlation concentrates gradient energy into a low-rank subspace, most of this noise falls in directions that carry little signal. Spectral gradient orthogonalization via polar decomposition is introduced as a post-processing step that recovers directional signal from the noisy gradient's low-rank structure at zero additional privacy cost. A phase transition governs the utility of this approach: orthogonalization improves accuracy only when the per-direction spectral signal-to-noise ratio (SNR) suffices for singular vector recovery; in low-SNR regimes, the directional bias of the gradient is replaced by a nearly random orthogonal update, and the transformation is harmful. The recovery threshold is determined by the spectral gap of the gradient and is surpassed at large batch sizes. Empirically, the benefit scales with model capacity: spectral orthogonalization achieves a +20.9% improvement over DP-SGD on WRN-28-10 (B = 4096) and +14.9% on ResNet-18, while reducing inter-run variance by a factor of two to three. In the fine-tuning regime, spectral orthogonalization matches the stability of DP-Adam while maintaining a first-order memory footprint. Combining spectral with temporal denoising yields 50.3% on CIFAR-10 (epsilon = 4), the highest accuracy in any tested configuration. These gains are specific to moderate-to-high-SNR regimes such as large-batch training of higher-capacity models. Small-batch or low-SNR settings are better served by DP-SGD or temporal denoising.

差分隐私梯度正交化大批次训练模型稳定

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