Curvature-Aware Clipping: Mitigating Bias in DPSGD via Approximate Second-Order Information
Organizations: Australian National University, Australia · CSIRO, Australia
Abstract
Differentially Private Stochastic Gradient Descent (DPSGD) has become the de facto standard for training deep neural networks with rigorous privacy guarantees. However, the utility of DPSGD is constrained by gradient clipping bias and DP noise. While existing work has tried to address these issues, the geometric misalignment caused by clipping bias remains a critical challenge. To bridge this gap, we propose a novel framework, Curvature-Aware Differentially Private Stochastic Gradient Descent (CA-DPSGD), which leverages approximate second-order information to construct an optimal, curvature-aware clipping space. CA-DPSGD integrates a rectified estimator for the curvature matrix and is coupled with an anisotropic noise mechanism to ensure privacy guarantees. Extensive empirical experiments demonstrate that our approach outperforms state-of-the-art clipping methods.
Figures & tables
| Dataset | Model | DPSGD | Auto-S | DP-PSAC | CA-DPSGD | |
|---|---|---|---|---|---|---|
| MNIST | 4-layer CNN | 98.07 0.10 | 98.13 0.04 | 98.17 0.10 | 98.32 0.12 | |
| Fashion-MNIST | 4-layer CNN | 86.45 0.42 | 86.46 0.07 | 86.56 0.13 | 86.72 0.18 | |
| CIFAR-10 | SimCLRv2 | 92.62 0.03 | 92.57 0.10 | 92.52 0.12 | 92.82 0.09 | |
| ImageNette | ResNet9 | 60.91 0.71 | 60.97 0.57 | 61.39 0.53 | 62.21 0.47 | |
| CelebA | ResNet9 | 88.49 0.13 | 88.61 0.13 | 88.55 0.12 | 88.67 0.12 | |
| MNLI | RoBERTa-base | 80.43 0.03 | 80.31 0.39 | 80.54 0.16 | 80.58 0.27 |
Appendix figures & tables4 assets
Supplementary material from the paper’s appendix.
Appendix
| Notation | Description |
|---|---|
| Privacy budget | |
| Failure factor | |
| Model parameters at iteration | |
| Dimensionality of model parameters | |
| Total number of samples in dataset | |
| Batch size |
Differentially Private Stochastic Gradient Descent
Differentially private stochastic gradient descent (DP-SGD) protects training data by clipping per-example gradients, adding calibrated noise, and accounting for privacy loss.
See all 52 papers