cs.LGOct 7, 2026

Closed-Form Noise Calibration Against Membership Inference for Random-Allocation DP-SGD

Authors: Murat Bilgehan Ertan, Marten van Dijk

Organizations: Affiliated with Vrije Universiteit Amsterdam.

Abstract

DP-SGD protects training data by adding Gaussian noise to clipped gradients. The amount of noise is usually chosen by running a numerical privacy accountant inside a search. We study DP-SGD with random allocation, where each epoch uses every record once, at a randomly chosen step. For this setting we give a one-line formula that bounds the accuracy of every membership inference attack (MIA) on the trained model. With MM steps per epoch, EE epochs and noise multiplier σσ, and with membership and non-membership equally likely a priori, the attack accuracy is at most 12+14(1+(e1/σ2−1)/M)E−1\frac12+\frac14\sqrt{(1+(e^{1/σ^2}-1)/M)^E-1}. The formula comes from the chi-square divergence between a Gaussian distribution and a Gaussian mixture that dominates random allocation. It is interpretable and gives σσ in about a microsecond. Where applicable, our formula needs at most about half the noise of the state-of-the-art closed-form bound. To measure how close the bound is, we also derive an exact expression for the attack accuracy of these two distributions and evaluate it numerically. Calibrating to this exact expression requires 13.0%13.0\% to 20.2%20.2\% less noise than the formula in our main experiments, and since it is exact, no accountant that knows only MM, EE and σσ can certify a smaller σσ. In training, the resulting σσ outperforms the formula and matches a published accountant in test accuracy. It is found in seconds and certified in minutes, whereas every search we ran with that accountant took longer or returned at least 0.62%0.62\% more noise. We show that MIAs on the trained models stay below the bound.

Figures & tables

Appendix figures & tables11 assets

Supplementary material from the paper’s appendix.

Appendix

Explore similar work

May 7, 2026cs.LG

Trade-off Functions for DP-SGD with Subsampling based on Random Shuffling: Tight Upper and Lower Bounds

We derive a tight analysis of the trade-off function for Differentially Private Stochastic Gradient Descent (DP-SGD) with subsampling based on random shuffling within the ff-DP framework. Our analysis covers the regime σ≥3/ln⁡Mσ\geq \sqrt{3/\ln M}, where σσ is the noise multiplier and MM is the number of rounds within a single epoch. Unlike ff-DP analyses for Poisson subsampling, which yield non-closed implicit formulas that can be machine computed but are non-transparent, random shuffling admits a tight analysis yielding transparent and interpretable closed-form bounds. Our concrete bounds, derived via the Berry-Esseen theorem, are tight up to constant factors within the proof framework. We demonstrate worked parameter settings for a single epoch (E=1E=1) with a corresponding trade-off function ≥1−a−δ\geq 1-a-δ, that is, only δδ below the ideal random guessing diagonal 1−a1-a: For δ=1/100δ= 1/100 and σ=1σ= 1, roughly M≈1.14×106M \approx 1.14\times 10^6 rounds and N≈1.14×107N \approx 1.14\times 10^7 training samples suffice to achieve meaningful differential privacy. This is in contrast to recent negative results for the regime σ≤1/2ln⁡Mσ\leq 1/\sqrt{2 \ln M}. Our concrete bounds can be composed over multiple epochs leading to δδ having a linear in EE dependency, which restricts E=O(M)E=O(\sqrt{M}). To go beyond Berry--Esseen, we introduce a new proof technique based on a generalization of the law of large numbers that yields an asymptotic random guessing diagonal-limit result: if E=cM2ME=c_M^2M with cM→0c_M\to 0, then the EE-fold composed trade-off function satisfies f⊗E(a)→1−af^{\otimes E}(a)\to 1-a uniformly in a∈[0,1]a\in[0,1] with δδ having only an O(E)O(\sqrt{E}) dependency. We compare this asymptotic regime with the corresponding Poisson subsampling asymptotic, and highlight the characterization of explicit convergence rates as an open question.
May 8, 2026cs.LG

Less Random, More Private: What is the Optimal Subsampling Scheme for DP-SGD?

Poisson subsampling is the default sampling scheme in differentially private machine learning, largely because its unstructured randomness yields tractable privacy amplification analyses. Yet this same randomness introduces substantial participation variance: each sample appears in very different numbers of training iterations. In this work, we show that this variance is not merely a practical artifact to be tolerated, but a fundamental source of suboptimal privacy amplification. We prove that Balanced Iteration Subsampling (BIS), a structured scheme in which each sample participates in exactly a fixed number of iterations, achieves stronger privacy amplification than Poisson subsampling and is optimal at both extremes of the noise spectrum (σ→0σ\to 0 and σ→∞σ\to \infty). Our analysis reveals that the privacy-noise tradeoff is governed not by maximizing randomness, but by eliminating participation variance while preserving uniform marginal participation across iterations. To translate this asymptotic theory into finite-noise guarantees, we introduce a practical near-exact Monte Carlo accountant for BIS, which removes the analytical slack of existing RDP and composition-based PLD analyses. Evaluations across more than 60 practical DP-SGD configurations show that BIS consistently outperforms Poisson subsampling in the low-noise regimes most relevant for high-utility private training, reducing the required noise multiplier by up to 9.6%9.6\%. These results overturn the common intuition that more sampling randomness necessarily yields stronger privacy amplification: in DP-SGD, structured participation can be both more practical and more private. Our implementation is available at https://github.com/dong-xin-ao-andy/bis-mc-accountant.
May 25, 2026cs.LG

From Privacy to Generalization: Linear Max-Information Bounds for Differentially Private Learning Algorithms

Understanding the relationship between generalization and privacy remains a challenge in modern machine learning theory, particularly for deep networks that are trained by variants of differentially private stochastic gradient descent (DP- SGD). In this work we make progress on this persistent open problem. First, we derive explicit upper bounds on the approximate max-information of any algorithm that fulfills (ε,δ)(ε, δ)-differential privacy or Rényi differential privacy, thereby going beyond the classical results for pure εε-differential privacy. Subsequently, we show even stronger guarantees for two common private learning algorithms, output perturbation with the Gaussian mechanism, and streaming DP-SGD, by exploiting the structure of their internal randomization. As an application of our results, we demonstrate how to obtain non-vacuous PAC-Bayes generalization bounds for deep networks, in which the prior distribution is learned by DP-SGD instead of the classical way of choosing it in a data-independent way.