DPDR: Gradient Decomposition and Reconstruction for Differentially Private Deep Learning
This addresses the problem of sub-optimal performance in differentially private deep learning for researchers and practitioners, offering an incremental improvement over existing methods.
The paper tackles the inefficiency of Differentially Private Stochastic Gradient Descent (DP-SGD) by proposing DPDR, a framework that decomposes gradients into common knowledge and incremental information to focus privacy budget on the latter, resulting in improved convergence rate and accuracy compared to state-of-the-art baselines.
Differentially Private Stochastic Gradients Descent (DP-SGD) is a prominent paradigm for preserving privacy in deep learning. It ensures privacy by perturbing gradients with random noise calibrated to their entire norm at each training step. However, this perturbation suffers from a sub-optimal performance: it repeatedly wastes privacy budget on the general converging direction shared among gradients from different batches, which we refer as common knowledge, yet yields little information gain. Motivated by this, we propose a differentially private training framework with early gradient decomposition and reconstruction (DPDR), which enables more efficient use of the privacy budget. In essence, it boosts model utility by focusing on incremental information protection and recycling the privatized common knowledge learned from previous gradients at early training steps. Concretely, DPDR incorporates three steps. First, it disentangles common knowledge and incremental information in current gradients by decomposing them based on previous noisy gradients. Second, most privacy budget is spent on protecting incremental information for higher information gain. Third, the model is updated with the gradient reconstructed from recycled common knowledge and noisy incremental information. Theoretical analysis and extensive experiments show that DPDR outperforms state-of-the-art baselines on both convergence rate and accuracy.