Privacy of Noisy Stochastic Gradient Descent: More Iterations without More Privacy Loss
Jason M. Altschuler, Kunal Talwar
Abstract
A central issue in machine learning is how to train models on sensitive user data. Industry has widely adopted a simple algorithm: Stochastic Gradient Descent with noise (a.k.a. Stochastic Gradient Langevin Dynamics). However, foundational theoretical questions about this algorithm's privacy loss remain open -- even in the seemingly simple setting of smooth convex losses over a bounded domain. Our main result resolves these questions: for a large range of parameters, we characterize the differential privacy up to a constant factor. This result reveals that all previous analyses for this setting have the wrong qualitative behavior. Specifically, while previous privacy analyses increase ad infinitum in the number of iterations, we show that after a small burn-in period, running SGD longer leaks no further privacy. Our analysis departs from previous approaches based on fast mixing, instead using techniques based on optimal transport (namely, Privacy Amplification by Iteration) and the Sampled Gaussian Mechanism (namely, Privacy Amplification by Sampling). Our techniques readily extend to other settings, e.g., strongly convex losses, non-uniform stepsizes, arbitrary batch sizes, and random or cyclic choice of batches.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext adcc8113-6d2c-48ab-933c-ef0acb519b3cCited by top-tier papers36
- Bounding training data reconstruction in DP-SGDJamie Hayes, Borja Balle, Saeed MahloujifarNeurIPS 2023 · 73 citations
- Differentially Private Learning Needs Hidden State (Or Much Faster Convergence)Jiayuan Ye, Reza ShokriNeurIPS 2022 · 62 citations
- Langevin Unlearning: A New Perspective of Noisy Gradient Descent for Machine UnlearningEli Chien, Haoyu Wang, Ziang Chen, Pan LiNeurIPS 2024 · 58 citations
- Provable Guarantees for Generative Behavior Cloning: Bridging Low-Level Stability and High-Level BehaviorAdam Block, Ali Jadbabaie, Daniel Pfrommer, Max Simchowitz et al.NeurIPS 2023 · 44 citations
- Scalable DP-SGD: Shuffling vs. Poisson SubsamplingLynn Chua, Badih Ghazi, Pritish Kamath, Ravi Kumar et al.NeurIPS 2024 · 29 citations
Builds on5
- Deep Learning with Differential PrivacyMartín Abadi, Andy Chu, Ian J. Goodfellow, H. Brendan McMahan et al.CCS 2016 · 7,620 citations
- Differential Privacy Dynamics of Langevin Diffusion and Noisy Gradient DescentRishav Chourasia, Jiayuan Ye, Reza ShokriNeurIPS 2021 · 95 citations
- Differentially Private Learning Needs Hidden State (Or Much Faster Convergence)Jiayuan Ye, Reza ShokriNeurIPS 2022 · 62 citations
- Faster Differentially Private Samplers via Rényi Divergence Analysis of Discretized Langevin MCMCArun Ganesh, Kunal TalwarNeurIPS 2020 · 44 citations
- Private stochastic convex optimization: optimal rates in linear timeVitaly Feldman, Tomer Koren, Kunal TalwarSTOC 2020 · 8 citations
Related papers
- Shifted Interpolation for Differential PrivacyJinho Bok, Weijie J. Su, Jason M. AltschulerICML 2024 · 12 citations
- An Improved Privacy and Utility Analysis of Differentially Private SGD with Bounded Domain and Smooth LossesHao Liang, Wanrong Zhang, Xinlei He, Kaishun Wu et al.AAAI 2026 · 4 citations
- Generalization of noisy SGD in unbounded non-convex settingsLeello Tadesse Dadi, Volkan CevherICML 2025
- Analyzing the Generalization Capability of SGLD Using Properties of Gaussian ChannelsHao Wang, Yizhe Huang, Rui Gao, Flávio P. CalmonNeurIPS 2021 · 32 citations
- Unified Mechanism-Specific Amplification by Subsampling and Group Privacy AmplificationJan Schuchardt, Mihail Stoian, Arthur Kosmala, Stephan GünnemannNeurIPS 2024 · 8 citations
