Enhancing DPSGD via Per-Sample Momentum and Low-Pass Filtering

Authors

  • Xincheng Xu School of Computing, Australian National University, Australia
  • Thilina Ranbaduge Data 61, CSIRO, Australia
  • Qing Wang School of Computing, Australian National University, Australia
  • Thierry Rakotoarivelo Data 61, CSIRO, Australia
  • David Smith Data 61, CSIRO, Australia

DOI:

https://doi.org/10.1609/aaai.v40i32.39951

Abstract

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.

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