Differential Private Stochastic Optimization with Heavy-tailed Data: Towards Optimal Rates

Authors

  • Puning Zhao Sun Yat-sen University
  • Jiafei Wu Zhejiang Lab
  • Zhe Liu Zhejiang Lab
  • Chong Wang Ningbo University
  • Rongfei Fan Beijing Institute of Technology
  • Qingming Li Zhejiang University

DOI:

https://doi.org/10.1609/aaai.v39i21.34440

Abstract

We study convex optimization problems under differential privacy (DP). With heavy-tailed gradients, existing works achieve suboptimal rates. The main obstacle is that existing gradient estimators have suboptimal tail property, resulting in a superfluous factor of d in the union bound. In this paper, we explore algorithms achieving optimal rates of DP optimization with heavy-tailed gradients. Our first method is a simple clipping approach. Under bounded p-th order moments of gradients, with n samples, it achieves minimax optimal population risk with epsilon less than 1/d. We then propose an iterative updating method, which is more complex but achieves this rate for all epsilon smaller than 1. The results significantly improve over existing methods. Such improvement relies on a careful treatment of the tail behavior of gradient estimators. Our results match the minimax lower bound, indicating that the theoretical limit of stochastic convex optimization under DP is achievable.

Downloads

Published

2025-04-11

How to Cite

Zhao, P., Wu, J., Liu, Z., Wang, C., Fan, R., & Li, Q. (2025). Differential Private Stochastic Optimization with Heavy-tailed Data: Towards Optimal Rates. Proceedings of the AAAI Conference on Artificial Intelligence, 39(21), 22795–22803. https://doi.org/10.1609/aaai.v39i21.34440

Issue

Section

AAAI Technical Track on Machine Learning VII