Differentially Private Learning Needs Hidden State (Or Much Faster Convergence)
Jiayuan Ye, Reza Shokri
Abstract
Prior work on differential privacy analysis of randomized SGD algorithms relies on composition theorems, where the implicit (unrealistic) assumption is that the internal state of the iterative algorithm is revealed to the adversary. As a result, the Rényi DP bounds derived by such composition-based analyses linearly grow with the number of training epochs. When the internal state of the algorithm is hidden, we prove a converging privacy bound for noisy stochastic gradient descent (on strongly convex smooth loss functions). We show how to take advantage of privacy amplification by sub-sampling and randomized post-processing, and prove the dynamics of privacy bound for "shuffle and partition" and "sample without replacement" stochastic mini-batch gradient descent schemes. We prove that, in these settings, our privacy bound converges exponentially fast and is substantially smaller than the composition bounds, notably after a few number of training epochs. Thus, unless the DP algorithm converges fast, our privacy analysis shows that hidden state analysis can significantly amplify differential privacy. Hidden-state privacy amplification by sub-sampling. Although privacy amplification by a single sub-sampling operation is well-studied [12, 27, 7, 1, 5, 38, 30, 20, 21] , to the best of our knowledge, no prior bounds are applicable to multiple epochs of hidden-state iterative resampling. This is because
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 e29025ae-681d-4f7e-8ac4-d8d3d81421eeCited by top-tier papers31
- Privacy of Noisy Stochastic Gradient Descent: More Iterations without More Privacy LossJason M. Altschuler, Kunal TalwarNeurIPS 2022 · 89 citations
- Bounding training data reconstruction in DP-SGDJamie Hayes, Borja Balle, Saeed MahloujifarNeurIPS 2023 · 73 citations
- Langevin Unlearning: A New Perspective of Noisy Gradient Descent for Machine UnlearningEli Chien, Haoyu Wang, Ziang Chen, Pan LiNeurIPS 2024 · 58 citations
- TAN Without a Burn: Scaling Laws of DP-SGDTom Sander, Pierre Stock, Alexandre SablayrollesICML 2023 · 58 citations
- Have it your way: Individualized Privacy Assignment for DP-SGDFranziska Boenisch, Christopher Mühl, Adam Dziedzic, Roy Rinberg et al.NeurIPS 2023 · 39 citations
Builds on15
- Deep Learning with Differential PrivacyMartín Abadi, Andy Chu, Ian J. Goodfellow, H. Brendan McMahan et al.CCS 2016 · 7,620 citations
- Extracting Training Data from Large Language ModelsNicholas Carlini, Florian Tramèr, Eric Wallace, Matthew Jagielski et al.USENIX Security 2021 · 2,866 citations
- Why Gradient Clipping Accelerates Training: A Theoretical Justification for AdaptivityJingzhao Zhang, Tianxing He, Suvrit Sra, Ali JadbabaieICLR 2020 · 598 citations
- Differentially Private Learning with Adaptive ClippingGalen Andrew, Om Thakkar, Brendan McMahan, Swaroop RamaswamyNeurIPS 2021 · 425 citations
- Differentially Private Learning Needs Better Features (or Much More Data)Florian Tramèr, Dan BonehICLR 2021 · 325 citations
Related papers
- Convergent Privacy Loss of Noisy-SGD without Convexity and SmoothnessEli Chien, Pan LiICLR 2025
- Differential Privacy Dynamics of Langevin Diffusion and Noisy Gradient DescentRishav Chourasia, Jiayuan Ye, Reza ShokriNeurIPS 2021 · 95 citations
- Convex Approximation of Two-Layer ReLU Networks for Hidden State Differential PrivacyRob Romijnders, Antti KoskelaNeurIPS 2025 · 2 citations
- Shifted Interpolation for Differential PrivacyJinho Bok, Weijie J. Su, Jason M. AltschulerICML 2024 · 12 citations
- Privacy Amplification via Random Check-InsBorja Balle, Peter Kairouz, Brendan McMahan, Om Dipakbhai Thakkar et al.NeurIPS 2020 · 86 citations
