Privacy of SGD under Gaussian or Heavy-Tailed Noise: Guarantees without Gradient Clipping

Umut Şimşekli, Mert Gürbüzbalaban, Sinan Yıldırım, Lingjiong Zhu
3/4/2024
stat.MLcs.CRcs.LG

No confident code implementation found for this paper.

Abstract

The injection of heavy-tailed noise into the iterates of stochastic gradient descent (SGD) has garnered growing interest in recent years due to its theoretical and empirical benefits for optimization and generalization. However, its implications for privacy preservation remain largely unexplored. Aiming to bridge this gap, we provide differential privacy (DP) guarantees for noisy SGD, when the injected noise follows an $α$-stable distribution, which includes a spectrum of heavy-tailed distributions (with infinite variance) as well as the light-tailed Gaussian distribution. Considering the $(ε, δ)$-DP framework, we show that SGD with heavy-tailed perturbations achieves $(0, O(1/n))$-DP for a broad class of loss functions which can be non-convex, where $n$ is the number of data points. As a remarkable byproduct, contrary to prior work that necessitates bounded sensitivity for the gradients or clipping the iterates, our theory can handle unbounded gradients without clipping, and reveals that under mild assumptions, such a projection step is not actually necessary. Our results suggest that, given other benefits of heavy-tails in optimization, heavy-tailed noising schemes can be a viable alternative to their light-tailed counterparts.

View on arXivView PDF

Code Implementations

No confident code match yet

We couldn't find an author-owned or strongly-evidenced community implementation for this paper. Any repos shown below are weak matches — verify before relying on them.

No code implementations found yet.

Know of an implementation? Let us know in the comments below!

Similar Papers

Cite this paper

@article{imekli2024privacy,
  title  = {Privacy of SGD under Gaussian or Heavy-Tailed Noise: Guarantees without Gradient Clipping},
  author = {Umut Şimşekli and Mert Gürbüzbalaban and Sinan Yıldırım and Lingjiong Zhu},
  year   = {2024},
  eprint = {2403.02051},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2403.02051v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2403.02051.svg)](https://codeofpaper.com/paper/2403.02051)

Discussion