Enhancing DPSGD via Per-Sample Momentum and Low-Pass Filtering
DOI:
https://doi.org/10.1609/aaai.v40i32.39951Abstract
Differentially Private Stochastic Gradient Descent (DPSGD) is widely used to train deep neural networks with formal privacy guarantees. However, the addition of differential privacy (DP) often degrades model accuracy by introducing both noise and bias. Existing techniques typically address only one of these issues, as reducing DP noise can exacerbate clipping bias and vice-versa. In this paper, we propose a novel method, DP-PMLF, which integrates per-sample momentum with a low-pass filtering strategy to simultaneously mitigate DP noise and clipping bias. Our approach uses per-sample momentum to smooth gradient estimates prior to clipping, thereby reducing sampling variance. It further employs a post-processing low-pass filter to attenuate high-frequency DP noise without consuming additional privacy budget. We provide a theoretical analysis demonstrating an improved convergence rate under rigorous DP guarantees, and our empirical evaluations reveal that DP-PMLF significantly enhances the privacy-utility trade-off compared to several state-of-the-art DPSGD variants.Downloads
Published
2026-03-14
How to Cite
Xu, X., Ranbaduge, T., Wang, Q., Rakotoarivelo, T., & Smith, D. (2026). Enhancing DPSGD via Per-Sample Momentum and Low-Pass Filtering. Proceedings of the AAAI Conference on Artificial Intelligence, 40(32), 27341–27349. https://doi.org/10.1609/aaai.v40i32.39951
Issue
Section
AAAI Technical Track on Machine Learning IX