Introduction
Suppose a learner receives samples from an unknown probability distribution p p p , which is known to belong to some class of probability distributions Q \mathcal{Q} Q . Can the learner identify a distribution p ^ \hat{p} p ^ , such that p ^ \hat{p} p ^ is close to the distribution p p p in total variation distance? This fundamental question is known as distribution learning or density estimation , and has seen significant study across a variety of adjacent research communities including computer science, statistics, and information theory (see, for example, [KMR+94, DL01, DDS12, Dia16, ABDH+20, LBD26]).
The problem of density estimation has also been studied extensively under the constraint of differential privacy (DP) [27 ] . Indeed, there is a rich body of literature on private learning of various classes of distributions. Some examples are private density estimation of structured univariate distributions [23 , 38 ] , high-dimensional Gaussians [32 , 17 , 15 , 2 , 7 KMV22, 34 , 6 , 29 ] , Gaussian Mixture Models [37 , 1 , 13 , 3 , 9 , 4 ] , and product distributions [32 , 17 , 45 ] .
One vibrant program within this area has focused on designing increasingly general and generic frameworks for learning distribution classes (see, for example, [17 , 7 , 29 , 12 , 3 , 4 ] ). At their core, many of these methods rely upon non-private learning algorithms as a black box, applying DP techniques on their outputs to design private algorithms. At a July 2022 workshop at the Fields Institute, Ashtiani conjectured that every learnable distribution class could be privately learned [Ash22].
Conjecture 1.1 (Ashtiani’s Conjecture). Every distribution class Q \mathcal{Q} Q that is learnable with respect to total variation distance is also learnable under the additional constraint of ( ε , δ ) (\varepsilon,\delta) ( ε , δ ) -differential privacy.
Our main result (Theorem 1.3 ) is a positive resolution to this conjecture: given a non-private algorithm for learning a distribution class up to arbitrarily small target error in total variation distance, we provide a private algorithm for learning the same distribution class.
A recent work by Bun, Kamath, Mouzakis, and Singhal [16 ] claimed to refute Ashtiani’s conjecture, which may seem to contradict our result. In fact, their work can be viewed as a weak refutation of Ashtiani’s conjecture. They constructed a distribution class which can be learned up to constant error in total variation distance, but cannot be learned privately to that same error. The distribution class in their counter-example, however, is not (non-privately) learnable to all values of error. In contrast, we use the traditional definition of learnability, wherein the distribution class must be learnable to arbitrarily small error [36 ] . Indeed, our private algorithm relies crucially upon non-private learning of the class to a higher accuracy than the target.
Our positive resolution of the conjecture may come as a surprise as private and non-private learnability are not necessarily equivalent for problems beyond density estimation. Indeed, in the standard binary classification setting, non-private learnability does not imply ( ε , δ ) (\varepsilon,\delta) ( ε , δ ) -DP learnability [19 , 8 ] . It is also remarkable that for (non-private) density estimation, a nice characterization of learnability is not known (and in some sense impossible [39 , 40 ] ) while our private to non-private reduction works anyway.
Results
In what follows, let X \mathcal{X} X be the domain of examples, which we assume is a standard Borel space, and denote by Δ ( X ) \Delta(\mathcal{X}) Δ ( X ) the set of all probability distributions on X \mathcal{X} X . Any Q ⊆ Δ ( X ) \mathcal{Q} \subseteq\Delta(\mathcal{X}) Q ⊆ Δ ( X ) is referred to as a distribution class . Denote by T V \mathrm{TV} TV the total variation distance: for p , q ∈ Δ ( X ) p,q \in\Delta(\mathcal{X}) p , q ∈ Δ ( X ) , T V ( p , q ) : = sup A ∈ B ∣ p ( A ) − q ( A ) ∣ \mathrm{TV}(p,q) := \sup_{A \in\mathcal{B}} \lvert p(A) - q(A) \rvert TV ( p , q ) := sup A ∈ B ∣ p ( A ) − q ( A )∣ where B \mathcal{B} B are the measurable sets of X \mathcal{X} X .
Definition 1.2 (Distribution learning). A distribution class Q ⊆ Δ ( X ) \mathcal{Q} \subseteq\Delta(\mathcal{X}) Q ⊆ Δ ( X ) is learnable if there exist a learner A : X ∗ → Δ ( X ) A : \mathcal{X}^{*} \to\Delta(\mathcal{X}) A : X ∗ → Δ ( X ) and a function m : ( 0 , 1 ) 2 → N m : (0,1)^{2} \to\mathbb{N} m : ( 0 , 1 ) 2 → N such that, for every α , β ∈ ( 0 , 1 ) \alpha,\beta\in(0,1) α , β ∈ ( 0 , 1 ) , every n ≥ m ( α , β ) n \ge m(\alpha,\beta) n ≥ m ( α , β ) , and every p ∈ Q p \in\mathcal{Q} p ∈ Q ,
Pr S ∼ p n [ T V ( A ( S ) , p ) ≥ α ] ≤ β . \Pr_{S \sim p^{n}}[\mathrm{TV}(A(S),p) \ge\alpha] \le\beta. S ∼ p n Pr [ TV ( A ( S ) , p ) ≥ α ] ≤ β . The sample complexity of learning Q \mathcal{Q} Q is the pointwise infimum of all functions m m m for which there exists a learner A A A satisfying the above guarantee.
Theorem 1.3 (Learnable classes are privately learnable; informal version of Theorem 6.1). Suppose Q ⊆ Δ ( X ) \mathcal{Q} \subseteq\Delta(\mathcal{X}) Q ⊆ Δ ( X ) is learnable with sample complexity m ( α , β ) m(\alpha,\beta) m ( α , β ) for α , β ∈ ( 0 , 1 ) \alpha,\beta\in(0,1) α , β ∈ ( 0 , 1 ) . Then, for every ε , δ ∈ ( 0 , 1 ) \varepsilon,\delta\in(0,1) ε , δ ∈ ( 0 , 1 ) , Q \mathcal{Q} Q is also learnable under ( ε , δ ) (\varepsilon,\delta) ( ε , δ ) -differential privacy, with sample complexity n ( α , β , ε , δ ) = n(\alpha,\beta,\varepsilon,\delta) = n ( α , β , ε , δ ) = O ( m ( 1 512 m ( α / 3 , 1 / 32 ) , 1 16 ) 1 ε log ( 4 δ ) log ( 4 β ) ) O\left(m\left(\frac{1}{512m(\alpha/3,1/32)},\frac{1}{16}\right)\frac{1}{\varepsilon}\log\left(\frac{4}{\delta}\right)\log\left(\frac{4}{\beta}\right)\right) O ( m ( 512 m ( α /3 , 1/32 ) 1 , 16 1 ) ε 1 log ( δ 4 ) log ( β 4 ) ) .
In particular, if a class Q \mathcal{Q} Q is learnable with sample complexity m ( α , β ) = poly ( 1 α , log 1 β ) m(\alpha,\beta) = \operatorname{poly}\left(\frac{1}{\alpha},\log\frac{1}{\beta}\right) m ( α , β ) = poly ( α 1 , log β 1 ) , our bound implies that Q \mathcal{Q} Q is ( ε , δ ) (\varepsilon,\delta) ( ε , δ ) -DP learnable with sample complexity m ( α , β , ε , δ ) = poly ( 1 α , log 1 β , 1 ε , log 1 δ ) m(\alpha,\beta,\varepsilon,\delta) = \operatorname{poly}\left(\frac{1}{\alpha},\log\frac{1}{\beta},\frac{1}{\varepsilon},\log\frac{1}{\delta}\right) m ( α , β , ε , δ ) = poly ( α 1 , log β 1 , ε 1 , log δ 1 ) .
As part of our approach, we develop private samplers [44 , 21 ] for any learnable class, which can be of independent interest. These samplers receive private data points and output a synthetic data set, generated from a distribution that is close to the underlying (unknown) distribution.
Definition 1.4 (Approximate multi-sampler). Let Q ⊆ Δ ( X ) Q \subseteq\Delta(\mathcal{X}) Q ⊆ Δ ( X ) , ℓ ∈ N \ell\in\mathbb{N} ℓ ∈ N , γ ∈ ( 0 , 1 ) \gamma\in(0,1) γ ∈ ( 0 , 1 ) . A randomized algorithm M : X n → X ℓ M : \mathcal{X}^{n} \to\mathcal{X}^{\ell} M : X n → X ℓ is a ( ℓ , γ ) (\ell,\gamma) ( ℓ , γ ) -sampler for Q Q Q with sample complexity n n n if for every p ∈ Q p \in Q p ∈ Q ,
TV ( L ( Z ) , L ( Z ′ ) ) ≤ γ , \operatorname{TV}(\mathcal{L}(Z),\mathcal{L}(Z')) \leq\gamma, TV ( L ( Z ) , L ( Z ′ )) ≤ γ , where Z ∼ p ℓ Z \sim p^{\ell} Z ∼ p ℓ , Z ′ = M ( S ) Z' = M(S) Z ′ = M ( S ) with S ∼ p n S \sim p^{n} S ∼ p n , and L ( ⋅ ) \mathcal{L}(\cdot) L ( ⋅ ) corresponds to the law of the random variable.
Theorem 1.5 (Differentially private synthetic data for learnable classes; informal version of Theorem 4.3 ). Let Q ⊆ Δ ( X ) Q \subseteq\Delta(\mathcal{X}) Q ⊆ Δ ( X ) be learnable with sample complexity m ( α , β ) m(\alpha,\beta) m ( α , β ) . For every ℓ ∈ N \ell\in\mathbb{N} ℓ ∈ N and γ , ε , δ ∈ ( 0 , 1 ) \gamma,\varepsilon,\delta\in(0,1) γ , ε , δ ∈ ( 0 , 1 ) , there exists a ( ε , δ ) (\varepsilon,\delta) ( ε , δ ) -differentially private ( ℓ , γ ) (\ell,\gamma) ( ℓ , γ ) -sampler for Q Q Q with sample complexity
n = O ( m ( γ 64 ℓ , 1 16 ) log ( 1 γ δ ) 1 ε ) . n = O\left(m\left(\frac{\gamma}{64\ell},\frac{1}{16}\right)\log\left(\frac{1}{\gamma\delta}\right)\frac{1}{\varepsilon}\right). n = O ( m ( 64 ℓ γ , 16 1 ) log ( γ δ 1 ) ε 1 ) .
For the problem of (non-private) density estimation, learnability in realizable setting is not equivalent to learnability in agnostic/robust setting [14 ] . Our next result is a private to non-private reduction for agnostic learning of distributions. The formal version and more discussions appear in Section C.
Theorem 1.6 (Agnostic learnability implies private agnostic learnability, Informal version of C.2). Suppose Q ⊆ Δ ( X ) Q \subseteq\Delta(\mathcal{X}) Q ⊆ Δ ( X ) is “agnostic” learnable with sample complexity m ( α , β ) m(\alpha,\beta) m ( α , β ) for α , β ∈ ( 0 , 1 ) \alpha,\beta\in(0,1) α , β ∈ ( 0 , 1 ) . Then, for every ε , δ ∈ ( 0 , 1 ) \varepsilon,\delta\in(0,1) ε , δ ∈ ( 0 , 1 ) , Q Q Q is also agnostic learnable under ( ε , δ ) (\varepsilon,\delta) ( ε , δ ) -differential privacy, with sample complexity
n ( α , β , ε , δ ) = O ( m ( Θ ( α ) , 1 16 ) m ( Θ ( α ) , β ) 1 ε log ( 1 α δ ) ) . n(\alpha,\beta,\varepsilon,\delta) = O\left(m\left(\Theta(\alpha),\frac{1}{16}\right)m\left(\Theta(\alpha),\beta\right)^{\frac{1}{\varepsilon}}\log\left(\frac{1}{\alpha\delta}\right)\right). n ( α , β , ε , δ ) = O ( m ( Θ ( α ) , 16 1 ) m ( Θ ( α ) , β ) ε 1 log ( α δ 1 ) ) .
Techniques
At its core, our algorithm falls into the subsample-and-aggregate framework [43 ] . Suppose we have a non-private learning algorithm A A A , which learns the distribution class Q Q Q in total variation distance. Consider the following intuitive yet naive approach:
Partition the dataset into k = O ( log ( 1 / δ ) / ε ) k = O(\log(1/\delta)/\varepsilon) k = O ( log ( 1/ δ ) / ε ) parts;
Run the non-private learning algorithm A A A on each part;
Privately aggregate the outputs.
Let q i q_i q i for i ∈ [ k ] i \in[k] i ∈ [ k ] be the output of the non-private learning algorithm on the i th i^{\text{th}} i th part of the dataset. If many of q i q_i q i ’s are exactly the same (i.e., the learner is globally stable [18 ] ), then they can be easily aggregated, for example, using DP stability-based mode [Vad17]. However, the non-private learner A A A only guarantees that, with high probability, most of q i q_i q i ’s are close to the ground truth p p p in TV, thus to one another. In other words, non-private density estimators may not be globally stable.
Another idea for aggregating q i q_i q i ’s is to first generate samples from each q i q_i q i , and then privately aggregate these samples. Assume Z i ∼ q i ℓ Z_i \sim q_i^{\ell} Z i ∼ q i ℓ is a sample of size ℓ \ell ℓ from q i q_i q i . If we are lucky and many of these Z i Z_i Z i ’s “collide” (take the exact same value for different choices of i i i ), then again we can privately aggregate them and will end up having a private sample set (synthetic data) from a distribution that is close to the ground truth distribution p p p . These public samples can then be fed to the non-private learner to find an approximation to p p p . In first glance, introducing the additional step of sampling from q i q_i q i ’s appears counterproductive. In fact, now Z i Z_i Z i ’s may not collide not only because q i q_i q i ’s are not colliding, but also because of the randomness of generating Z i Z_i Z i ’s. Indeed, this would be the case if Z i Z_i Z i ’s are generated independently for each i i i . However, as we discuss below, allowing Z i Z_i Z i ’s to be correlated (across values of i i i ) gives an extra flexibility that proves to be crucial for stabilizing the outcomes of non-private learners.
To make the idea concrete we turn to couplings . Couplings construct a joint distribution over a collection of random variables such that their marginal distributions are preserved. The key observation is that a suitable TV coupling can convert closeness in total variation distance into a high probability guarantee of exact equality between sampled outcomes. Specifically, after obtaining q 1 , … , q k q_1,\ldots,q_k q 1 , … , q k non privately, we jointly draw ( Z 1 , … , Z k ) (Z_1,\ldots,Z_k) ( Z 1 , … , Z k ) from a TV coupling of these distributions satisfying Definition 3.1 . The coupling is designed to satisfy two key desiderata: (1) each marginal has the correct law, so that Z i ∼ q i ℓ Z_i \sim q_i^\ell Z i ∼ q i ℓ is a valid synthetic dataset from the learner output on the i t h i^{\mathrm{th}} i th block, and (2) assuming q i q_i q i ’s are all close in total variation distance, synthetic datasets Z i Z_i Z i ’s would take the exact same value with high probability. Thus, a TV coupling can convert total variation closeness of the q i q_i q i ’s into the sufficiently many exact repetitions of the Z i Z_i Z i ’s needed by the stability-based mode.
Indeed, the use of correlated sampling to convert closeness in TV into agreement of sampled outcomes already appears in the user-level private learning algorithms of Ghazi, Kumar, and Manurangsi [28 ] . Kalavasis, Karbasi, Moran, and Velegkas [KKMV23] subsequently use a related coupling approach to reduce TV-indistinguishable PAC learning to approximate differentially private learning. A similar concept has been used in [CCWL+26] for private aggregation of teacher models. In our setting, we apply this principle to synthetic datasets.
Our constructions of a couplings are based on the Poisson coupling of Angel and Spinka [10 ] , which ensures that for all i , j ∈ [ k ] i,j \in[k] i , j ∈ [ k ] simultaneously, Pr [ Z i ≠ Z j ] ≤ 2 TV ( q i ℓ , q j ℓ ) \Pr[Z_i \ne Z_j] \le2\operatorname{TV}(q_i^\ell,q_j^\ell) Pr [ Z i = Z j ] ≤ 2 TV ( q i ℓ , q j ℓ ) . When all relevant distributions admit densities with respect to a common σ \sigma σ -finite reference measure, the Poisson coupling of Angel and Spinka [10 ] applies directly, as in [KKMV23]. This already covers many familiar distribution classes. For our main result, however, we must handle arbitrary learnable distribution classes, for which such a common dominating measure need not exist. For instance, P = { δ x : x ∈ R } \mathcal{P} = \{\delta_x : x \in\mathbb{R}\} P = { δ x : x ∈ R } is trivially learnable with a single sample but there is no σ \sigma σ -finite measure dominating all point masses. We, therefore, develop a more general version of the Poisson coupling that removes the common-dominating-measure assumption while retaining the same marginal and pairwise agreement guarantees on general standard Borel spaces.
The coupling methods allow us to turn closeness of TV distances between q i q_i q i ’s to agreement probability lower bounds between Z i Z_i Z i ’s. Private aggregation of Z i Z_i Z i ’s then gives the private multi-sampler of Theorem 1.5 . In order to have a private distribution learner, one can simply feed a privatized sample set to the non-private learner. However, this approach would result in a poor dependency on β \beta β , the failure probability. To overcome this issue, we use a boosting argument.
This results in the logarithmic dependence of sample complexity on 1 / β 1/\beta 1/ β in Theorem 1.3 .
AI Disclosure
The initial resolution for Ashtiani’s conjecture in the case of countable domains was independently obtained by two authors prompting GPT-5.6 Sol. An internal Gemini model also produced a positive resolution of the conjecture by discovering a similar algorithm to the one presented. All authors digested the result and verified it to be correct.
Additional prompting, verification, and derivations from the authors resulted in the presented measure-theoretic arguments that constitute a full resolution to the conjecture for general domains. Care was taken to write the paper in a way that communicates the ideas most clearly to a human reader. The authors also conducted a literature search to attribute the main algorithmic ideas to [10 ] and [KKMV23]. All writing in this paper is done by the authors, who take full responsibility for the claims made herein.
Preliminaries
Let Y \mathcal{Y} Y be a standard Borel space, and let B ( Y ) \mathcal{B}(\mathcal{Y}) B ( Y ) denote the Borel sets and Δ ( Y ) \Delta(\mathcal{Y}) Δ ( Y ) the probability measures on Y \mathcal{Y} Y , with the σ \sigma σ -algebra generated by the map P ↦ P ( B ) P \mapsto P(B) P ↦ P ( B ) for B ∈ B ( Y ) B \in\mathcal{B}(\mathcal{Y}) B ∈ B ( Y ) . Subsets and finite products carry the induced and product σ \sigma σ -algebras, respectively.
For a random variable Z Z Z , we write L ( Z ) \mathcal{L}(Z) L ( Z ) for its law, i.e., L ( Z ) ( B ) = Pr [ Z ∈ B ] \mathcal{L}(Z)(B) = \Pr[Z \in B] L ( Z ) ( B ) = Pr [ Z ∈ B ] for every measurable set B B B . For P , Q ∈ Δ ( Y ) P,Q \in\Delta(\mathcal{Y}) P , Q ∈ Δ ( Y ) , their total variation distance is TV ( P , Q ) : = sup B ∈ B ( Y ) ∣ P ( B ) − Q ( B ) ∣ \operatorname{TV}(P,Q) := \sup_{B \in\mathcal{B}(\mathcal{Y})} |P(B)-Q(B)| TV ( P , Q ) := sup B ∈ B ( Y ) ∣ P ( B ) − Q ( B ) ∣ .
Our coupling constructions use two standard measure-theoretic objects. First, we will use Markov kernels to formalize randomized mappings whose output distributions depend measurably on their inputs. In particular, they allow us to view our coupling procedures as randomized maps from tuples of distributions to joint synthetic datasets.
Definition 2.1 (Markov kernel). Let ( X , A ) (\mathcal{X},\mathcal{A}) ( X , A ) and ( Y , B ) (\mathcal{Y},\mathcal{B}) ( Y , B ) be measurable spaces. A Markov kernel K : X ⇝ Y K : \mathcal{X} \rightsquigarrow\mathcal{Y} K : X ⇝ Y assigns to each x ∈ X x \in\mathcal{X} x ∈ X a probability measure K ( x ; ⋅ ) K(x;\mathord\cdot) K ( x ; ⋅ ) on Y \mathcal{Y} Y , such that, for every B ∈ B B \in\mathcal{B} B ∈ B , the map x ↦ K ( x ; B ) x \mapsto K(x;B) x ↦ K ( x ; B ) is A \mathcal{A} A -measurable.
Second, our explicit coupling constructions are based on the Poisson coupling of Angel and Spinka [10 ] , which is built from Poisson point processes. We recall their basic definition below.
Informally, a Poisson point process with intensity measure μ \mu μ is a random cloud of points scattered over the space according to μ \mu μ , where the expected number of points in a region B B B is μ ( B ) \mu(B) μ ( B ) , and the number of points falling in disjoint regions are independent.
Definition 2.2 (Poisson point process). Let ( Y , B ) (\mathcal{Y},\mathcal{B}) ( Y , B ) be a measurable space and let μ \mu μ be a σ \sigma σ -finite measure on Y \mathcal{Y} Y . A Poisson point process A \mathcal{A} A on Y \mathcal{Y} Y with intensity measure μ \mu μ is a random counting measure such that:
for every measurable B ⊆ Y B \subseteq\mathcal{Y} B ⊆ Y with μ ( B ) < ∞ \mu(B) < \infty μ ( B ) < ∞ , A ( B ) ∼ Poisson ( μ ( B ) ) \mathcal{A}(B) \sim\operatorname{Poisson}(\mu(B)) A ( B ) ∼ Poisson ( μ ( B ))
for every collection of pairwise disjoint measurable sets B 1 , … , B m B_1,\ldots,B_m B 1 , … , B m , the random variables A ( B 1 ) , … , A ( B m ) \mathcal{A}(B_1),\ldots,\mathcal{A}(B_m) A ( B 1 ) , … , A ( B m ) are independent.
We recall the celebrated definition of differential privacy, as well as its useful post-processing and parallel composition property.
Definition 2.3 (Differential privacy (DP) [27 , 25 ] ). Let ( X , A ) (\mathcal{X},\mathcal{A}) ( X , A ) and ( Y , B ) (\mathcal{Y},\mathcal{B}) ( Y , B ) be measurable spaces, and let ϵ > 0 \epsilon> 0 ϵ > 0 and δ ∈ ( 0 , 1 ) \delta\in(0,1) δ ∈ ( 0 , 1 ) . A randomized algorithm A : X ∗ → Y A : \mathcal{X}^{*} \to\mathcal{Y} A : X ∗ → Y is ( ϵ , δ ) (\epsilon,\delta) ( ϵ , δ ) -differentially private if, for every n ∈ N n \in\mathbb{N} n ∈ N , every pair of neighboring datasets S , S ′ ∈ X n S,S' \in\mathcal{X}^{n} S , S ′ ∈ X n differing in at most one element, and every measurable set B ⊆ Y B \subseteq\mathcal{Y} B ⊆ Y , we have
Pr [ A ( S ) ∈ B ] ≤ e ϵ Pr [ A ( S ′ ) ∈ B ] + δ \Pr[A(S) \in B] \le e^{\epsilon}\Pr[A(S') \in B] + \delta Pr [ A ( S ) ∈ B ] ≤ e ϵ Pr [ A ( S ′ ) ∈ B ] + δ where the probabilities are over the internal randomness of A A A .
Proposition 2.4 (Post-processing). Let ( X , A ) (\mathcal{X},\mathcal{A}) ( X , A ) , ( Y , B ) (\mathcal{Y},\mathcal{B}) ( Y , B ) , and ( Z , C ) (\mathcal{Z},\mathcal{C}) ( Z , C ) be measurable spaces. Suppose A : X ∗ → Y A : \mathcal{X}^{*} \to\mathcal{Y} A : X ∗ → Y is ( ϵ , δ ) (\epsilon,\delta) ( ϵ , δ ) -differentially private, and let K : Y ⇝ Z K : \mathcal{Y} \rightsquigarrow\mathcal{Z} K : Y ⇝ Z be a data-independent Markov kernel. Then the randomized algorithm K ∘ A : S ↦ Z K \circ A : S \mapsto Z K ∘ A : S ↦ Z with Z ∼ K ( A ( S ) ; ⋅ ) Z \sim K(A(S);\mathord\cdot) Z ∼ K ( A ( S ) ; ⋅ ) is also ( ϵ , δ ) (\epsilon,\delta) ( ϵ , δ ) -DP.
Lemma 2.5 (Parallel composition [41 ] ). Let ϵ ≥ 0 \epsilon\ge0 ϵ ≥ 0 , δ ∈ [ 0 , 1 ) \delta\in[0,1) δ ∈ [ 0 , 1 ) , and let I 1 , … , I k I_1,\ldots,I_k I 1 , … , I k be a fixed partition of [ n ] [n] [ n ] . For each i ∈ [ k ] i \in[k] i ∈ [ k ] , suppose A i : X ∗ → Y A_i : \mathcal{X}^{*} \to\mathcal{Y} A i : X ∗ → Y is ( ϵ , δ ) (\epsilon,\delta) ( ϵ , δ ) -differentially private. Then, for any S ∈ X n S \in\mathcal{X}^{n} S ∈ X n , the algorithm A ( S ) : = ( A 1 ( S I 1 ) , … , A k ( S I k ) ) A(S) := (A_1(S_{I_1}),\ldots,A_k(S_{I_k})) A ( S ) := ( A 1 ( S I 1 ) , … , A k ( S I k )) , where the invocations use independent randomness, is ( ϵ , δ ) (\epsilon,\delta) ( ϵ , δ ) -DP.
Couplings
As discussed in § 1.2 , our private aggregation procedure requires a correlated sampling mechanism that converts TV closeness of the learner outputs into exact agreement of the sampled synthetic datasets. We now formalize the properties of the coupling needed for the subsequent privacy and utility analyses. In addition to having the correct marginals and a small pairwise disagreement probability discussed above, we further require the joint law of every finite collection of coupled samples to depend measurably on the input distributions. This ensures that the coupling can be applied to random data-dependent learner outputs and then supplied to subsequent randomized algorithms.1 Throughout, we assume the domain Y \mathcal{Y} Y to be standard Borel.
Definition 3.1 (Grand coupling scheme). Let Y \mathcal{Y} Y be standard Borel and P ⊆ Δ ( Y ) \mathcal{P} \subseteq\Delta(\mathcal{Y}) P ⊆ Δ ( Y ) . A grand coupling scheme on P \mathcal{P} P is a family of random variables ( Z P ) P ∈ P (Z_P)_{P\in\mathcal{P}} ( Z P ) P ∈ P , on a common probability space, with joint law Γ P \Gamma_{\mathcal{P}} Γ P , such that:
(G1) (Marginals.) Z P ∼ P Z_P \sim P Z P ∼ P for every P ∈ P P \in\mathcal{P} P ∈ P .
(G2) (Agreement.) Pr [ Z P ≠ Z Q ] ≤ 2 TV ( P , Q ) \Pr[Z_P \ne Z_Q] \le2\operatorname{TV}(P,Q) Pr [ Z P = Z Q ] ≤ 2 TV ( P , Q ) for all P , Q ∈ P P,Q \in\mathcal{P} P , Q ∈ P .
(G3) (Measurable marginal kernels.) For every k ≥ 1 k \ge1 k ≥ 1 , the map P ↦ L ( Z P 1 , … , Z P k ) \mathbf{P} \mapsto\mathcal{L}(Z_{P_1},\ldots,Z_{P_k}) P ↦ L ( Z P 1 , … , Z P k ) defines a Markov kernel Γ k : P k ⇝ Y k \Gamma_k:\mathcal{P}^k \rightsquigarrow\mathcal{Y}^k Γ k : P k ⇝ Y k , where P = ( P 1 , … , P k ) ∈ P k \mathbf{P}=(P_1,\ldots,P_k)\in\mathcal{P}^k P = ( P 1 , … , P k ) ∈ P k .
Couple k ( P ) \operatorname{Couple}_k(\mathbf{P}) Couple k ( P ) is then defined as a sample from Γ k ( P ; ⋅ ) \Gamma_k(\mathbf{P};\cdot) Γ k ( P ; ⋅ ) .
We present two constructions of this interface, based on the Poisson coupling of [10 ] . Full details and complete proofs for both constructions are deferred to Appendix § 7 . The first, presented in Appendix § 7 .§ 1 , is the original construction taken verbatim from [10 ] , which applies when the subset of relevant distributions P \mathcal{P} P admits a common σ \sigma σ -finite dominating measure ν \nu ν .
We illustrate how Construction A.1 works with a simple example. Consider a set of distributions P \mathcal{P} P on domain [ 0 , 1 ] [0,1] [ 0 , 1 ] , where for each distribution P ∈ P P \in\mathcal{P} P ∈ P , we have density f P f_P f P with respect to the Lebesgue measure bounded by a constant M M M . Draw an infinite sequence of i.i.d. pairs ( Y i , U i ) (Y_i,U_i) ( Y i , U i ) , distributed uniformly over [ 0 , 1 ] × [ 0 , M ] [0,1] \times[0,M] [ 0 , 1 ] × [ 0 , M ] . This sequence of pairs is shared across all P ∈ P P \in\mathcal{P} P ∈ P . To draw a sample from a particular P P P , we output the Y i Y_i Y i with the minimum i i i such that U i ≤ f P ( Y i ) U_i \le f_P(Y_i) U i ≤ f P ( Y i ) . Intuitively, to sample from a distribution P P P , this process outputs the first coordinate of the first point that lies under P P P ’s density. By a standard rejection sampling argument, the resulting output Z P Z_P Z P has law P P P , giving the marginal property (G1). For P , Q ∈ P P,Q \in\mathcal{P} P , Q ∈ P , the areas of the intersection and union of their acceptance regions are 1 − TV ( P , Q ) 1-\operatorname{TV}(P,Q) 1 − TV ( P , Q ) and 1 + TV ( P , Q ) 1+\operatorname{TV}(P,Q) 1 + TV ( P , Q ) , respectively. Hence, the first proposal accepted by either distribution is accepted by both with probability 1 − TV ( P , Q ) 1 + TV ( P , Q ) \frac{1-\operatorname{TV}(P,Q)}{1+\operatorname{TV}(P,Q)} 1 + TV ( P , Q ) 1 − TV ( P , Q ) , and thus yields the disagreement property (G2), Pr [ Z P ≠ Z Q ] ≤ 2 TV ( P , Q ) 1 + TV ( P , Q ) ≤ 2 TV ( P , Q ) \Pr[Z_P \ne Z_Q] \le\frac{2\operatorname{TV}(P,Q)}{1+\operatorname{TV}(P,Q)} \le2\operatorname{TV}(P,Q) Pr [ Z P = Z Q ] ≤ 1 + TV ( P , Q ) 2 TV ( P , Q ) ≤ 2 TV ( P , Q ) .
More precisely, Construction A.1 yields a grand coupling scheme on P \mathcal{P} P , using a single Poisson point process shared across all P ∈ P P \in\mathcal{P} P ∈ P . Each P ∈ P P \in\mathcal{P} P ∈ P defines its own acceptance region through its density d P d ν \frac{\mathrm{d}P}{\mathrm{d}\nu} d ν d P , and the coupled sample Z P Z_P Z P is the y y y -coordinate of the first point of the shared Poisson process that falls in that region, i.e., the point in the region with the smallest t t t -coordinate.
Proposition 3.2 (Informal version of Corollary A.3). Suppose Y \mathcal{Y} Y is standard Borel and there exists a σ \sigma σ -finite measure ν \nu ν such that P ≪ ν P \ll\nu P ≪ ν for all P ∈ P P \in\mathcal{P} P ∈ P . Then the family of random variables ( Z P ) P ∈ P (Z_P)_{P\in\mathcal{P}} ( Z P ) P ∈ P defined in Construction A.1 yields a grand coupling scheme Γ P \Gamma_{\mathcal{P}} Γ P on P \mathcal{P} P .
Construction A.1 already captures a wide range of distribution classes. For example, a family of distributions on R d \mathbb{R}^{d} R d that are all absolutely continuous with respect to Lebesgue measure admits a common σ \sigma σ -finite dominating measure. However, an arbitrary learnable distribution class need not admit such a measure. For instance, P = { δ x : x ∈ R } \mathcal{P} = \{\delta_x : x \in\mathbb{R}\} P = { δ x : x ∈ R } is trivially learnable with a single sample but there is no σ \sigma σ -finite measure dominating all point masses. Therefore, the first construction can not handle such classes.
The key observation behind our second construction is that while a global measure ν \nu ν may not exist over an arbitrary family, any finite tuple of distributions P = ( P 1 , … , P k ) ∈ Δ ( Y ) k \mathbf{P} = (P_1, \ldots, P_k) \in\Delta(\mathcal{Y})^k P = ( P 1 , … , P k ) ∈ Δ ( Y ) k always admits a canonical finite dominating measure, ν P = ∑ i = 1 k P i \nu_{\mathbf{P}} = \sum_{i=1}^{k} P_i ν P = ∑ i = 1 k P i . Returning to the simple example before, for any fixed set of k k k distributions P \mathbf{P} P , we can imagine sampling candidate points Y i Y_i Y i from a uniform mixture over P \mathbf{P} P . We can therefore apply the same Poisson coupling construction to each finite tuple respectively, using the tuple-dependent dominating measure ν P \nu_{\mathbf{P}} ν P . This yields a coupling Γ k ( P ; ⋅ ) ∈ Δ ( Y k ) \Gamma_k(\mathbf{P}; \cdot) \in\Delta(\mathcal{Y}^k) Γ k ( P ; ⋅ ) ∈ Δ ( Y k ) of P 1 , … , P k P_1, \ldots, P_k P 1 , … , P k . We then show that these couplings vary measurably with P \mathbf{P} P with the standard Borel machinery, so that Γ k : Δ ( Y ) k ⇝ Y k \Gamma_k : \Delta(\mathcal{Y})^k \rightsquigarrow\mathcal{Y}^k Γ k : Δ ( Y ) k ⇝ Y k is indeed a Markov kernel.
The remaining issue is that, the reference measure ν P \nu_{\mathbf{P}} ν P , and thus the Poisson process used in the construction, depend on the tuple P \mathbf{P} P – it essentially corresponds to a mixture over these k k k distributions. In our later privacy and utility analyses, however, we will need our coupling to handle when we enlarge a tuple of k k k distributions into a tuple of k + 1 k+1 k + 1 distributions without changing the joint distribution of the coordinates already present. This will capture adjoining the ground truth distribution in the utility analysis, or jointly realizing two neighboring executions in the privacy analysis. A priori, the coupling constructed for ( P 1 , P 2 ) (P_1, P_2) ( P 1 , P 2 ) need not coincide with the first two coordinates of the coupling constructed for ( P 1 , P 2 , P 3 ) (P_1, P_2, P_3) ( P 1 , P 2 , P 3 ) . Fortunately, the construction is insensitive to this choice. More formally, the finite-dimensional kernels are projectively consistent, namely, projecting the coupling of a larger tuple onto any subset of its coordinates gives exactly the coupling obtained by applying the construction directly to the corresponding subtuple.2
Finally, Kolmogorov’s Extension Theorem (Corollary A.6 ) then guarantees the existence of a single, universal joint measure Γ ∞ ∈ Δ ( Y Δ ( Y ) ) \Gamma_\infty\in\Delta(\mathcal{Y}^{\Delta(\mathcal{Y})}) Γ ∞ ∈ Δ ( Y Δ ( Y ) ) . In other words, Γ ∞ \Gamma_\infty Γ ∞ jointly realizes a random variable Z P Z_{\mathbf{P}} Z P for every P ∈ Δ ( Y ) \mathbf{P} \in\Delta(\mathcal{Y}) P ∈ Δ ( Y ) , with finite-dimensional marginals given by the kernels Γ k \Gamma_k Γ k , such that Z P ∼ P Z_{\mathbf{P}} \sim P Z P ∼ P for every P ∈ Δ ( Y ) \mathbf{P} \in\Delta(\mathcal{Y}) P ∈ Δ ( Y ) , and every pair satisfies the desired disagreement bound. In particular, restricting this universal coupling to any P ⊆ Δ ( Y ) \mathcal{P} \subseteq\Delta(\mathcal{Y}) P ⊆ Δ ( Y ) therefore yields a data-independent grand coupling scheme Γ P \Gamma_{\mathcal{P}} Γ P , without requiring a common dominating measure.
Proposition 3.3 (Informal version of Corollary A.7 and A.6). Suppose Y \mathcal{Y} Y is standard Borel. The family of random variables ( Z P ) P ∈ Δ ( Y ) ∼ Γ ∞ (Z_{\mathbf{P}})_{\mathbf{P}\in\Delta(\mathcal{Y})} \sim\Gamma_\infty ( Z P ) P ∈ Δ ( Y ) ∼ Γ ∞ induced by Construction A.4x constitutes a universal grand coupling scheme on Δ ( Y ) \Delta(\mathcal{Y}) Δ ( Y ) . Consequently, on any subset P ⊆ Δ ( Y ) \mathcal{P} \subseteq\Delta(\mathcal{Y}) P ⊆ Δ ( Y ) , its restriction ( Z P ) P ∈ P (Z_{\mathbf{P}})_{\mathbf{P}\in\mathcal{P}} ( Z P ) P ∈ P yields a valid grand coupling scheme Γ P \Gamma_{\mathcal{P}} Γ P on P \mathcal{P} P .
We emphasize that this grand coupling scheme serves purely as an analytical tool. Its purpose is to bring independent runs of the algorithms on two inputs into a shared probability space, for our privacy and utility analysis. By the nature of the coupling, the marginals exactly match the true target distributions (G1 in Definition 3.1 ). Consequently, all results derived within this joint probability space apply directly to the actual execution of the algorithms.
Private Synthetic Data Sampler
In this section, we apply the abstract grand coupling schemes developed in § 3 to construct a differentially private synthetic data sampler. Here, we set Y = X ℓ \mathcal{Y}=\mathcal{X}^{\ell} Y = X ℓ , where X \mathcal{X} X is the data universe and ℓ \ell ℓ is the size of the synthetic dataset. Let F ⊆ Δ ( X ) \mathcal{F}\subseteq\Delta(\mathcal{X}) F ⊆ Δ ( X ) contain the ground-truth distribution class Q \mathcal{Q} Q and all possible outputs of the non-private learner, and set P = { q ℓ : q ∈ F } \mathcal{P}=\{q^{\ell}:q\in\mathcal{F}\} P = { q ℓ : q ∈ F } . The goal is to privately output a synthetic dataset Z ∈ X ℓ Z\in\mathcal{X}^{\ell} Z ∈ X ℓ of arbitrary target size ℓ \ell ℓ such that TV ( L ( Z ) , p ℓ ) ≤ β \operatorname{TV}(\mathcal{L}(Z),p^{\ell})\leq\beta TV ( L ( Z ) , p ℓ ) ≤ β , given a dataset S ∼ p n S\sim p^{n} S ∼ p n from an unknown distribution p ∈ Q p\in\mathcal{Q} p ∈ Q .
Technical Intuition. At a high level, we partition the samples into disjoint blocks, run the non-private learner on each block, and use our grand coupling to generate correlated synthetic datasets from the resulting learner outputs.
For the utility analysis, we would like to show not only that the synthetic datasets Z 1 , … , Z k Z_1,\ldots,Z_k Z 1 , … , Z k agree with one another frequently, but that their common values “behaves like an i.i.d. sample” from the ground truth distribution p p p . To this end, we introduce an analysis-only “ghost” sample Z ∗ ∼ p ℓ Z^*\sim p^{\ell} Z ∗ ∼ p ℓ from the ground truth distribution and consider the augmented vector ( Z 1 , … , Z k , Z ∗ ) (Z_1,\ldots,Z_k,Z^*) ( Z 1 , … , Z k , Z ∗ ) with joint law Γ k + 1 ( ( q 1 ℓ , … , q k ℓ , p ℓ ) ; ⋅ ) \Gamma_{k+1}((q_1^{\ell},\ldots,q_k^{\ell},p^{\ell});\cdot) Γ k + 1 (( q 1 ℓ , … , q k ℓ , p ℓ ) ; ⋅ ) , which is the corresponding finite-dimensional marginal of our grand coupling Γ P \Gamma_{\mathcal{P}} Γ P from Proposition 3.3 . Note that the augmented vector is drawn from the same grand coupling, so adjoining this extra “ghost” coordinate does not change the joint law of the synthetic datasets produced by the algorithm, i.e., its marginal on the first k k k coordinates is exactly Γ k ( ( q 1 ℓ , … , q k ℓ ) ; ⋅ ) \Gamma_k((q_1^{\ell},\ldots,q_k^{\ell});\cdot) Γ k (( q 1 ℓ , … , q k ℓ ) ; ⋅ ) .
Moreover, by the disagreement bound, for every learn output q i q_i q i , Pr [ Z i ≠ Z ∗ ] ≤ 2 TV ( q i ℓ , p ℓ ) ≤ 2 ℓ TV ( q i , p ) \Pr[Z_i\ne Z^*]\leq2\operatorname{TV}(q_i^{\ell},p^{\ell})\leq2\ell\operatorname{TV}(q_i,p) Pr [ Z i = Z ∗ ] ≤ 2 TV ( q i ℓ , p ℓ ) ≤ 2 ℓ TV ( q i , p ) . Thus, when most blockwise learner outputs are accurate, many of the corresponding synthetic dataset agree with Z ∗ Z^* Z ∗ exactly. With appropriate accuracy parameters, we show that a large majority agree with Z ∗ Z^* Z ∗ , providing the frequency gap required by StabilityBasedMode to output Z ∗ Z^* Z ∗ , with high probability. The additional coordinate Z ∗ Z^* Z ∗ is used only in the analysis and is not generated by the algorithm.
For the privacy analysis, fix neighboring datasets S , S ′ S,S' S , S ′ and use the same partition and base-learner randomness in both executions. Changing one record affects at most one block, so the resulting tuples q = ( q 1 , … , q k ) \mathbf{q}=(q_1,\ldots,q_k) q = ( q 1 , … , q k ) and q ′ = ( q 1 ′ , … , q k ′ ) \mathbf{q}'=(q'_1,\ldots,q'_k) q ′ = ( q 1 ′ , … , q k ′ ) differ in at most one coordinate j j j . For the analysis only, we take a single draw ( Z P ) P ∈ P ∼ Γ P (Z_P)_{P\in\mathcal{P}}\sim\Gamma_{\mathcal{P}} ( Z P ) P ∈ P ∼ Γ P and define Z ~ i = Z q i ℓ \widetilde{Z}_i=Z_{q_i^{\ell}} Z i = Z q i ℓ , and Z ~ i ′ = Z ( q i ′ ) ℓ \widetilde{Z}'_i=Z_{(q'_i)^{\ell}} Z i ′ = Z ( q i ′ ) ℓ . By (G3) and the definition of Couple k \operatorname{Couple}_k Couple k , ( Z ~ 1 , … , Z ~ k ) (\widetilde{Z}_1,\ldots,\widetilde{Z}_k) ( Z 1 , … , Z k ) has joint law Γ k ( ( q 1 , … , q k ) ; ⋅ ) \Gamma_k((q_1,\ldots,q_k);\cdot) Γ k (( q 1 , … , q k ) ; ⋅ ) , and ( Z ~ 1 ′ , … , Z ~ k ′ ) (\widetilde{Z}'_1,\ldots,\widetilde{Z}'_k) ( Z 1 ′ , … , Z k ′ ) has joint law Γ k ( ( ( q 1 ′ ) ℓ , … , ( q k ′ ) ℓ ) ; ⋅ ) \Gamma_k(((q'_1)^{\ell},\ldots,(q'_k)^{\ell});\cdot) Γ k ((( q 1 ′ ) ℓ , … , ( q k ′ ) ℓ ) ; ⋅ ) . These are exactly the joint laws used by the coupled sampling step in the two executions. For every unchanged coordinate i ≠ j i\ne j i = j , we have q i = q i ′ q_i=q'_i q i = q i ′ , and hence Z ~ i = Z ~ i ′ \widetilde{Z}_i=\widetilde{Z}'_i Z i = Z i ′ for every realization of the shared draw.3
Thus, the two inputs to StabilityBasedMode are neighboring for every realization, so its privacy guarantee applies to each fixed coupling draw.
Road Map. We present the formal construction in two stages to isolate the coupling randomness from the statistical randomness of the dataset partition and base learners.
AGGREGATE (Algorithm 1): Operates on a fixed tuple q = ( q 1 , … , q k ) ∈ F k \mathbf{q}=(q_1,\ldots,q_k)\in\mathcal{F}^k q = ( q 1 , … , q k ) ∈ F k . It draws ( Z 1 , … , Z k ) (Z_1,\ldots,Z_k) ( Z 1 , … , Z k ) via the grand coupling and aggregates them with the stability-based mode. Its privacy (under tuple-level change-one adjacency) follows directly from the construction of the
grand coupling, while its utility follows from the disagreement bound and the accuracy of the stability-based mode mechanism.
Sampler (Algorithm 2): Operates on raw datasets S ∈ X n S \in\mathcal{X}^n S ∈ X n , partitioning S S S to form q \mathbf{q} q before invoking Aggregate. It inherits the guarantees of Stage 1, and we can lift the privacy to standard change-one adjacency on X n \mathcal{X}^n X n since changing one data point affects a single block in q \mathbf{q} q , while utility follows by combining Hoeffding’s inequality over the block partition with the utility guarantees of Aggregate.
We defer the detailed proofs and discussions in Appendix § 8 .
Tuple-Level Primitive: Aggregate
We first present Aggregate (Algorithm 1), a mechanism operating on a fixed tuple of candidate distributions q = ( q 1 , … , q k ) ∈ F k \mathbf{q}=(q_1,\ldots,q_k)\in\mathcal{F}^k q = ( q 1 , … , q k ) ∈ F k to output a synthetic candidate Z ∈ X ℓ Z\in\mathcal{X}^{\ell} Z ∈ X ℓ . Analysis in this subsection is purely with respect to tuple-level adjacency, where neighboring inputs q , q ′ \mathbf{q},\mathbf{q}' q , q ′ differ in a single distribution coordinate q j q_j q j .
Algorithm 1 Aggregate
Require: Distributions q 1 , … , q k ∈ F q_1,\ldots,q_k\in\mathcal{F} q 1 , … , q k ∈ F .
Ensure: Synthetic dataset Z ∈ X ℓ Z\in\mathcal{X}^{\ell} Z ∈ X ℓ .
Jointly sample ( Z 1 , … , Z k ) ← Couple k ( q 1 ℓ , … , q k ℓ ) (Z_1,\ldots,Z_k)\leftarrow\operatorname{Couple}_k(q_1^{\ell},\ldots,q_k^{\ell}) ( Z 1 , … , Z k ) ← Couple k ( q 1 ℓ , … , q k ℓ ) // Couple k \operatorname{Couple}_k Couple k is a finite-dimensional marginal of a valid grand coupling scheme on P = { q ℓ : q ∈ F } \mathcal{P}=\{q^{\ell}:q\in\mathcal{F}\} P = { q ℓ : q ∈ F } , which can be instantiated by either Couple_I in Algorithm 5 or Couple_II in Algorithm 6
Z = StabilityBasedMode ( Z 1 , … , Z k ) Z=\operatorname{StabilityBasedMode}(Z_1,\ldots,Z_k) Z = StabilityBasedMode ( Z 1 , … , Z k )
return Z Z Z .
By instantiating Couple k \operatorname{Couple}_k Couple k with a valid grand coupling scheme Γ P \Gamma_{\mathcal{P}} Γ P 4 , it produces a joint measurable vector ( Z 1 , … , Z k ) (Z_1,\ldots,Z_k) ( Z 1 , … , Z k ) , for any input tuple q \mathbf{q} q , such that Z i ∼ q i ℓ Z_i\sim q_i^{\ell} Z i ∼ q i ℓ and Pr [ Z i ≠ Z j ] ≤ 2 TV ( q i ℓ , q j ℓ ) \Pr[Z_i\ne Z_j]\le2\operatorname{TV}(q_i^{\ell},q_j^{\ell}) Pr [ Z i = Z j ] ≤ 2 TV ( q i ℓ , q j ℓ ) for all i ≠ j i\ne j i = j . We next show that: (i) the privacy guarantee of StabilityBasedMode transfers directly to Aggregate, and (ii) if a large majority of the input distributions q i q_i q i are close to a target ground-truth distribution p p p in TV, the output of Aggregate matches an ideal draw from p ℓ p^{\ell} p ℓ with high probability.
Theorem 4.1 (Informal version of Corollary B.4). For every ε > 0 \varepsilon>0 ε > 0 , δ , β ∈ ( 0 , 1 ) \delta,\beta\in(0,1) δ , β ∈ ( 0 , 1 ) , ℓ ∈ N + \ell\in\mathbb{N}^{+} ℓ ∈ N + , and k = Ω ( log ( 4 / β ) + log ( 1 / δ ) ε 2 ) k=\Omega\left(\frac{\log(4/\beta)+\log(1/\delta)}{\varepsilon^2}\right) k = Ω ( ε 2 l o g ( 4/ β ) + l o g ( 1/ δ ) ) , Aggregate is ( ε , δ ) (\varepsilon,\delta) ( ε , δ ) -DP5 . Moreover, for any p ∈ Q p\in\mathcal{Q} p ∈ Q and input tuple q ∈ F k \mathbf{q}\in\mathcal{F}^k q ∈ F k , if at least 7 / 8 7/8 7/8 of the input distributions are O ( β / ℓ ) O(\beta/\ell) O ( β / ℓ ) -close to p p p in TV, then the law of its output is β / 2 \beta/2 β /2 -close to p ℓ p^{\ell} p ℓ in TV.
Proof Sketch. Privacy: Since the joint law of Couple k ( q ) \operatorname{Couple}_k(\mathbf{q}) Couple k ( q ) is a finite-dimensional marginal of the global measure Γ P \Gamma_{\mathcal{P}} Γ P 6 , we evaluate executions on neighboring tuples q \mathbf{q} q and q ′ \mathbf{q}' q ′ on a single shared probability space. Specifically, we construct an analysis coupling from a single draw ( Z P ) P ∈ P ∼ Γ P (Z_P)_{P\in\mathcal{P}}\sim\Gamma_{\mathcal{P}} ( Z P ) P ∈ P ∼ Γ P , setting Z ~ i = Z q i ℓ \widetilde{Z}_i=Z_{q_i^{\ell}} Z i = Z q i ℓ for input q \mathbf{q} q and Z ~ i ′ = Z ( q i ′ ) ℓ \widetilde{Z}'_i=Z_{(q'_i)^{\ell}} Z i ′ = Z ( q i ′ ) ℓ for input q ′ \mathbf{q}' q ′ . For every unchanged coordinate i ≠ j i\ne j i = j , q i = q i ′ q_i = q_i' q i = q i ′ implies Z q i ℓ = Z ( q i ′ ) ℓ Z_{q_i}^{\ell} = Z_{(q_i')^{\ell}} Z q i ℓ = Z ( q i ′ ) ℓ pointwise. Consequently, the realized input vectors ( Z ~ 1 , … , Z ~ k ) (\widetilde{Z}_1,\ldots,\widetilde{Z}_k) ( Z 1 , … , Z k ) and ( Z ~ 1 ′ , … , Z ~ k ′ ) (\widetilde{Z}'_1,\ldots,\widetilde{Z}'_k) ( Z 1 ′ , … , Z k ′ ) differ in at most a single coordinate j j j for every realization of Γ P \Gamma_{\mathcal{P}} Γ P . Therefore, since neighboring input tuples yield pointwise neighboring vectors under Γ P \Gamma_{\mathcal{P}} Γ P , taking expectations over Γ P \Gamma_{\mathcal{P}} Γ P transfer the DP guarantee of StabilityBasedMode directly to Aggregate.
Utility: As in the privacy analysis, the utility guarantee relies on constructing an analysis coupling from the grand coupling scheme Γ P \Gamma_{\mathcal{P}} Γ P . Specifically, for input distributions q = ( q 1 , … , q k ) ∈ F k \mathbf{q}=(q_1,\ldots,q_k)\in\mathcal{F}^k q = ( q 1 , … , q k ) ∈ F k and a target distribution p ∈ F p\in\mathcal{F} p ∈ F , we take a single draw ( Z P ) P ∈ P ∼ Γ P (Z_P)_{P\in\mathcal{P}}\sim\Gamma_{\mathcal{P}} ( Z P ) P ∈ P ∼ Γ P and define Z ~ i = Z q i ℓ \widetilde{Z}_i=Z_{q_i}^{\ell} Z i = Z q i ℓ for each i ∈ [ k ] i\in[k] i ∈ [ k ] , alongside an unobserved “ghost” sample Z ~ ∗ = Z p ℓ \widetilde{Z}_*=Z_{p^\ell} Z ∗ = Z p ℓ , indexed by the ground-truth product distribution p ℓ ∈ P p^\ell\in\mathcal{P} p ℓ ∈ P . Since L ( C o u p l e k ( q ) ) \mathcal{L}(\mathrm{Couple}_k(\mathbf{q})) L ( Couple k ( q )) and L ( C o u p l e 1 ( p ) ) \mathcal{L}(\mathrm{Couple}_1(p)) L ( Couple 1 ( p )) are simply a finite-dimensional marginal of the global measure Γ P \Gamma_{\mathcal{P}} Γ P , our construction from the grand coupling scheme Γ P \Gamma_{\mathcal{P}} Γ P satisfies ( Z ~ 1 , … , Z ~ k ) ∼ L ( C o u p l e k ( q ) ) (\widetilde{Z}_1,\ldots,\widetilde{Z}_k)\sim\mathcal{L}(\mathrm{Couple}_k(\mathbf{q})) ( Z 1 , … , Z k ) ∼ L ( Couple k ( q )) and Z ~ ∗ ∼ L ( C o u p l e 1 ( p ) ) = p ℓ \widetilde{Z}_*\sim\mathcal{L}(\mathrm{Couple}_1(p))=p^\ell Z ∗ ∼ L ( Couple 1 ( p )) = p ℓ , by property G3 and G1 from Definition 3.1 .
For every accurate input distribution where TV ( q i , p ) ≤ η \operatorname{TV}(q_i,p)\leq\eta TV ( q i , p ) ≤ η , property G2 bounds the disagreement by Pr [ Z ~ i ≠ Z ~ ∗ ] ≤ 2 TV ( q i ℓ , p ℓ ) ≤ 2 ℓ η \Pr[\widetilde{Z}_i\ne\widetilde{Z}_*]\leq2\operatorname{TV}(q_i^\ell,p^\ell)\leq2\ell\eta Pr [ Z i = Z ∗ ] ≤ 2 TV ( q i ℓ , p ℓ ) ≤ 2 ℓ η . When 7 / 8 7/8 7/8 of the q i q_i q i are η \eta η -close to p p p , an overwhelming majority of the coupled samples Z ~ i \widetilde{Z}_i Z i collapse onto the exact same realization Z ~ ∗ ∼ p ℓ \widetilde{Z}_*\sim p^\ell Z ∗ ∼ p ℓ pointwise. This manufactures the frequency gap required for StabilityBasedMode [46 Vad17] to output Z ~ ∗ \widetilde{Z}_* Z ∗ with probability ≥ 1 − δ 0 \geq1-\delta_0 ≥ 1 − δ 0 .
We emphasize that neither Z ~ ∗ \widetilde{Z}_* Z ∗ nor Z ~ i ′ \widetilde{Z}'_i Z i ′ is computed by the algorithm, and they are introduced purely for the analysis. Full technical details appear in Proposition B.2 , Proposition B.1 , and Corollary B.4 .
Algorithm 2 Sampler
Require: Dataset S = ( X 1 , … , X n ) S=(X_1,\ldots,X_n) S = ( X 1 , … , X n ) with n = b k n=bk n = bk . Number of blocks k k k .
Ensure: Synthetic samples Z ∈ X ℓ Z\in\mathcal{X}^{\ell} Z ∈ X ℓ
Let a random seed (a data-independent permutation) be Π ∼ U ( S n ) \Pi\sim U(S_n) Π ∼ U ( S n )
Partition S S S into k k k disjoint blocks S 1 , … , S k ∈ X b S_1,\ldots,S_k\in\mathcal{X}^b S 1 , … , S k ∈ X b , by setting S i ← ( X Π ( ( i − 1 ) b + 1 ) , … , X Π ( i b ) ) S_i\leftarrow(X_{\Pi((i-1)b+1)},\ldots,X_{\Pi(ib)}) S i ← ( X Π (( i − 1 ) b + 1 ) , … , X Π ( ib ) )
Let learners’ random seeds (data- and permutation-independent) be U 1 , … , U k ∼ i . i . d . U ( [ 0 , 1 ] ) U_1,\ldots,U_k \mathrel{\overset{\mathrm{i.i.d.}}{\sim}} U([0,1]) U 1 , … , U k ∼ i.i.d. U ([ 0 , 1 ]) .
For each i ∈ [ k ] i\in[k] i ∈ [ k ] , compute q i ← L ( S i ; U i ) q_i\leftarrow L(S_i;U_i) q i ← L ( S i ; U i )
Z ← Aggregate ( q 1 , … , q k ) Z\leftarrow\operatorname{Aggregate}(q_1,\ldots,q_k) Z ← Aggregate ( q 1 , … , q k )
return Z ∈ X ℓ Z\in\mathcal{X}^{\ell} Z ∈ X ℓ
End-to-End Private Correlated Sampler: Sampler
In the preceding subsection, we established the privacy and utility guarantees of Aggregate conditioned on a fixed tuple of candidate distributions q = ( q 1 , … , q k ) \mathbf{q}=(q_1,\ldots,q_k) q = ( q 1 , … , q k ) . We now lift these guarantees to the full end-to-end pipeline, Sampler (Algorithm 2), where input distributions are learned from disjoint blocks of a raw dataset S = ( X 1 , … , X n ) ∈ X n S=(X_1,\ldots,X_n)\in\mathcal{X}^n S = ( X 1 , … , X n ) ∈ X n . The algorithm partitions S S S into disjoint blocks, runs the base learner on each block to obtain the random tuple q \mathbf{q} q , and then applies Aggregate. Here, privacy is naturally defined at the dataset level, where two datasets S , S ′ ∈ X n S,S'\in\mathcal{X}^n S , S ′ ∈ X n are adjacent if they differ in a single record.
While much of the analysis relies on standard techniques, the primary technical step lies in explicitly controlling the sources of randomness to bridge Aggregate and Sampler, ensuring that the resulting random tuple satisfies the conditions required by Aggregate.
Theorem 4.3 (Informal version of Corollary B.7). Fix a base learner L L L for Q Q Q . For every ε , δ , β ∈ ( 0 , 1 ) \varepsilon,\delta,\beta\in(0,1) ε , δ , β ∈ ( 0 , 1 ) and ℓ ∈ N + \ell\in\mathbb{N}^{+} ℓ ∈ N + , let b b b be a sample size such that, for every p ∈ Q p\in Q p ∈ Q , L L L achieves TV ( L ( S i ) , p ) ≤ O ( β / ℓ ) \operatorname{TV}(L(S_i),p)\le O(\beta/\ell) TV ( L ( S i ) , p ) ≤ O ( β / ℓ ) with probability at least 15 / 16 15/16 15/16 given S i ∼ p b S_i\sim p^b S i ∼ p b . Then, Sampler satisfies dataset-level ( ε , δ ) (\varepsilon,\delta) ( ε , δ ) -DP.
Moreover, for every p ∈ Q p\in Q p ∈ Q , given S ∼ p n S\sim p^n S ∼ p n with n = Ω ( b log ( 4 / δ β ) ε ) n=\Omega\left(\frac{b\log(4/\delta\beta)}{\varepsilon}\right) n = Ω ( ε b l o g ( 4/ δ β ) ) , it outputs a synthetic dataset Z ∈ X ℓ Z\in\mathcal{X}^{\ell} Z ∈ X ℓ satisfying TV ( L ( Z ) , p ℓ ) ≤ 3 β / 4 \operatorname{TV}(L(Z),p^{\ell})\le3\beta/4 TV ( L ( Z ) , p ℓ ) ≤ 3 β /4 . Thus, Sampler is an ( ℓ , 3 β / 4 ) (\ell,3\beta/4) ( ℓ , 3 β /4 ) -approximate sampler for Q Q Q in the sense of Definition 1.4 .
Proof Sketch. Privacy: Fix adjacent datasets S , S ′ ∈ X n S,S'\in\mathcal{X}^n S , S ′ ∈ X n differing at record j j j . For any realization of the permutation Π \Pi Π , the record that differs in the two worlds falls into a unique block index a a a . Since all other blocks, S i S_i S i for i ≠ a i\ne a i = a , remain identical, and the base learner’s internal randomness U 1 , … , U k U_1,\ldots,U_k U 1 , … , U k is drawn independently of input data S S S , the output tuples q = L ( S ) \mathbf{q}=L(S) q = L ( S ) and q ′ = L ( S ′ ) \mathbf{q}'=L(S') q ′ = L ( S ′ ) differ in at most coordinate a a a . Taking an expectation over the data-independent randomness, i.e., the random permutation for data partition and the learners’ random seeds, the tuple-level DP guarantee of Aggregate, shown in Theorem 4.1 , directly lifts to dataset-level ( ε , δ ) (\varepsilon,\delta) ( ε , δ ) -DP for Sampler.
Utility: Define indicator variables I i = 1 { TV ( q i , p ) ≤ η } I_i=1\{\operatorname{TV}(q_i,p)\le\eta\} I i = 1 { TV ( q i , p ) ≤ η } for η = O ( β / ℓ ) \eta=O(\beta/\ell) η = O ( β / ℓ ) . Since S ∼ p n S\sim p^n S ∼ p n i.i.d., the blocks S i S_i S i are i.i.d. draws from p b p^b p b , making I 1 , … , I k I_1,\ldots,I_k I 1 , … , I k i.i.d. Bernoulli variables with mean E [ I i ] ≥ 15 / 16 \mathbb{E}[I_i]\ge15/16 E [ I i ] ≥ 15/16 . By Hoeffding’s inequality, setting k = O ( log ( 1 / β ) ) k=O(\log(1/\beta)) k = O ( log ( 1/ β )) ensures that ∑ i = 1 k I i ≥ 7 8 k \sum_{i=1}^k I_i\ge\frac{7}{8}k ∑ i = 1 k I i ≥ 8 7 k holds with probability at least 1 − β / 4 1-\beta/4 1 − β /4 . Conditioning on this good event = { ∣ { i : TV ( q i , p ) ≤ η } ∣ ≥ 7 8 k } =\{|\{i:\operatorname{TV}(q_i,p)\le\eta\}|\ge\frac{7}{8}k\} = { ∣ { i : TV ( q i , p ) ≤ η } ∣ ≥ 8 7 k } , Theorem 4.1 guarantees that the output distribution L ( Z ∣ good event ) \mathcal{L}(Z\mid\text{good event}) L ( Z ∣ good event ) satisfies TV ( L ( Z ∣ good event ) , p ℓ ) ≤ β / 2 \operatorname{TV}(\mathcal{L}(Z\mid\text{good event}),p^{\ell})\le\beta/2 TV ( L ( Z ∣ good event ) , p ℓ ) ≤ β /2 . Combining the failure probability of the concentration with the conditional TV error yields TV ( L ( Z ) , p ℓ ) ≤ β / 2 + β / 4 = 3 4 β \operatorname{TV}(\mathcal{L}(Z),p^{\ell})\le\beta/2+\beta/4=\frac{3}{4}\beta TV ( L ( Z ) , p ℓ ) ≤ β /2 + β /4 = 4 3 β .
Full technical details appear in Proposition B.5 and Proposition B.6 .
Remark 4.4. Since private and non-private PAC-learnability are known to not be equivalent [19 , 8 ] , it is instructive to understand where the present approach fails in supervised settings.
Algorithm 3 BlackBoxPrivateLearner
Require: Dataset S = ( X 1 , … , X n ) S=(X_1,\ldots,X_n) S = ( X 1 , … , X n ) Ensure: A distribution q ^ ∈ Δ ( X ) \hat{q}\in\Delta(\mathcal{X}) q ^ ∈ Δ ( X )
Z ← Sampler ( S ) Z\leftarrow\operatorname{Sampler}(S) Z ← Sampler ( S )
Pick a learner’s random seed (data-independent) V ∼ U ( [ 0 , 1 ] ) V\sim U([0,1]) V ∼ U ([ 0 , 1 ]) and q ^ ← L ( Z ; V ) \hat{q}\leftarrow L(Z;V) q ^ ← L ( Z ; V )
return q ^ \hat{q} q ^
This becomes clear by looking at Algorithm 2. The algorithm crucially relies on the fact that a distribution learning algorithm outputs a density that is close to the ground truth in TV-distance. However, in supervised settings, the input consists of domain elements and their labels, while the learner outputs a function attempting to capture the rule that associates the former to the latter. This does not capture the distribution of the unlabeled examples, and therefore cannot be used as the basis for a sampler.
Main algorithm and reduction
We now present our main application, a black-box reduction that transforms any non-private base learner L L L into a DP learner via Sampler. Let m ( α , β ) m(\alpha,\beta) m ( α , β ) denote the non-private sample complexity of L L L for outputting an α \alpha α -accurate estimate in terms of TV, with probability at least 1 − β 1-\beta 1 − β .
The algorithm is conceptually simple, where we first invoke Sampler to generate a private synthetic dataset Z ∈ X ℓ Z\in\mathcal{X}^{\ell} Z ∈ X ℓ , and then apply the non-private learner L L L to the private synthetic dataset Z Z Z using an independent random seed V V V .
The complete procedure is described in Algorithm 3.
The results follows immediately from the guarantees of Sampler. Privacy follows by post-processing, and utility follows by coupling its output to an i.i.d. sample from p ℓ p^{\ell} p ℓ and union-bounding the coupling mismatch with the learner’s failure event. The main privacy and utility guarantees of PrivateLearner are established in the following theorem.
Theorem 5.1 (Main Reduction). Let α , β , δ , ε ∈ ( 0 , 1 ) \alpha,\beta,\delta,\varepsilon\in(0,1) α , β , δ , ε ∈ ( 0 , 1 ) be parameters, and L L L be a non-private learner for Q \mathcal{Q} Q with sample complexity m ( α , β ) m(\alpha,\beta) m ( α , β ) . Then, BlackBoxPrivateLearner (Algorithm 3) is ( ε , δ ) (\varepsilon,\delta) ( ε , δ ) -DP. Moreover, there is a sample size
n = O ( m ( β 64 m ( α , β 4 ) , 1 16 ) 1 ε log ( 4 β δ ) ) n=O\left(m\left(\frac{\beta}{64m\left(\alpha,\frac{\beta}{4}\right)},\frac{1}{16}\right)\frac{1}{\varepsilon}\log\left(\frac{4}{\beta\delta}\right)\right) n = O m 64 m ( α , 4 β ) β , 16 1 ε 1 log ( β δ 4 ) such that, for every p ∈ Q p\in\mathcal{Q} p ∈ Q , given S ∼ p n S\sim p^n S ∼ p n , the output q ^ \hat{q} q ^ satisfies TV ( q ^ , p ) < α \operatorname{TV}(\hat{q},p)<\alpha TV ( q ^ , p ) < α with probability at least 1 − β 1-\beta 1 − β .
Proof. Set ℓ = m ( α , β 4 ) \ell=m\left(\alpha,\frac{\beta}{4}\right) ℓ = m ( α , 4 β ) , η = β 64 ℓ \eta=\frac{\beta}{64\ell} η = 64 ℓ β , b = m ( η , 1 16 ) b=m\left(\eta,\frac{1}{16}\right) b = m ( η , 16 1 ) , and δ 0 = min { δ , β 4 } \delta_0=\min\left\{\delta,\frac{\beta}{4}\right\} δ 0 = min { δ , 4 β } . By Proposition B.5 , Sampler is ( ε , δ 0 ) (\varepsilon,\delta_0) ( ε , δ 0 ) -DP. Applying L L L to its output using independent randomness preserves privacy by post-processing (Proposition 2.4 ). Since δ 0 ≤ δ \delta_0\le\delta δ 0 ≤ δ , the resulting learner is ( ε , δ ) (\varepsilon,\delta) ( ε , δ ) -DP.
For utility, take k = ⌈ max { 8 ε log ( 4 β δ ) , 128 log ( 4 β ) } ⌉ k=\left\lceil\max\left\{\frac{8}{\varepsilon}\log\left(\frac{4}{\beta\delta}\right),128\log\left(\frac{4}{\beta}\right)\right\}\right\rceil k = ⌈ max { ε 8 log ( β δ 4 ) , 128 log ( β 4 ) } ⌉ and n = b k n=bk n = bk . This choice satisfies both the stability requirement and the concentration bound in Corollary B.7. Hence, for every p ∈ Q p\in\mathcal{Q} p ∈ Q and S ∼ p n S\sim p^n S ∼ p n , the synthetic output Z Z Z satisfies TV ( L ( Z ) , p ℓ ) ≤ 3 β 4 \operatorname{TV}(L(Z),p^{\ell})\le\frac{3\beta}{4} TV ( L ( Z ) , p ℓ ) ≤ 4 3 β . By the choice of ℓ \ell ℓ , the learner fails to achieve TV error strictly below α \alpha α on an i.i.d. sample from p ℓ p^{\ell} p ℓ with probability at most β 4 \frac{\beta}{4} 4 β . Transferring this failure bound using union bound across the TV distance gives Pr [ TV ( q ^ , p ) ≥ α ] ≤ β 4 + 3 β 4 = β \Pr[\operatorname{TV}(\hat{q},p)\ge\alpha]\le\frac{\beta}{4}+\frac{3\beta}{4}=\beta Pr [ TV ( q ^ , p ) ≥ α ] ≤ 4 β + 4 3 β = β .
Finally, substituting the value of b b b and k = O ( 1 ε log ( 4 β δ ) ) k = O\left(\frac{1}{\varepsilon}\log\left(\frac{4}{\beta\delta}\right)\right) k = O ( ε 1 log ( β δ 4 ) ) into n = b k n = bk n = bk gives the claimed sample complexity.
A Boosted Private Learner
In the previous section, the PrivateLearner requires the block-level target TV error to satisfy η ≲ β / ℓ \eta\lesssim\beta/\ell η ≲ β / ℓ . When the non-private sample complexity scales polynomially with accuracy, i.e., m ( α , β ) = poly ( 1 / α , log ( 1 / β ) ) m(\alpha,\beta) = \operatorname{poly}(1/\alpha,\log(1/\beta)) m ( α , β ) = poly ( 1/ α , log ( 1/ β )) , setting η ∝ β \eta\propto\beta η ∝ β introduces an undesirable polynomial dependence 1 / β Ω ( 1 ) 1/\beta^{\Omega(1)} 1/ β Ω ( 1 ) into the overall sample complexity n n n . To get rid of this polynomial dependence and achieve a better logarithmic overhead O ( log ( 1 / β ) ) O(\log(1/\beta)) O ( log ( 1/ β )) , we introduce a boosted version, BoostedLearner in Algorithm 4.
The core idea is confidence boosting: amplifying a learner’s constant success probability to 1 − β 1-\beta 1 − β through independent repetition and robust aggregation. We run PrivateLearner independently on T = O ( log ( 1 / β ) ) T = O(\log(1/\beta)) T = O ( log ( 1/ β )) disjoint blocks S 1 , … , S T S_1,\ldots,S_T S 1 , … , S T with a fixed constant success probability 1 − β 0 1-\beta_0 1 − β 0 (e.g., β 0 = 1 / 8 \beta_0 = 1/8 β 0 = 1/8 ), obtaining estimates p ^ 1 , … , p ^ T \hat{p}_1,\ldots,\hat{p}_T p ^ 1 , … , p ^ T . This requires a block accuracy of only η ∼ 1 / ℓ \eta\sim1/\ell η ∼ 1/ ℓ , independent of the final target failure probability β \beta β . For each t ∈ [ T ] t \in[T] t ∈ [ T ] , N t = ∣ { i ∈ [ T ] : TV ( p ^ i , p ^ t ) ≤ 2 α 0 } ∣ N_t = |\{i \in[T] : \operatorname{TV}(\hat{p}_i,\hat{p}_t) \le2\alpha_0\}| N t = ∣ { i ∈ [ T ] : TV ( p ^ i , p ^ t ) ≤ 2 α 0 } ∣ counts how many estimates are within TV distance 2 α 0 2\alpha_0 2 α 0 of p ^ t \hat{p}_t p ^ t , where α 0 = α / 3 \alpha_0 = \alpha/3 α 0 = α /3 . We select j j j maximizing N j N_j N j and return p ^ j \hat{p}_j p ^ j .
By doing so, if more than half of the estimates are α 0 \alpha_0 α 0 -close to p p p , they are all within 2 α 0 2\alpha_0 2 α 0 of one another. Thus, every such estimate has N t > T / 2 N_t > T/2 N t > T /2 , and the maximizing estimate also has N j > T / 2 N_j > T/2 N j > T /2 . Among the estimates counted by N j N_j N j , at least one must therefore be α 0 \alpha_0 α 0 -close to p p p , and thus TV ( p ^ j , p ) ≤ 2 α 0 + α 0 = α \operatorname{TV}(\hat{p}_j,p) \le2\alpha_0+\alpha_0 = \alpha TV ( p ^ j , p ) ≤ 2 α 0 + α 0 = α . Each run fails with probability at most β 0 \beta_0 β 0 , and by independence, the probability of failing to obtain a majority of accurate estimates decreases exponentially in T T T . See D for the formal proof.
Theorem 6.1 (Boosted Private Learner). Let α , β , ε , δ ∈ ( 0 , 1 ) \alpha,\beta,\varepsilon,\delta\in(0,1) α , β , ε , δ ∈ ( 0 , 1 ) be parameters, and L L L be a non-private learner for Q \mathcal{Q} Q with sample complexity m ( α , β ) m(\alpha,\beta) m ( α , β ) . Then, BoostedLearner (Algorithm 4) is ( ε , δ ) (\varepsilon,\delta) ( ε , δ ) -DP. Moreover, there is a sample size
n = O ( m ( 1 512 m ( α / 3 , 1 / 32 ) , 1 16 ) 1 ε log ( 4 δ ) log ( 4 β ) ) n = O\left(m\left(\frac{1}{512m(\alpha/3,1/32)},\frac{1}{16}\right)\frac{1}{\varepsilon}\log\left(\frac{4}{\delta}\right)\log\left(\frac{4}{\beta}\right)\right) n = O ( m ( 512 m ( α /3 , 1/32 ) 1 , 16 1 ) ε 1 log ( δ 4 ) log ( β 4 ) )
such that, for every p ∈ Q p \in\mathcal{Q} p ∈ Q , given S ∼ p n S \sim p^n S ∼ p n , the output q ^ \hat{q} q ^ satisfies TV ( q ^ , p ) < α \operatorname{TV}(\hat{q}, p) < \alpha TV ( q ^ , p ) < α with probability at least 1 − β 1-\beta 1 − β . In particular, if m ( α , β ) = poly ( 1 α , log ( 1 β ) ) m(\alpha,\beta)=\operatorname{poly}\left(\frac{1}{\alpha},\log\left(\frac{1}{\beta}\right)\right) m ( α , β ) = poly ( α 1 , log ( β 1 ) ) , then n = poly ( 1 α , log ( 1 β ) , 1 ε , log ( 1 δ ) ) n=\operatorname{poly}\left(\frac{1}{\alpha},\log\left(\frac{1}{\beta}\right),\frac{1}{\varepsilon},\log\left(\frac{1}{\delta}\right)\right) n = poly ( α 1 , log ( β 1 ) , ε 1 , log ( δ 1 ) ) .
Note that the DP guarantee is non-trivial only when δ = o ( 1 / n ) \delta=o(1/n) δ = o ( 1/ n ) , since allowing δ = Ω ( 1 / n ) \delta=\Omega(1/n) δ = Ω ( 1/ n ) permits the learner to randomly select a user’s data to output each run [42 ] . Our sample complexity bound is meaningful in this regime because of its dependence on δ \delta δ only through a log ( 1 / δ ) \log(1/\delta) log ( 1/ δ ) factor. Specifically, rewriting the sample complexity in Theorem 6.1 as n ( δ ) = C log ( 4 / δ ) n(\delta)=C\log(4/\delta) n ( δ ) = C log ( 4/ δ ) , where C C C collects other factors independent of δ \delta δ , we have δ ⋅ n ( δ ) = C δ log ( 4 / δ ) → 0 \delta\cdot n(\delta)=C\delta\log(4/\delta)\to0 δ ⋅ n ( δ ) = C δ log ( 4/ δ ) → 0 as δ → 0 \delta\to0 δ → 0 , that is, δ = o ( 1 / n ) \delta=o(1/n) δ = o ( 1/ n ) . In contrast, a dependence of the form n ( δ ) = C / δ n(\delta)=C/\delta n ( δ ) = C / δ would give δ ⋅ n ( δ ) = C \delta\cdot n(\delta)=C δ ⋅ n ( δ ) = C , so δ = Θ ( 1 / n ) \delta=\Theta(1/n) δ = Θ ( 1/ n ) , which falls back to the trivial regime.
Acknowledgements
We thank David Woodruff for discussions and assistance with the internal version of Gemini used. We also thank Matthew Joseph and Chris Liaw for their comments on a preliminary draft of this paper.
Poisson Coupling Details
We present two constructions of the grand coupling interface in Definition 3.1 , both based on the Poisson coupling of [10 ] . Subsection A.1 treats families P ⊆ Δ ( Y ) \mathcal{P} \subseteq\Delta(\mathcal{Y}) P ⊆ Δ ( Y ) that admit a common σ \sigma σ -finite dominating measure. For such families, Corollary A.3 establishes a grand coupling scheme on P \mathcal{P} P . Subsection A.2 treats the general case, without assuming a common dominating measure. For every standard Borel space Y \mathcal{Y} Y , Corollary A.7 establishes a universal grand coupling scheme on Δ ( Y ) \Delta(\mathcal{Y}) Δ ( Y ) . Restricting this scheme to any family P ⊆ Δ ( Y ) \mathcal{P} \subseteq\Delta(\mathcal{Y}) P ⊆ Δ ( Y ) gives a grand coupling scheme on P \mathcal{P} P .
A Globally Dominated Poisson Coupling
We first formalize the setting where all distributions in the family are dominated by a single reference measure. Let ( Y , B ) (\mathcal{Y},\mathcal{B}) ( Y , B ) be a standard Borel space and let P ⊆ Δ ( Y ) \mathcal{P} \subseteq\Delta(\mathcal{Y}) P ⊆ Δ ( Y ) be a collection of probability measures of interest. We use the original construction taken verbatim from [10 ] .
Construction A.1. Suppose there exists a σ \sigma σ -finite measure ν \nu ν such that P ≪ ν P \ll\nu P ≪ ν for every P ∈ P P \in\mathcal{P} P ∈ P . We write f P = d P d ν f_P = \frac{dP}{d\nu} f P = d ν d P for all P ∈ P P \in\mathcal{P} P ∈ P .
Let A \mathcal{A} A be a Poisson point process on Y × R + × R + \mathcal{Y} \times\mathbb{R}_{+} \times\mathbb{R}_{+} Y × R + × R + with intensity ν ( d y ) d s d t \nu(dy)dsdt ν ( d y ) d s d t (Definition 2.2 ). For a realization a a a of the Poisson point process A \mathcal{A} A , we let a P = { ( y , s , t ) ∈ a : s ≤ f P ( y ) } a_P = \{(y,s,t) \in a : s \le f_P(y)\} a P = {( y , s , t ) ∈ a : s ≤ f P ( y )} for all P ∈ P P \in\mathcal{P} P ∈ P , and let Φ ( P , a ) \Phi(P,a) Φ ( P , a ) denote the y y y -coordinate of the point in a P a_P a P having the smallest t t t -coordinate. More formally, if ( y ∗ , s ∗ , t ∗ ) (y^*,s^*,t^*) ( y ∗ , s ∗ , t ∗ ) is the unique point satisfying t ∗ = min { t : ( y , s , t ) ∈ a , s ≤ f P ( y ) } t^* = \min\{t : (y,s,t) \in a, s \le f_P(y)\} t ∗ = min { t : ( y , s , t ) ∈ a , s ≤ f P ( y )} , then Φ ( P , a ) = y ∗ \Phi(P,a) = y^* Φ ( P , a ) = y ∗ . Note that on the null event that there is a tie we can define Φ ( P , a ) \Phi(P,a) Φ ( P , a ) arbitrarily.7
Using the same random process A \mathcal{A} A for every P ∈ P P \in\mathcal{P} P ∈ P , we define the random variable Z P = Φ ( P , A ) Z_P = \Phi(P,\mathcal{A}) Z P = Φ ( P , A ) and let Γ = L ( ( Z P ) P ∈ P ) \Gamma= \mathcal{L}((Z_P)_{P \in\mathcal{P}}) Γ = L (( Z P ) P ∈ P ) be their joint law.
Algorithm 5 Couple I from construction A.1
Input: Distributions q 1 , … , q k ∈ F q_1,\ldots,q_k \in\mathcal{F} q 1 , … , q k ∈ F . A fixed σ \sigma σ -finite measure ν \nu ν on X ℓ \mathcal{X}^{\ell} X ℓ dominating every relevant q ℓ ∈ { q ℓ : q ∈ F } q^{\ell} \in\{q^{\ell}:q \in\mathcal{F}\} q ℓ ∈ { q ℓ : q ∈ F } .
Output: Synthetic samples ( Z 1 , … , Z k ) (Z_1,\ldots,Z_k) ( Z 1 , … , Z k ) such that Z i ∼ q i ℓ Z_i \sim q_i^{\ell} Z i ∼ q i ℓ and Pr [ Z i ≠ Z j ] ≤ 2 TV ( q i ℓ , q j ℓ ) \Pr[Z_i \ne Z_j] \le2\operatorname{TV}(q_i^{\ell},q_j^{\ell}) Pr [ Z i = Z j ] ≤ 2 TV ( q i ℓ , q j ℓ ) for all i , j ∈ [ k ] i,j \in[k] i , j ∈ [ k ] .
Let a random seed be A \mathcal{A} A , a Poisson point process with intensity ν ( d y ) d s d t \nu(dy)dsdt ν ( d y ) d s d t
Z i = Φ ( q i ℓ , A ) Z_i = \Phi(q_i^{\ell},\mathcal{A}) Z i = Φ ( q i ℓ , A ) for all i ∈ [ k ] i \in[k] i ∈ [ k ] according to construction A.1 ▹ \triangleright ▹ Conditional on a realization A = a \mathcal{A}=a A = a , i.e. conditional on a realization of the PPP, the output z i = Φ ( q i ℓ , a ) z_i=\Phi(q_i^{\ell},a) z i = Φ ( q i ℓ , a ) is deterministic.
return ( Z 1 , … , Z k ) (Z_1,\ldots,Z_k) ( Z 1 , … , Z k ) . ▹ \triangleright ▹ Equivalently, ( Z 1 , … , Z k ) (Z_1,\ldots,Z_k) ( Z 1 , … , Z k ) has the ( q 1 ℓ , … , q k ℓ ) (q_1^{\ell},\ldots,q_k^{\ell}) ( q 1 ℓ , … , q k ℓ ) -coordinate marginal of the coupling Γ \Gamma Γ from construction A.1
Theorem A.2 (Proposition 5 and Theorem 2 in [10 ] ). Suppose Y \mathcal{Y} Y is standard Borel and there exists a σ \sigma σ -finite measure ν \nu ν such that P ≪ ν P \ll\nu P ≪ ν for all P ∈ P P \in\mathcal{P} P ∈ P . The law Γ \Gamma Γ defined in Construction A.1, is a coupling of ( P ) P ∈ P (P)_{P \in\mathcal{P}} ( P ) P ∈ P , i.e. Z P ∼ P Z_P \sim P Z P ∼ P for every P ∈ P P \in\mathcal{P} P ∈ P . Moreover, for every P , Q ∈ P P,Q \in\mathcal{P} P , Q ∈ P ,
Pr [ Z P ≠ Z Q ] ≤ 2 TV ( P , Q ) 1 + TV ( P , Q ) ≤ 2 TV ( P , Q ) \Pr[Z_P \ne Z_Q] \le\frac{2\operatorname{TV}(P,Q)}{1+\operatorname{TV}(P,Q)} \le2\operatorname{TV}(P,Q) Pr [ Z P = Z Q ] ≤ 1 + TV ( P , Q ) 2 TV ( P , Q ) ≤ 2 TV ( P , Q )
Theorem A.2 establishes the core marginal and agreement guarantees for the coupling construction. However, to formally show that it constitutes a grand coupling scheme on P \mathcal{P} P , we must also verify the measurability requirement, as formalized below.
Corollary A.3 (Grand coupling scheme Γ P \Gamma_{\mathcal{P}} Γ P from Construction A.1). Suppose Y \mathcal{Y} Y is standard Borel and there exists a σ \sigma σ -finite measure ν \nu ν such that P ≪ ν P \ll\nu P ≪ ν for all P ∈ P P \in\mathcal{P} P ∈ P . Then the family of random variables ( Z P ) P ∈ P (Z_P)_{P \in\mathcal{P}} ( Z P ) P ∈ P defined in Construction A.1 constitutes a grand coupling scheme Γ P \Gamma_{\mathcal{P}} Γ P on P \mathcal{P} P .
Proof. G1 and G2 follow immediately from Theorem A.2. It suffices to show G3 that the map P ↦ L ( Z P 1 , … , Z P k ) \mathbf{P} \mapsto\mathcal{L}(Z_{P_1},\ldots,Z_{P_k}) P ↦ L ( Z P 1 , … , Z P k ) , where P = ( P 1 , … , P k ) \mathbf{P}=(P_1,\ldots,P_k) P = ( P 1 , … , P k ) with P i ∈ P P_i \in\mathcal{P} P i ∈ P , is a Markov kernel.
Since Y \mathcal{Y} Y is standard Borel and P ≪ ν P \ll\nu P ≪ ν for all P ∈ P P \in\mathcal{P} P ∈ P , the Radon–Nikodym derivatives may be chosen jointly measurably in ( P , y ) (P,y) ( P , y ) [Kal02], so write f ( P i , y ) = d P i d ν ( y ) f(P_i,y)=\frac{dP_i}{d\nu}(y) f ( P i , y ) = d ν d P i ( y ) . Let A \mathcal{A} A be a Poisson point process on Y × R + × R + \mathcal{Y} \times\mathbb{R}_{+} \times\mathbb{R}_{+} Y × R + × R + with intensity ν ( d y ) d s d t \nu(dy)dsdt ν ( d y ) d s d t . For every i ∈ [ k ] i \in[k] i ∈ [ k ] , define A i = { ( y , s , t ) ∈ A : s ≤ f ( P i , y ) } \mathcal{A}_i=\{(y,s,t)\in\mathcal{A}:s\leq f(P_i,y)\} A i = {( y , s , t ) ∈ A : s ≤ f ( P i , y )} . We define the random variable Z i Z_i Z i by Z i = y Z_i=y Z i = y , where ( y , s , t ) (y,s,t) ( y , s , t ) is the point of A i \mathcal{A}_i A i with the minimal t t t among all points of A i \mathcal{A}_i A i . Equivalently, we write Z i = Φ ( P i , A ) Z_i=\Phi(P_i,\mathcal{A}) Z i = Φ ( P i , A ) , where Φ \Phi Φ denotes this deterministic rule.
Now define Φ ( P , A ) = ( Φ ( P 1 , A ) , … , Φ ( P k , A ) ) \Phi(\mathbf{P},\mathcal{A})=(\Phi(P_1,\mathcal{A}),\ldots,\Phi(P_k,\mathcal{A})) Φ ( P , A ) = ( Φ ( P 1 , A ) , … , Φ ( P k , A )) and for measurable B ⊆ Y k B\subseteq\mathcal{Y}^k B ⊆ Y k , let Γ k ( P ; B ) = Pr A [ Φ ( P , A ) ∈ B ] \Gamma_k(\mathbf{P};B)=\Pr_{\mathcal{A}}[\Phi(\mathbf{P},\mathcal{A})\in B] Γ k ( P ; B ) = Pr A [ Φ ( P , A ) ∈ B ] , i.e. Γ k ( P ; ⋅ ) = L ( Φ ( P , A ) ) \Gamma_k(\mathbf{P};\cdot)=\mathcal{L}(\Phi(\mathbf{P},\mathcal{A})) Γ k ( P ; ⋅ ) = L ( Φ ( P , A )) . Since f ( P i , y ) f(P_i,y) f ( P i , y ) is jointly measurable, the indicator 1 { s ≤ f ( P i , y ) } \mathbb{1}\{s\leq f(P_i,y)\} 1 { s ≤ f ( P i , y )} is also jointly measurable; applying a measurable filter to a point process yields another valid point process, and applying the argmin on t t t -coordinate is also measurable; and thus, the deterministic map ( P , a ) ↦ Φ ( P , a ) (\mathbf{P},a)\mapsto\Phi(\mathbf{P},a) ( P , a ) ↦ Φ ( P , a ) is measurable. Therefore, P ↦ Γ k ( P ; B ) \mathbf{P}\mapsto\Gamma_k(\mathbf{P};B) P ↦ Γ k ( P ; B ) is also measurable for every measurable B ⊆ Y k B\subseteq\mathcal{Y}^k B ⊆ Y k , so Γ k \Gamma_k Γ k is a Markov kernel.
Thus, the joint law of ( Z P ) P ∈ P (Z_P)_{P\in\mathcal{P}} ( Z P ) P ∈ P yields the desired grand coupling Γ P \Gamma_{\mathcal{P}} Γ P on Y \mathcal{Y} Y .
Projectively Consistent Poisson Couplings on Standard Borel Spaces
The preceding construction requires a single σ \sigma σ -finite measure dominating every distribution in the family under consideration. However, such a measure may not generally exist. For example, the uncountable family of all point masses over the real line admits no common σ \sigma σ -finite dominating measure. In fact, on a standard Borel space, a family P ⊆ Δ ( Y ) \mathcal{P}\subseteq\Delta(\mathcal{Y}) P ⊆ Δ ( Y ) admits a common σ \sigma σ -finite dominating measure if and only if P \mathcal{P} P is TV separable. Thus the core issue is not uncountability per se, but rather TV nonseparability. For any finite tuple of probability measures P = ( P 1 , … , P k ) ∈ Δ ( Y ) k \mathbf{P}=(P_1,\ldots,P_k)\in\Delta(\mathcal{Y})^k P = ( P 1 , … , P k ) ∈ Δ ( Y ) k , however, there is always a canonical finite dominating measure, ν P = ∑ i P i \nu_{\mathbf{P}}=\sum_i P_i ν P = ∑ i P i . Thus, for each fixed finite tuple, the Poisson construction A.1 can be applied with the tuple-dependent reference measure ν P \nu_{\mathbf{P}} ν P .
Construction A.4. Fix k ≥ 1 k\geq1 k ≥ 1 and a tuple P = ( P 1 , … , P k ) ∈ Δ ( Y ) k \mathbf{P}=(P_1,\ldots,P_k)\in\Delta(\mathcal{Y})^k P = ( P 1 , … , P k ) ∈ Δ ( Y ) k . Define the P \mathbf{P} P -dependent measure ν P = ∑ i P i \nu_{\mathbf{P}}=\sum_i P_i ν P = ∑ i P i , which is finite, and let f i P = d P i d ν P f_i^{\mathbf{P}}=\frac{dP_i}{d\nu_{\mathbf{P}}} f i P = d ν P d P i be the corresponding densities. Let A P \mathcal{A}_{\mathbf{P}} A P be a Poisson point process on Y × R + × R + \mathcal{Y}\times\mathbb{R}_{+}\times\mathbb{R}_{+} Y × R + × R + with intensity ν P ( d y ) d s d t \nu_{\mathbf{P}}(dy)dsdt ν P ( d y ) d s d t (Definition 2.2 ). Unlike Construction A.1, the law of A P \mathcal{A}_{\mathbf{P}} A P depends on the entire tuple P \mathbf{P} P .
*For a realization a P \mathbf{a}_{\mathbf{P}} a P of the Poisson point process A P \mathcal{A}_{\mathbf{P}} A P , we let Φ i ( P , a P ) \Phi_i(\mathbf{P},\mathbf{a}_{\mathbf{P}}) Φ i ( P , a P ) denote the y y y -coordinate of the point in a P , i \mathbf{a}_{\mathbf{P},i} a P , i having the smallest t t t -coordinate, where the acceptance regions are a P , i = { ( y , s , t ) ∈ a P : s ≤ f i P ( y ) } \mathbf{a}_{\mathbf{P},i}=\{(y,s,t)\in\mathbf{a}_{\mathbf{P}}:s\leq f_i^{\mathbf{P}}(y)\} a P , i = {( y , s , t ) ∈ a P : s ≤ f i P ( y )} for all i ∈ [ k ] i\in[k] i ∈ [ k ] .8
*Using the same random process A P \mathcal{A}_{\mathbf{P}} A P for every i ∈ [ k ] i\in[k] i ∈ [ k ] ,9 we define the random variable Z P , i = Φ i ( P , A P ) Z_{\mathbf{P},i}=\Phi_i(\mathbf{P},\mathcal{A}_{\mathbf{P}}) Z P , i = Φ i ( P , A P ) , and for every measurable B ⊆ Y k B\subseteq\mathcal{Y}^k B ⊆ Y k , define Γ k ( P ; B ) = Pr [ ( Z P , 1 , … , Z P , k ) ∈ B ] \Gamma_k(\mathbf{P};B)=\Pr[(Z_{\mathbf{P},1},\ldots,Z_{\mathbf{P},k})\in B] Γ k ( P ; B ) = Pr [( Z P , 1 , … , Z P , k ) ∈ B ] , i.e.,*
( Z P , 1 , … , Z P , k ) ∼ Γ k ( P ; ⋅ ) (Z_{\mathrm{P},1},\ldots,Z_{\mathrm{P},k}) \sim\Gamma_k(\mathbf{P};\cdot) ( Z P , 1 , … , Z P , k ) ∼ Γ k ( P ; ⋅ ) . Thus, Γ k ( P ; ⋅ ) \Gamma_k(\mathbf{P};\cdot) Γ k ( P ; ⋅ ) is the joint law induced by the P \mathbf{P} P -specific Poisson coupling.
Algorithm 6 Couple II from construction A.4
Input: Distributions q 1 , … , q k ∈ F q_1,\ldots,q_k \in\mathcal{F} q 1 , … , q k ∈ F .
Output: Synthetic samples ( Z 1 , … , Z k ) (Z_1,\ldots,Z_k) ( Z 1 , … , Z k ) such that Z i ∼ q i ℓ Z_i \sim q_i^\ell Z i ∼ q i ℓ and Pr [ Z i ≠ Z j ] ≤ 2 TV ( q i ℓ , q j ℓ ) \Pr[Z_i \ne Z_j] \le2\operatorname{TV}(q_i^\ell,q_j^\ell) Pr [ Z i = Z j ] ≤ 2 TV ( q i ℓ , q j ℓ ) for all i , j ∈ [ k ] i,j \in[k] i , j ∈ [ k ] .
Set P i = q i ℓ P_i=q_i^\ell P i = q i ℓ for all i ∈ [ k ] i\in[k] i ∈ [ k ] , ν P = ∑ i P i \nu_{\mathbf{P}}=\sum_i P_i ν P = ∑ i P i , and f i ν P = d P i d ν P f_i^{\nu_{\mathbf{P}}}=\frac{\mathrm{d}P_i}{\mathrm{d}\nu_{\mathbf{P}}} f i ν P = d ν P d P i
Let a random seed be A P \mathcal{A}_{\mathbf{P}} A P , a Poisson point process with intensity ν P ( d y ) d s d t \nu_{\mathbf{P}}(\mathrm{d}y)\mathrm{d}s\mathrm{d}t ν P ( d y ) d s d t
Z i = Φ i ( P , A P ) Z_i=\Phi_i(\mathbf{P},\mathcal{A}_{\mathbf{P}}) Z i = Φ i ( P , A P ) for all i ∈ [ k ] i\in[k] i ∈ [ k ] according to construction A.4 ▹ \triangleright ▹ Conditional on a realization A P = a P \mathcal{A}_{\mathbf{P}}=a_{\mathbf{P}} A P = a P , the output z i = Φ i ( P , a P ) z_i=\Phi_i(\mathbf{P},a_{\mathbf{P}}) z i = Φ i ( P , a P ) is deterministic.
return ( Z 1 , … , Z k ) (Z_1,\ldots,Z_k) ( Z 1 , … , Z k ) . ▹ \triangleright ▹ Equivalently, ( Z 1 , … , Z k ) ∼ Γ k ( P ; ⋅ ) (Z_1,\ldots,Z_k)\sim\Gamma_k(\mathbf{P};\cdot) ( Z 1 , … , Z k ) ∼ Γ k ( P ; ⋅ ) from construction A.4
The key extra ingredients are to show that these tuple-specific couplings vary measurably with P \mathbf{P} P and are projective consistent across different tuples. Measurability allows the construction to be applied to (data-dependent) random distributions, while projective consistency allows finite couplings built from different tuples to be embedded into a common larger coupling.
For a measurable map T : S → T T:\mathcal{S}\to\mathcal{T} T : S → T and a measure μ \mu μ on S \mathcal{S} S , we write T # μ T_{\#}\mu T # μ for the pushforward measure of μ \mu μ under T T T , defined by ( T # μ ) ( B ) = μ ( T − 1 ( B ) ) (T_{\#}\mu)(B)=\mu(T^{-1}(B)) ( T # μ ) ( B ) = μ ( T − 1 ( B )) for all measurable B ⊆ T B\subseteq\mathcal{T} B ⊆ T . For finite products, write pr i ( y 1 , … , y k ) = y i \operatorname{pr}_i(y_1,\ldots,y_k)=y_i pr i ( y 1 , … , y k ) = y i for the i i i -th coordinate projection.
Proposition A.5 (Projectively consistent Poisson coupling kernels). Let Y \mathcal{Y} Y be standard Borel. For every k ≥ 1 k\ge1 k ≥ 1 , Construction A.4 defines a map Γ k : Δ ( Y ) k → Δ ( Y k ) \Gamma_k:\Delta(\mathcal{Y})^k\to\Delta(\mathcal{Y}^k) Γ k : Δ ( Y ) k → Δ ( Y k ) , with the following properties. Write Γ k ( P ; ⋅ ) \Gamma_k(\mathbf{P};\cdot) Γ k ( P ; ⋅ ) for the measure Γ k ( P ) \Gamma_k(\mathbf{P}) Γ k ( P ) , and let D i j = { ( y 1 , … , y k ) ∈ Y k : y i ≠ y j } D_{ij}=\{(y_1,\ldots,y_k)\in\mathcal{Y}^k:y_i\ne y_j\} D ij = {( y 1 , … , y k ) ∈ Y k : y i = y j } .
(1) (Measurability.) For every measurable B ⊆ Y k B\subseteq\mathcal{Y}^k B ⊆ Y k , the map P ↦ Γ k ( P ; B ) \mathbf{P}\mapsto\Gamma_k(\mathbf{P};B) P ↦ Γ k ( P ; B ) is measurable with respect to the evaluation σ \sigma σ -algebra on Δ ( Y ) k \Delta(\mathcal{Y})^k Δ ( Y ) k ; that is, Γ k \Gamma_k Γ k is a Markov kernel from Δ ( Y ) k \Delta(\mathcal{Y})^k Δ ( Y ) k to Y k \mathcal{Y}^k Y k .
(2) (Marginals.) For every P = ( P 1 , … , P k ) \mathbf{P}=(P_1,\ldots,P_k) P = ( P 1 , … , P k ) and i ∈ [ k ] i\in[k] i ∈ [ k ] , ( pr i ) # Γ k ( P ; ⋅ ) = P i (\operatorname{pr}_i)_{\#}\Gamma_k(\mathbf{P};\cdot)=P_i ( pr i ) # Γ k ( P ; ⋅ ) = P i .
(3) (Disagreement bound.) For every P \mathbf{P} P and i , j ∈ [ k ] i,j\in[k] i , j ∈ [ k ] , Γ k ( P ; D i j ) ≤ 2 TV ( P i , P j ) 1 + TV ( P i , P j ) ≤ 2 TV ( P i , P j ) \Gamma_k(\mathbf{P};D_{ij})\le\frac{2\operatorname{TV}(P_i,P_j)}{1+\operatorname{TV}(P_i,P_j)}\le2\operatorname{TV}(P_i,P_j) Γ k ( P ; D ij ) ≤ 1 + TV ( P i , P j ) 2 TV ( P i , P j ) ≤ 2 TV ( P i , P j ) .
(4) (Projective consistency.) For every nonempty E = { i 1 , … , i k ′ } ⊆ [ k ] E=\{i_1,\ldots,i_{k'}\}\subseteq[k] E = { i 1 , … , i k ′ } ⊆ [ k ] with i 1 < ⋯ < i k ′ i_1<\cdots<i_{k'} i 1 < ⋯ < i k ′ and P ′ = ( P i ) i ∈ E \mathbf{P}'=(P_i)_{i\in E} P ′ = ( P i ) i ∈ E in the original order, ( pr E ) # Γ k ( P ; ⋅ ) = Γ k ′ ( P ′ ; ⋅ ) (\operatorname{pr}_E)_{\#}\Gamma_k(\mathbf{P};\cdot)=\Gamma_{k'}(\mathbf{P}';\cdot) ( pr E ) # Γ k ( P ; ⋅ ) = Γ k ′ ( P ′ ; ⋅ ) , where pr E ( y 1 , … , y k ) = ( y i 1 , … , y i k ′ ) \operatorname{pr}_E(y_1,\ldots,y_k)=(y_{i_1},\ldots,y_{i_{k'}}) pr E ( y 1 , … , y k ) = ( y i 1 , … , y i k ′ ) .
(5) (Permutation equivariance.) For every permutation σ \sigma σ of [ k ] [k] [ k ] : if ( Z 1 , … , Z k ) ∼ Γ k ( ( P 1 , … , P k ) ; ⋅ ) (Z_1,\ldots,Z_k)\sim\Gamma_k((P_1,\ldots,P_k);\cdot) ( Z 1 , … , Z k ) ∼ Γ k (( P 1 , … , P k ) ; ⋅ ) , then ( Z σ ( 1 ) , … , Z σ ( k ) ) ∼ Γ k ( ( P σ ( 1 ) , … , P σ ( k ) ) ; ⋅ ) (Z_{\sigma(1)},\ldots,Z_{\sigma(k)})\sim\Gamma_k((P_{\sigma(1)},\ldots,P_{\sigma(k)});\cdot) ( Z σ ( 1 ) , … , Z σ ( k ) ) ∼ Γ k (( P σ ( 1 ) , … , P σ ( k ) ) ; ⋅ ) .
Proof. Now, we prove these properties.
Measurability Fix k ≥ 1 k\ge1 k ≥ 1 and let P = ( P 1 , … , P k ) ∈ Δ ( Y ) k \mathbf{P}=(P_1,\ldots,P_k)\in\Delta(\mathcal{Y})^k P = ( P 1 , … , P k ) ∈ Δ ( Y ) k . Define ν P = ∑ i P i \nu_{\mathbf{P}}=\sum_i P_i ν P = ∑ i P i , so P i ≪ ν P P_i\ll\nu_{\mathbf{P}} P i ≪ ν P for all i ∈ [ k ] i\in[k] i ∈ [ k ] . Since Y \mathcal{Y} Y is standard Borel, the Radon–Nikodym derivatives may be chosen jointly measurably in ( P , y ) (\mathbf{P},y) ( P , y ) , so write f i ( P , y ) = d P i d ν P ( y ) f_i(\mathbf{P},y)=\frac{\mathrm{d}P_i}{\mathrm{d}\nu_{\mathbf{P}}}(y) f i ( P , y ) = d ν P d P i ( y ) . Since P i ≪ ν P P_i\ll\nu_{\mathbf{P}} P i ≪ ν P , we may take 0 ≤ f i ( P , y ) ≤ 1 0\le f_i(\mathbf{P},y)\le1 0 ≤ f i ( P , y ) ≤ 1 ν P \nu_{\mathbf{P}} ν P -a.e.
For each P \mathbf{P} P , let A P \mathcal{A}_{\mathbf{P}} A P be a Poisson point process on Y × R + × R + \mathcal{Y}\times\mathbb{R}_+\times\mathbb{R}_+ Y × R + × R + with intensity ν P ( d y ) d s d t \nu_{\mathbf{P}}(\mathrm{d}y)\mathrm{d}s\mathrm{d}t ν P ( d y ) d s d t . For every i ∈ [ k ] i\in[k] i ∈ [ k ] , define A P , i = { ( y , s , t ) ∈ A P : s ≤ f i ( P , y ) } \mathcal{A}_{\mathbf{P},i}=\{(y,s,t)\in\mathcal{A}_{\mathbf{P}}:s\le f_i(\mathbf{P},y)\} A P , i = {( y , s , t ) ∈ A P : s ≤ f i ( P , y )} . We define the random variable Z i Z_i Z i by Z i = y Z_i=y Z i = y , where ( y , s , t ) (y,s,t) ( y , s , t ) is the point of A P , i \mathcal{A}_{\mathbf{P},i} A P , i with the minimal t t t among all points of A P , i \mathcal{A}_{\mathbf{P},i} A P , i . Equivalently, we write Z i = Φ i ( P , A P ) Z_i=\Phi_i(\mathbf{P},\mathcal{A}_{\mathbf{P}}) Z i = Φ i ( P , A P ) , where Φ i \Phi_i Φ i denotes this deterministic rule. Now define Φ ( P , A P ) = ( Φ 1 ( P , A P ) , … , Φ k ( P , A P ) ) \Phi(\mathbf{P},\mathcal{A}_{\mathbf{P}})=(\Phi_1(\mathbf{P},\mathcal{A}_{\mathbf{P}}),\ldots,\Phi_k(\mathbf{P},\mathcal{A}_{\mathbf{P}})) Φ ( P , A P ) = ( Φ 1 ( P , A P ) , … , Φ k ( P , A P )) and for measurable B ⊆ Y k B\subseteq\mathcal{Y}^k B ⊆ Y k , let Γ k ( P ; B ) = Pr A P [ Φ ( P , A P ) ∈ B ] \Gamma_k(\mathbf{P};B)=\Pr_{\mathcal{A}_{\mathbf{P}}}[\Phi(\mathbf{P},\mathcal{A}_{\mathbf{P}})\in B] Γ k ( P ; B ) = Pr A P [ Φ ( P , A P ) ∈ B ] , i.e. Γ k ( P ; ⋅ ) = L ( Φ ( P , A P ) ) \Gamma_k(\mathbf{P};\cdot)=\mathcal{L}(\Phi(\mathbf{P},\mathcal{A}_{\mathbf{P}})) Γ k ( P ; ⋅ ) = L ( Φ ( P , A P )) . Since Y \mathcal{Y} Y is standard Borel, the intensity measure ν P ( d y ) d s d t \nu_{\mathbf{P}}(dy)dsdt ν P ( d y ) d s d t depends measurably on P \mathbf{P} P , and since the law of any PPP is uniquely and completely determined by its intensity measure, it follows that the P \mathbf{P} P -dependent PPP law varies measurably with P \mathbf{P} P . Also, f i ( P , y ) f_i(\mathbf{P},y) f i ( P , y ) is jointly measurable, so the indicator 1 { g ( P , y , s ) ≥ 0 } \mathbf{1}\{g(\mathbf{P},y,s)\geq0\} 1 { g ( P , y , s ) ≥ 0 } is also jointly measurable; applying a measurable filter to a point process yields another valid point process, and applying the argmin on t t t -coordinate is also measurable; and thus, the deterministic map ( P , a ) ↦ Φ ( P , a ) (\mathbf{P},a)\mapsto\Phi(\mathbf{P},a) ( P , a ) ↦ Φ ( P , a ) is measurable. Therefore, P ↦ Γ k ( P ; B ) \mathbf{P}\mapsto\Gamma_k(\mathbf{P};B) P ↦ Γ k ( P ; B ) is also measurable for every measurable B ⊆ Y k B\subseteq\mathcal{Y}^k B ⊆ Y k , so Γ k \Gamma_k Γ k is a Markov kernel.
2. Marginal To see that Z i Z_i Z i has the required law (so that the above is indeed a coupling), we note that A P , i \mathcal{A}_{\mathbf{P},i} A P , i has intensity f i ( P , y ) ν P ( d y ) d t = P i ( d y ) d t f_i(\mathbf{P},y)\nu_{\mathbf{P}}(dy)dt=P_i(dy)dt f i ( P , y ) ν P ( d y ) d t = P i ( d y ) d t , so the y y y -coordinate of its first point, i.e. Φ i ( P , A P ) \Phi_i(\mathbf{P},\mathcal{A}_{\mathbf{P}}) Φ i ( P , A P ) , has law P i P_i P i . Hence, ( pr i ) # Γ k ( P ; ⋅ ) = P i (\operatorname{pr}_i)_{\#}\Gamma_k(\mathbf{P};\cdot)=P_i ( pr i ) # Γ k ( P ; ⋅ ) = P i
3. Disagreement bound Fix i , j ∈ [ k ] i,j\in[k] i , j ∈ [ k ] , and denote g = f i ( P , y ) g=f_i(\mathbf{P},y) g = f i ( P , y ) , h = f j ( P , y ) h=f_j(\mathbf{P},y) h = f j ( P , y ) , α = TV ( P i , P j ) \alpha=\operatorname{TV}(P_i,P_j) α = TV ( P i , P j ) . Consider the first point ( y , s ) (y,s) ( y , s ) to appear that has s ≤ g ( y ) ∨ h ( y ) s\leq g(y)\vee h(y) s ≤ g ( y ) ∨ h ( y ) . If it happens that s ≤ g ( y ) ∧ h ( y ) s\leq g(y)\wedge h(y) s ≤ g ( y ) ∧ h ( y ) , then we get the agreement Z i = Z j = y Z_i=Z_j=y Z i = Z j = y ; otherwise, this point determines either the value of Z i Z_i Z i or Z j Z_j Z j . Hence, for any measurable A A A , the intensity of the region accepted by both is ∫ A ( g ∧ h ) d ν P \int_A(g\wedge h)d\nu_{\mathbf{P}} ∫ A ( g ∧ h ) d ν P and the intensity of the region accepted by at least one is ∫ A ( g ∨ h ) d ν P \int_A(g\vee h)d\nu_{\mathbf{P}} ∫ A ( g ∨ h ) d ν P . Therefore,
Pr [ Z i = Z j ] ≥ Pr [ Z i = Z j ∈ A and the same point determines both Z i and Z j ] = ∫ A ( g ∧ h ) d ν P ∫ A ( g ∨ h ) d ν P . \Pr[Z_i=Z_j]\geq\Pr[Z_i=Z_j\in A\text{ and the same point determines both }Z_i\text{ and }Z_j]=\frac{\int_A(g\wedge h)d\nu_{\mathbf{P}}}{\int_A(g\vee h)d\nu_{\mathbf{P}}}. Pr [ Z i = Z j ] ≥ Pr [ Z i = Z j ∈ A and the same point determines both Z i and Z j ] = ∫ A ( g ∨ h ) d ν P ∫ A ( g ∧ h ) d ν P .
Note that ∫ Y ( g ∨ h ) d ν P = 1 + α \int_{\mathcal{Y}}(g\vee h)d\nu_{\mathbf{P}}=1+\alpha ∫ Y ( g ∨ h ) d ν P = 1 + α and ∫ Y ( g ∧ h ) d ν P = 1 − α \int_{\mathcal{Y}}(g\wedge h)d\nu_{\mathbf{P}}=1-\alpha ∫ Y ( g ∧ h ) d ν P = 1 − α , so it follows that Pr [ Z i = Z j ] ≥ 1 − α 1 + α \Pr[Z_i=Z_j]\geq\frac{1-\alpha}{1+\alpha} Pr [ Z i = Z j ] ≥ 1 + α 1 − α , i.e. Pr [ Z i ≠ Z j ] ≤ 1 − 1 − α 1 + α ≤ 2 α \Pr[Z_i\neq Z_j]\leq1-\frac{1-\alpha}{1+\alpha}\leq2\alpha Pr [ Z i = Z j ] ≤ 1 − 1 + α 1 − α ≤ 2 α . Thus, Γ k ( P ; D i j ) ≤ 2 TV ( P i , P j ) 1 + TV ( P i , P j ) ≤ 2 TV ( P i , P j ) \Gamma_k(\mathbf{P};D_{ij})\leq\frac{2\operatorname{TV}(P_i,P_j)}{1+\operatorname{TV}(P_i,P_j)}\leq2\operatorname{TV}(P_i,P_j) Γ k ( P ; D ij ) ≤ 1 + TV ( P i , P j ) 2 TV ( P i , P j ) ≤ 2 TV ( P i , P j ) , for D i j = { ( y 1 , … , y k ) ∈ Y k : y i ≠ y j } D_{ij}=\{(y_1,\ldots,y_k)\in\mathcal{Y}^k:y_i\neq y_j\} D ij = {( y 1 , … , y k ) ∈ Y k : y i = y j } .
4. Projective consistency Let E ⊆ [ k ] E\subseteq[k] E ⊆ [ k ] be nonempty. Let P ′ = ( P i ) i ∈ E \mathbf{P}'=(P_i)_{i\in E} P ′ = ( P i ) i ∈ E and define ν P ′ = ∑ i ∈ E P i \nu_{\mathbf{P}'}=\sum_{i\in E}P_i ν P ′ = ∑ i ∈ E P i and f i ( P ′ , y ) = d P i d ν P ′ ( y ) f_i(\mathbf{P}',y)=\frac{dP_i}{d\nu_{\mathbf{P}'}}(y) f i ( P ′ , y ) = d ν P ′ d P i ( y ) for i ∈ E i\in E i ∈ E . Thus, by the chain rule, we have f i ( P , y ) = r ( y ) f i ( P ′ , y ) f_i(\mathbf{P},y)=r(y)f_i(\mathbf{P}',y) f i ( P , y ) = r ( y ) f i ( P ′ , y ) , where r ( y ) = d ν P ′ d ν P ( y ) r(y)=\frac{d\nu_{\mathbf{P}'}}{d\nu_{\mathbf{P}}}(y) r ( y ) = d ν P d ν P ′ ( y ) ν P \nu_{\mathbf{P}} ν P -a.e., for all i ∈ E i\in E i ∈ E . Let Y + = { y : r ( y ) > 0 } \mathcal{Y}_+=\{y:r(y)>0\} Y + = { y : r ( y ) > 0 } . Since ν P ′ ( Y ∖ Y + ) = 0 \nu_{\mathbf{P}'}(\mathcal{Y}\setminus\mathcal{Y}_+)=0 ν P ′ ( Y ∖ Y + ) = 0 and P i ≪ ν P ′ P_i\ll\nu_{\mathbf{P}'} P i ≪ ν P ′ for every i ∈ E i\in E i ∈ E , we also have P i ( Y ∖ Y + ) = 0 P_i(\mathcal{Y}\setminus\mathcal{Y}_+)=0 P i ( Y ∖ Y + ) = 0 . Equivalently, the portion of the acceptance region for coordinate i i i lying over Y ∖ Y + \mathcal{Y}\setminus\mathcal{Y}_+ Y ∖ Y + has zero intensity. Hence, almost surely, no point with r ( y ) = 0 r(y)=0 r ( y ) = 0 is accepted by any coordinate i ∈ E i\in E i ∈ E , and these points may be discarded.
Note that under the change of variable ( y , s , t ) ↦ ( y , s r ( y ) , t ) (y,s,t)\mapsto\left(y,\frac{s}{r(y)},t\right) ( y , s , t ) ↦ ( y , r ( y ) s , t ) , the restriction of A P \mathcal{A}_{\mathbf{P}} A P on Y + × R + × R + \mathcal{Y}_+\times\mathbb{R}_+\times\mathbb{R}_+ Y + × R + × R + is transformed into a PPP with intensity ν P ′ ( d y ) d u d t \nu_{\mathbf{P}'}(dy)dudt ν P ′ ( d y ) d u d t , for u = s / r ( y ) u=s/r(y) u = s / r ( y ) , since ν P ( d y ) d s d t = ν P ( d y ) r ( y ) d u d t = ν P ′ ( d y ) d u d t \nu_{\mathbf{P}}(dy)dsdt=\nu_{\mathbf{P}}(dy)r(y)dudt=\nu_{\mathbf{P}'}(dy)dudt ν P ( d y ) d s d t = ν P ( d y ) r ( y ) d u d t = ν P ′ ( d y ) d u d t . Moreover, for all i ∈ E i\in E i ∈ E , s ≤ f i P ( y ) s\leq f_i^{\mathbf{P}}(y) s ≤ f i P ( y ) if and only if s r ( y ) ≤ f i ( P ′ , y ) ⟺ u ≤ f i ( P ′ , y ) \frac{s}{r(y)}\leq f_i(\mathbf{P}',y)\Longleftrightarrow u\leq f_i(\mathbf{P}',y) r ( y ) s ≤ f i ( P ′ , y ) ⟺ u ≤ f i ( P ′ , y ) , so after rescaling the s s s -coordinate, the transformation does not change the y y y - and t t t -coordinate, i.e. it preserves the first accepted point for every i ∈ E i\in E i ∈ E . In other words, if some point ( y , s , t ) (y,s,t) ( y , s , t ) was the first accepted point for coordinate i i i in the full construction, its transformed version ( y , s / r ( y ) , t ) (y,s/r(y),t) ( y , s / r ( y ) , t ) has the same y y y and t t t , and is accepted in the subtuple construction. Therefore, the joint law of the coordinates indexed by E E E obtained from the full construction is exactly the law obtained by applying the Poisson construction directly to P ′ \mathbf{P}' P ′ . Hence, ( pr E ) # Γ k ( P ; ⋅ ) = Γ k ′ ( P ′ ; ⋅ ) (\operatorname{pr}_E)_{\#}\Gamma_k(\mathbf{P};\cdot)=\Gamma_{k'}(\mathbf{P}';\cdot) ( pr E ) # Γ k ( P ; ⋅ ) = Γ k ′ ( P ′ ; ⋅ ) , for k ′ = ∣ E ∣ k'=|E| k ′ = ∣ E ∣ .
5. Permutation equivariance Let P = ( P 1 , … , P k ) \mathbf{P}=(P_1,\ldots,P_k) P = ( P 1 , … , P k ) and denote P σ = ( P σ ( 1 ) , … , P σ ( k ) ) \mathbf{P}_{\sigma}=(P_{\sigma(1)},\ldots,P_{\sigma(k)}) P σ = ( P σ ( 1 ) , … , P σ ( k ) ) for an arbitrary permutation σ \sigma σ of [ k ] [k] [ k ] . Note that the dominating measure is invariant under permutations, i.e. ν P = ∑ i P i = ∑ i P σ ( i ) = ν P σ \nu_{\mathbf{P}}=\sum_iP_i=\sum_iP_{\sigma(i)}=\nu_{\mathbf{P}_{\sigma}} ν P = ∑ i P i = ∑ i P σ ( i ) = ν P σ . Moreover, for every i ∈ [ k ] i\in[k] i ∈ [ k ] , f i ( P σ , y ) = d P σ ( i ) d ν P σ = d P σ ( i ) d ν P = f σ ( i ) ( P , ⋅ ) f_i(\mathbf{P}_{\sigma},y)=\frac{dP_{\sigma(i)}}{d\nu_{\mathbf{P}_{\sigma}}}=\frac{dP_{\sigma(i)}}{d\nu_{\mathbf{P}}}=f_{\sigma(i)}(\mathbf{P},\cdot) f i ( P σ , y ) = d ν P σ d P σ ( i ) = d ν P d P σ ( i ) = f σ ( i ) ( P , ⋅ ) ν P \nu_{\mathbf{P}} ν P -a.e. Also note that a Poisson point process A P \mathcal{A}_{\mathbf{P}} A P on Y × R + × R + \mathcal{Y}\times\mathbb{R}_+\times\mathbb{R}_+ Y × R + × R + with intensity ν P ( d y ) d s d t \nu_{\mathbf{P}}(dy)dsdt ν P ( d y ) d s d t is also the Poisson point process used in the construction for P σ \mathbf{P}_{\sigma} P σ , since ν P = ν P σ \nu_{\mathbf{P}}=\nu_{\mathbf{P}_{\sigma}} ν P = ν P σ . Thus, the acceptance regions are the same a.s. as well, i.e. A P σ , i = { ( y , s , t ) ∈ A P : s ≤ f i ( P σ , y ) } = { ( y , s , t ) ∈ A P : s ≤ f σ ( i ) ( P , y ) } = A P , σ ( i ) \mathcal{A}_{\mathbf{P}_{\sigma},i}=\{(y,s,t)\in\mathcal{A}_{\mathbf{P}}:s\leq f_i(\mathbf{P}_{\sigma},y)\}=\{(y,s,t)\in\mathcal{A}_{\mathbf{P}}:s\leq f_{\sigma(i)}(\mathbf{P},y)\}=\mathcal{A}_{\mathbf{P},\sigma(i)} A P σ , i = {( y , s , t ) ∈ A P : s ≤ f i ( P σ , y )} = {( y , s , t ) ∈ A P : s ≤ f σ ( i ) ( P , y )} = A P , σ ( i ) . Hence, if we similarly define Z σ , i = Φ i ( P σ , A P ) Z_{\sigma,i}=\Phi_i(\mathbf{P}_{\sigma},\mathcal{A}_{\mathbf{P}}) Z σ , i = Φ i ( P σ , A P ) , then we have Z σ , i = Φ i ( P σ , A P ) = Φ σ ( i ) ( P , A P ) = Z σ ( i ) Z_{\sigma,i}=\Phi_i(\mathbf{P}_{\sigma},\mathcal{A}_{\mathbf{P}})=\Phi_{\sigma(i)}(\mathbf{P},\mathcal{A}_{\mathbf{P}})=Z_{\sigma(i)} Z σ , i = Φ i ( P σ , A P ) = Φ σ ( i ) ( P , A P ) = Z σ ( i ) , where Z i = Φ i ( P , A P ) Z_i=\Phi_i(\mathbf{P},\mathcal{A}_{\mathbf{P}}) Z i = Φ i ( P , A P ) . Thus, ( Z σ , 1 , … , Z σ , k ) ∼ Γ k ( P σ ; ⋅ ) (Z_{\sigma,1},\ldots,Z_{\sigma,k})\sim\Gamma_k(\mathbf{P}_{\sigma};\cdot) ( Z σ , 1 , … , Z σ , k ) ∼ Γ k ( P σ ; ⋅ ) □
There are two aspects of A.5 worth remarking. First, the standard Borel machinery is not needed to construct the coupling for a fixed tuple P \mathbf{P} P . In fact, for fixed P \mathbf{P} P , the Construction A.4 is simply the globally-dominated construction as in Construction A.1 using the P \mathbf{P} P -dependent dominating measure ν P = ∑ i P i \nu_{\mathbf{P}} = \sum_i P_i ν P = ∑ i P i . The standard-Borel assumption is used to ensure that the relevant Radon-Nikodym derivatives and Poisson law can be parameterized measurably in P \mathbf{P} P so that Γ k \Gamma_k Γ k is indeed a Markov kernel. This is what allows us later to sample from Γ k ( P ; ⋅ ) \Gamma_k(\mathbf{P};\cdot) Γ k ( P ; ⋅ ) when P \mathbf{P} P itself is a random data-dependent tuple of learner outputs.
Second, projective consistency allows couplings laws associated with different tuples to be realized as marginals of a single larger coupling. For example, given P = ( P 1 , … , P k ) \mathbf{P} = (P_1,\ldots,P_k) P = ( P 1 , … , P k ) and Q = ( Q 1 , … , Q k ) \mathbf{Q} = (Q_1,\ldots,Q_k) Q = ( Q 1 , … , Q k ) , and consider Γ 2 k ( ( P 1 , … , P k , Q 1 , … , Q k ) ) ; ⋅ ) \Gamma_{2k}((P_1,\ldots,P_k,Q_1,\ldots,Q_k));\cdot) Γ 2 k (( P 1 , … , P k , Q 1 , … , Q k )) ; ⋅ ) . Projective consistency guarantees that its first k k k coordinates have law Γ k ( P ; ⋅ ) \Gamma_k(\mathbf{P};\cdot) Γ k ( P ; ⋅ ) , and its last k k k coordinates have law Γ k ( Q ; ⋅ ) \Gamma_k(\mathbf{Q};\cdot) Γ k ( Q ; ⋅ ) . Thus, the larger coupling jointly realizes the two coupling laws without changing either of their original joint distributions. This is the key property that we will need for privacy, where we couple two neighboring executions, and utility analyses, where we enlarge the current execution by adjoining the ground-truth law.
It is also worth noting the differences between the two constructions. Under a common σ \sigma σ -finite dominating measure ν \nu ν , construction A.1 produces a single coupling ( Z P ) P ∈ P (Z_P)_{P\in\mathcal{P}} ( Z P ) P ∈ P for arbitrary (potentially uncountable) family P ⊆ Δ ( Y ) \mathcal{P} \subseteq\Delta(\mathcal{Y}) P ⊆ Δ ( Y ) , using one single global Poisson point process. On the other hand, construction A.4 assumes no global dominating measure , and for each finite tuple P \mathbf{P} P , constructs a tuple-specific coupling using the canonical dominating measure ν P \nu_{\mathbf{P}} ν P that is also tuple-dependent. Although for a different finite tuple Q \mathbf{Q} Q , the constructions generally uses a different reference measure ν Q \nu_{\mathbf{Q}} ν Q and hence a different PPP law and coupling, the projection consistency allows us to create a larger coupling of the concatenated tuple, such that the coordinates corresponding to Q \mathbf{Q} Q have law Γ k ( Q ; ⋅ ) \Gamma_k(\mathbf{Q};\cdot) Γ k ( Q ; ⋅ ) , and the rest have law Γ k ( P ; ⋅ ) \Gamma_k(\mathbf{P};\cdot) Γ k ( P ; ⋅ ) .
Despite this bottom-up approach, we can still establish a post-hoc universal coupling over all of Δ ( Y ) \Delta(\mathcal{Y}) Δ ( Y ) via the Kolmogorov Extension Theorem [Kal02]. Specifically, the projective consistency established in Proposition A.5 allows us to stitch the entire family of finite-dimensional couplings ( Γ k ) k ≥ 1 (\Gamma_k)_{k\geq1} ( Γ k ) k ≥ 1 together into a single, global probability measure over the uncountable product space Y Δ ( Y ) \mathcal{Y}^{\Delta(\mathcal{Y})} Y Δ ( Y ) , yielding a simultaneous coupling for every possible distribution in Δ ( Y ) \Delta(\mathcal{Y}) Δ ( Y ) . Consequently, the conceptual takeaway for both constructions is identical. In Construction A.1, one conceptually draws the family ( Z P ) P ∈ P (Z_P)_{P\in\mathcal{P}} ( Z P ) P ∈ P from the joint law Γ \Gamma Γ . In Construction A.4, one conceptually draws the entire universe ( Z P ) P ∈ Δ ( Y ) (Z_P)_{P\in\Delta(\mathcal{Y})} ( Z P ) P ∈ Δ ( Y ) from Γ ∞ \Gamma_\infty Γ ∞ , and trivially restricts to the coordinates ( Z P ) P ∈ P (Z_P)_{P\in\mathcal{P}} ( Z P ) P ∈ P for any subset P \mathcal{P} P of interest. The only caveat lies in how the (conceptual) couplings are constructed. In Construction A.1, we assume a dominating measure ν \nu ν such that P ≪ ν \mathcal{P}\ll\nu P ≪ ν for all P ∈ P P\in\mathcal{P} P ∈ P , and construct the coupling directly based on a single PPP A \mathcal{A} A depending only on ν \nu ν . In Construction A.4, we instead build a P \mathbf{P} P -dependent measure ν P \nu_{\mathbf{P}} ν P and a corresponding PPP A P \mathcal{A}_{\mathbf{P}} A P to define a coupling Γ k ( P ; ⋅ ) \Gamma_k(\mathbf{P};\cdot) Γ k ( P ; ⋅ ) for each finite tuple P \mathbf{P} P , and then KET guarantees that this family of finite-dimensional couplings ( Γ k ) k ≥ 1 (\Gamma_k)_{k\geq1} ( Γ k ) k ≥ 1 ultimately induces a global coupling Γ ∞ \Gamma_\infty Γ ∞ over the entire space Δ ( Y ) \Delta(\mathcal{Y}) Δ ( Y ) .
Corollary A.6. In the setup above, let ( Γ k ) k ≥ 1 (\Gamma_k)_{k\geq1} ( Γ k ) k ≥ 1 be the family of coupling kernels from Proposition A.5 . Then there exists a probability measure Γ ∞ \Gamma_\infty Γ ∞ on Y Δ ( Y ) \mathcal{Y}^{\Delta(\mathcal{Y})} Y Δ ( Y ) equipped with the product σ \sigma σ -algebra such that if ( Z P ) P ∈ Δ ( Y ) ∼ Γ ∞ (Z_P)_{P\in\Delta(\mathcal{Y})}\sim\Gamma_\infty ( Z P ) P ∈ Δ ( Y ) ∼ Γ ∞ , then for every k ≥ 1 k\geq1 k ≥ 1 and every finite collection of probability measures P = ( P 1 , … , P k ) \mathbf{P}=(P_1,\ldots,P_k) P = ( P 1 , … , P k ) , ( Z P 1 , … , Z P k ) ∼ Γ k ( P ; ⋅ ) (Z_{P_1},\ldots,Z_{P_k})\sim\Gamma_k(\mathbf{P};\cdot) ( Z P 1 , … , Z P k ) ∼ Γ k ( P ; ⋅ ) . In particular, for every P ∈ Δ ( Y ) P\in\Delta(\mathcal{Y}) P ∈ Δ ( Y ) , Z P ∼ P Z_P\sim P Z P ∼ P and for every P , Q ∈ Δ ( Y ) P,Q\in\Delta(\mathcal{Y}) P , Q ∈ Δ ( Y ) , Pr [ Z P ≠ Z Q ] ≤ 2 TV ( P , Q ) \Pr[Z_P\ne Z_Q]\leq2\operatorname{TV}(P,Q) Pr [ Z P = Z Q ] ≤ 2 TV ( P , Q ) . In other words, Γ ∞ \Gamma_\infty Γ ∞ is a grand coupling of all probability measures in Δ ( Y ) \Delta(\mathcal{Y}) Δ ( Y ) , i.e. Γ ∞ \Gamma_\infty Γ ∞ is fixed by the family ( Γ k ) k (\Gamma_k)_k ( Γ k ) k and does not depend on any particular finite tuple.
Proof. Consider index set T = Δ ( Y ) T=\Delta(\mathcal{Y}) T = Δ ( Y ) and the coordinate spaces S P = Y S_P=\mathcal{Y} S P = Y for P ∈ T P\in T P ∈ T . For every finite subset I = { P 1 , … , P k } ⊆ T I=\{P_1,\ldots,P_k\}\subseteq T I = { P 1 , … , P k } ⊆ T , we choose an arbitrary ordering ( P 1 , … , P k ) (P_1,\ldots,P_k) ( P 1 , … , P k ) . Under the natural isomorphism, ( y P ) P ∈ I → ( y P i ) i ∈ [ k ] (y_P)_{P\in I}\to(y_{P_i})_{i\in[k]} ( y P ) P ∈ I → ( y P i ) i ∈ [ k ] , we define μ I \mu_I μ I to be the probability measure on S I = ∏ P i ∈ I Y = S_I=\prod_{P_i\in I}\mathcal{Y}= S I = ∏ P i ∈ I Y = Y I \mathcal{Y}^{I} Y I , corresponding to Γ k ( ( P 1 , … , P k ) ; ⋅ ) \Gamma_k((P_1,\ldots,P_k);\cdot) Γ k (( P 1 , … , P k ) ; ⋅ ) . Note that such definition is independent of the chosen ordering, since for any other ordering of the form ( P σ ( 1 ) , … , P σ ( k ) ) (P_{\sigma(1)},\ldots,P_{\sigma(k)}) ( P σ ( 1 ) , … , P σ ( k ) ) with arbitrary permutation σ \sigma σ of [ k ] [k] [ k ] , Proposition A.5 shows that ( Z σ ( 1 ) , … , Z σ ( k ) ) ∼ Γ k ( ( P σ ( 1 ) , … , P σ ( k ) ) ; ⋅ ) (Z_{\sigma(1)},\ldots,Z_{\sigma(k)}) \sim\Gamma_k((P_{\sigma(1)},\ldots,P_{\sigma(k)});\cdot) ( Z σ ( 1 ) , … , Z σ ( k ) ) ∼ Γ k (( P σ ( 1 ) , … , P σ ( k ) ) ; ⋅ ) if ( Z 1 , … , Z k ) ∼ Γ k ( ( P 1 , … , P k ) ; ⋅ ) (Z_1,\ldots,Z_k) \sim\Gamma_k((P_1,\ldots,P_k);\cdot) ( Z 1 , … , Z k ) ∼ Γ k (( P 1 , … , P k ) ; ⋅ ) . Hence, the two orderings induce the same law on the coordinate-indexed space Y I \mathcal{Y}^{I} Y I .
Now let J ⊆ I J \subseteq I J ⊆ I . Using the same ordering, projective consistency in Proposition A.5 implies ( pr J ) # μ I = μ J (\operatorname{pr}_J)_{\#}\mu_I=\mu_J ( pr J ) # μ I = μ J , where pr J : Y I ↦ Y J \operatorname{pr}_J:\mathcal{Y}^{I}\mapsto\mathcal{Y}^{J} pr J : Y I ↦ Y J is the coordinate projection. Hence, ( μ I ) I (\mu_I)_I ( μ I ) I is a projective family, and thus, the Kolmogorov Extension Theorem [Kal02] yields random variables ( Z P ) P ∈ Δ ( Y ) (Z_P)_{P\in\Delta(\mathcal{Y})} ( Z P ) P ∈ Δ ( Y ) with Z P ∈ Y Z_P\in\mathcal{Y} Z P ∈ Y such that for every finite collection of distinct measures I = { P 1 , … , P k } I=\{P_1,\ldots,P_k\} I = { P 1 , … , P k } , ( Z P 1 , … , Z P k ) ∼ Γ k ( ( P 1 , … , P k ) ; ⋅ ) (Z_{P_1},\ldots,Z_{P_k})\sim\Gamma_k((P_1,\ldots,P_k);\cdot) ( Z P 1 , … , Z P k ) ∼ Γ k (( P 1 , … , P k ) ; ⋅ ) . Let Γ ∞ = Law ( ( Z P ) P ∈ Δ ( Y ) ) \Gamma_\infty=\operatorname{Law}((Z_P)_{P\in\Delta(\mathcal{Y})}) Γ ∞ = Law (( Z P ) P ∈ Δ ( Y ) ) , which is a probability measure on Y Δ ( Y ) \mathcal{Y}^{\Delta(\mathcal{Y})} Y Δ ( Y ) equipped with its product σ \sigma σ -algebra.
It remains to remove the distinctness assumption. Fix an arbitrary tuple P = ( P 1 , … , P k ) ∈ T k \mathbf{P}=(P_1,\ldots,P_k)\in\mathcal{T}^k P = ( P 1 , … , P k ) ∈ T k and let Q 1 , … , Q n Q_1,\ldots,Q_n Q 1 , … , Q n be the distinct measures (in the same ordering) in P \mathbf{P} P , where P i = Q π ( i ) P_i=Q_{\pi(i)} P i = Q π ( i ) for some map π : [ k ] ↦ [ n ] \pi:[k]\mapsto[n] π : [ k ] ↦ [ n ] . Now define a duplication enlargement map T π : Y n ↦ Y k T_\pi:\mathcal{Y}^n\mapsto\mathcal{Y}^k T π : Y n ↦ Y k , where T π ( y 1 , … , y n ) = ( y π ( 1 ) , … , y π ( k ) ) T_\pi(y_1,\ldots,y_n)=(y_{\pi(1)},\ldots,y_{\pi(k)}) T π ( y 1 , … , y n ) = ( y π ( 1 ) , … , y π ( k ) ) . Since Q Q Q ’s are distinct, it follows that ( Z Q 1 , … , Z Q n ) ∼ Γ n ( ( Q 1 , … , Q n ) ; ⋅ ) (Z_{Q_1},\ldots,Z_{Q_n})\sim\Gamma_n((Q_1,\ldots,Q_n);\cdot) ( Z Q 1 , … , Z Q n ) ∼ Γ n (( Q 1 , … , Q n ) ; ⋅ ) . Moreover, since ( Z P ) P ∈ Δ ( Y ) ∼ Γ ∞ (Z_P)_{P\in\Delta(\mathcal{Y})}\sim\Gamma_\infty ( Z P ) P ∈ Δ ( Y ) ∼ Γ ∞ is indexed by the probability measures themselves, P i = Q π ( i ) P_i=Q_{\pi(i)} P i = Q π ( i ) implies that Z P i = Z Q π ( i ) Z_{P_i}=Z_{Q_{\pi(i)}} Z P i = Z Q π ( i ) is the same coordinate of the grand coupling, so it follows that ( Z P 1 , … , Z P k ) = T π ( Z Q 1 , … , Z Q n ) (Z_{P_1},\ldots,Z_{P_k})=T_\pi(Z_{Q_1},\ldots,Z_{Q_n}) ( Z P 1 , … , Z P k ) = T π ( Z Q 1 , … , Z Q n ) pointwise. Therefore, Law Γ ∞ ( Z P 1 , … , Z P k ) = ( T π ) # Γ n ( ( Q 1 , … , Q n ) ; ⋅ ) \operatorname{Law}_{\Gamma_\infty}(Z_{P_1},\ldots,Z_{P_k})=(T_\pi)_{\#}\Gamma_n((Q_1,\ldots,Q_n);\cdot) Law Γ ∞ ( Z P 1 , … , Z P k ) = ( T π ) # Γ n (( Q 1 , … , Q n ) ; ⋅ ) .
Next, it suffices to show that ( T π ) # Γ n ( ( Q 1 , … , Q n ) ; ⋅ ) = Γ k ( P ; ⋅ ) (T_\pi)_{\#}\Gamma_n((Q_1,\ldots,Q_n);\cdot)=\Gamma_k(\mathbf{P};\cdot) ( T π ) # Γ n (( Q 1 , … , Q n ) ; ⋅ ) = Γ k ( P ; ⋅ ) . Now consider E = { r 1 , … , r n } E=\{r_1,\ldots,r_n\} E = { r 1 , … , r n } where r j = min { i ∈ [ k ] : P i = Q j } r_j=\min\{i\in[k]:P_i=Q_j\} r j = min { i ∈ [ k ] : P i = Q j } is the first representative coordinate for Q j Q_j Q j in P \mathbf{P} P . By projection consistency, we have ( pr E ) # Γ k ( P ; ⋅ ) = Γ n ( ( Q 1 , … , Q n ) ; ⋅ ) (\operatorname{pr}_E)_{\#}\Gamma_k(\mathbf{P};\cdot)=\Gamma_n((Q_1,\ldots,Q_n);\cdot) ( pr E ) # Γ k ( P ; ⋅ ) = Γ n (( Q 1 , … , Q n ) ; ⋅ ) since Q Q Q ’s are distinct. Also note that Γ k ( P ; D i j ) ≤ 2 TV ( P i , P j ) = 0 \Gamma_k(\mathbf{P};D_{ij})\leq2\operatorname{TV}(P_i,P_j)=0 Γ k ( P ; D ij ) ≤ 2 TV ( P i , P j ) = 0 if P i = P j P_i=P_j P i = P j , so for ( Y 1 , … , Y k ) ∼ Γ k ( P ; ⋅ ) (Y_1,\ldots,Y_k)\sim\Gamma_k(\mathbf{P};\cdot) ( Y 1 , … , Y k ) ∼ Γ k ( P ; ⋅ ) , Y i = Y r π ( i ) Y_i=Y_{r_{\pi(i)}} Y i = Y r π ( i ) a.s., and thus ( Y 1 , … , Y k ) = T π ( Y r 1 , … , Y r n ) (Y_1,\ldots,Y_k)=T_\pi(Y_{r_1},\ldots,Y_{r_n}) ( Y 1 , … , Y k ) = T π ( Y r 1 , … , Y r n ) a.s. Hence, Γ k ( P ; ⋅ ) = ( T π ) # Law ( Y r 1 , … , Y r n ) = ( T π ) # Γ n ( ( Q 1 , … , Q n ) ; ⋅ ) = Law Γ ∞ ( Z P 1 , … , Z P k ) \Gamma_k(\mathbf{P};\cdot)=(T_\pi)_{\#}\operatorname{Law}(Y_{r_1},\ldots,Y_{r_n})=(T_\pi)_{\#}\Gamma_n((Q_1,\ldots,Q_n);\cdot)=\operatorname{Law}_{\Gamma_\infty}(Z_{P_1},\ldots,Z_{P_k}) Γ k ( P ; ⋅ ) = ( T π ) # Law ( Y r 1 , … , Y r n ) = ( T π ) # Γ n (( Q 1 , … , Q n ) ; ⋅ ) = Law Γ ∞ ( Z P 1 , … , Z P k ) . Thus, we have shown that for every k ≥ 1 k\geq1 k ≥ 1 and every tuple P ∈ Δ ( Y ) k \mathbf{P}\in\Delta(\mathcal{Y})^k P ∈ Δ ( Y ) k that could contain repeated measures, ( Z P 1 , … , Z P k ) ∼ Γ k ( P ; ⋅ ) (Z_{P_1},\ldots,Z_{P_k})\sim\Gamma_k(\mathbf{P};\cdot) ( Z P 1 , … , Z P k ) ∼ Γ k ( P ; ⋅ ) .
Finally, for all P ∈ Δ ( Y ) P\in\Delta(\mathcal{Y}) P ∈ Δ ( Y ) , taking I = { P } I=\{P\} I = { P } yields Z P ∼ Γ 1 ( P ; ⋅ ) Z_P\sim\Gamma_1(P;\cdot) Z P ∼ Γ 1 ( P ; ⋅ ) , and by Proposition A.5 , Γ 1 ( P ; ⋅ ) = P \Gamma_1(P;\cdot)=P Γ 1 ( P ; ⋅ ) = P , so Z P ∼ P Z_P\sim P Z P ∼ P . For all P , Q ∈ Δ ( Y ) P,Q\in\Delta(\mathcal{Y}) P , Q ∈ Δ ( Y ) , taking I = { P , Q } I=\{P,Q\} I = { P , Q } yields ( Z P , Z Q ) ∼ Γ 2 ( ( P , Q ) ; ⋅ ) (Z_P,Z_Q)\sim\Gamma_2((P,Q);\cdot) ( Z P , Z Q ) ∼ Γ 2 (( P , Q ) ; ⋅ ) , so Pr [ Z P ≠ Z Q ] = Γ 2 ( ( P , Q ) ; D 12 ) \Pr[Z_P\ne Z_Q]=\Gamma_2((P,Q);D_{12}) Pr [ Z P = Z Q ] = Γ 2 (( P , Q ) ; D 12 ) for D 12 = { ( y 1 , y 2 ) ∈ Y 2 : y 1 ≠ y 2 } D_{12}=\{(y_1,y_2)\in\mathcal{Y}^2:y_1\ne y_2\} D 12 = {( y 1 , y 2 ) ∈ Y 2 : y 1 = y 2 } . By the disagreement bound in Proposition A.5 , it follows that Pr [ Z P ≠ Z Q ] = Γ 2 ( ( P , Q ) ; D 12 ) ≤ 2 TV ( P , Q ) \Pr[Z_P\ne Z_Q]=\Gamma_2((P,Q);D_{12})\leq2\operatorname{TV}(P,Q) Pr [ Z P = Z Q ] = Γ 2 (( P , Q ) ; D 12 ) ≤ 2 TV ( P , Q ) . □ \square □
Having established the existence of the global measure Γ ∞ \Gamma_\infty Γ ∞ , we now formally confirm that it constitutes a grand coupling scheme on Δ ( Y ) \Delta(\mathcal{Y}) Δ ( Y ) , and trivially after restriction, on any subset P \mathcal{P} P .
Corollary A.7 (Grand coupling scheme Γ P \Gamma_{\mathcal{P}} Γ P from Construction A.4). The family of random variables ( Z P ) P ∈ Δ ( Y ) ∼ Γ ∞ (Z_P)_{P\in\Delta(\mathcal{Y})}\sim\Gamma_\infty ( Z P ) P ∈ Δ ( Y ) ∼ Γ ∞ constructed in Corollary A.6 constitutes a universal grand coupling scheme on the entire space Δ ( Y ) \Delta(\mathcal{Y}) Δ ( Y ) . Consequently, its trivial restriction ( Z P ) P ∈ P (Z_P)_{P\in\mathcal{P}} ( Z P ) P ∈ P yields a valid grand coupling scheme Γ P \Gamma_{\mathcal{P}} Γ P on any subset P ⊆ Δ ( Y ) \mathcal{P}\subseteq\Delta(\mathcal{Y}) P ⊆ Δ ( Y ) .
Proof. G1 and G2 directly follows from Corollary A.6 , and G3 follows from Proposition A.5 . The restriction to any subset P \mathcal{P} P preserves all three properties trivially. Thus, the joint law of ( Z P ) P ∈ P (Z_P)_{P\in\mathcal{P}} ( Z P ) P ∈ P yields the desired grand coupling scheme Γ P \Gamma_{\mathcal{P}} Γ P . □ \square □
Private synthetic data sampler
In this section, we apply the grand coupling schemes developed in § 3 to construct a differentially private synthetic data sampler. In particular, we formally prove the privacy and utility guarantees of Aggregate (Algorithm 1) and Sampler (Algorithm 2), presented in § 4 .
Throughout, let Y = X ℓ \mathcal{Y}=\mathcal{X}^{\ell} Y = X ℓ , where X \mathcal{X} X is the data universe and ℓ \ell ℓ is the size of the synthetic dataset, and the space of probability measures of interest is P = { q ℓ : q ∈ F } \mathcal{P}=\{q^{\ell}:q\in\mathcal{F}\} P = { q ℓ : q ∈ F } , where F \mathcal{F} F contains the ground-truth distribution class Q \mathcal{Q} Q and all possible outputs of the base learner.
Aggregate: Tuple-Level Privacy and Utility
We establish the privacy and utility guarantees of Aggregate (Algorithm 1), instantiated with any grand coupling scheme on P \mathcal{P} P satisfying Definition 3.1 .
Tuple-Level Privacy. Recall that we use a change-by-one adjacency notion for neighboring tuples, i.e., q = ( q 1 , … , q k ) \mathbf{q}=(q_1,\ldots,q_k) q = ( q 1 , … , q k ) and q ′ = ( q 1 ′ , … , q k ′ ) \mathbf{q}'=(q'_1,\ldots,q'_k) q ′ = ( q 1 ′ , … , q k ′ ) are neighboring if they differ in at most one coordinate.
The privacy argument relies on using the grand coupling scheme Γ P \Gamma_{\mathcal{P}} Γ P to jointly evaluate the algorithm on both q \mathbf{q} q and q ′ \mathbf{q}' q ′ . Since the joint law of Couple k ( q ) \operatorname{Couple}_k(\mathbf{q}) Couple k ( q ) , and likewise for q ′ \mathbf{q}' q ′ , in Aggregate (Algorithm 1) is simply a finite-dimensional marginal of the global measure Γ P 10 \Gamma_{\mathcal{P}}^{10} Γ P 10 , we can evaluate both executions on a single probability space.
Specifically, rather than drawing independent samples from L ( Couple k ( q ) ) \mathcal{L}(\operatorname{Couple}_k(\mathbf{q})) L ( Couple k ( q )) and L ( Couple k ( q ′ ) ) \mathcal{L}(\operatorname{Couple}_k(\mathbf{q}')) L ( Couple k ( q ′ )) , respectively, we construct an analysis coupling from a single draw ( Z P ) P ∈ P ∼ Γ P (Z_P)_{P\in\mathcal{P}}\sim\Gamma_{\mathcal{P}} ( Z P ) P ∈ P ∼ Γ P , setting Z ~ i = Z q i ℓ \widetilde{Z}_i=Z_{q_i}^{\ell} Z i = Z q i ℓ for input q \mathbf{q} q and Z ~ i ′ = Z ( q i ′ ) ℓ \widetilde{Z}'_i=Z_{(q'_i)^{\ell}} Z i ′ = Z ( q i ′ ) ℓ for input q ′ \mathbf{q}' q ′ . For every unchanged coordinate i ≠ j i\ne j i = j , q i = q i ′ q_i=q'_i q i = q i ′ implies Z q i ℓ = Z ( q i ′ ) ℓ Z_{q_i}^{\ell}=Z_{(q'_i)^{\ell}} Z q i ℓ = Z ( q i ′ ) ℓ pointwise. Consequently, the realized input vectors ( Z ~ 1 , … , Z ~ k ) (\widetilde{Z}_1,\ldots,\widetilde{Z}_k) ( Z 1 , … , Z k ) and ( Z ~ 1 ′ , … , Z ~ k ′ ) (\widetilde{Z}'_1,\ldots,\widetilde{Z}'_k) ( Z 1 ′ , … , Z k ′ ) differ in at most a single coordinate j j j for every realization of Γ P \Gamma_{\mathcal{P}} Γ P . Therefore, since neighboring input tuples yield pointwise neighboring vectors under Γ P \Gamma_{\mathcal{P}} Γ P , the ( ε , δ ) (\varepsilon,\delta) ( ε , δ ) -DP guarantee of StabilityBasedMode transfers directly to Aggregate.
Proposition B.1 (Privacy of Aggregate). If StabilityBasedMode is ( ε , δ ) (\varepsilon,\delta) ( ε , δ ) -DP, then Aggregate in Algorithm 1 is ( ε , δ ) (\varepsilon,\delta) ( ε , δ ) -DP.10
Proof. Fix a pair of neighboring tuples q = ( q 1 , … , q k ) \mathbf{q}=(q_1,\ldots,q_k) q = ( q 1 , … , q k ) and q ′ = ( q 1 ′ , … , q k ′ ) \mathbf{q}'=(q'_1,\ldots,q'_k) q ′ = ( q 1 ′ , … , q k ′ ) , where q i = q i ′ q_i=q'_i q i = q i ′ for all i ≠ j i\ne j i = j . Let P i = q i ℓ P_i=q_i^{\ell} P i = q i ℓ and P i ′ = ( q i ′ ) ℓ P'_i=(q'_i)^{\ell} P i ′ = ( q i ′ ) ℓ . Hence, P i = P i ′ P_i=P'_i P i = P i ′ for all i ≠ j i\ne j i = j . We denote P = ( P 1 , … , P k ) \mathbf{P}=(P_1,\ldots,P_k) P = ( P 1 , … , P k ) and P ′ = ( P 1 ′ , … , P k ′ ) \mathbf{P}'=(P'_1,\ldots,P'_k) P ′ = ( P 1 ′ , … , P k ′ ) .
By Definition 3.1 , let ( Z P ) P ∈ P (Z_P)_{P\in\mathcal{P}} ( Z P ) P ∈ P be the grand coupling scheme on P \mathcal{P} P used in Couple k \operatorname{Couple}_k Couple k , and, on the common probability space where ( Z P ) P ∈ P (Z_P)_{P\in\mathcal{P}} ( Z P ) P ∈ P is defined, we construct the vectors Z ~ = ( Z P 1 , … , Z P k ) \widetilde{\mathbf{Z}}=(Z_{P_1},\ldots,Z_{P_k}) Z = ( Z P 1 , … , Z P k ) and Z ~ ′ = ( Z P 1 ′ , … , Z P k ′ ) \widetilde{\mathbf{Z}}'=(Z_{P'_1},\ldots,Z_{P'_k}) Z ′ = ( Z P 1 ′ , … , Z P k ′ ) . Note that the joint law of the output of Couple k ( q ) \operatorname{Couple}_k(\mathbf{q}) Couple k ( q ) is exactly the marginal law L ( Z P 1 , … , Z P k ) \mathcal{L}(Z_{P_1},\ldots,Z_{P_k}) L ( Z P 1 , … , Z P k ) , and similarly for q ′ \mathbf{q}' q ′ , by property G3, so it follows that Z ~ = d Couple k ( q ) \widetilde{\mathbf{Z}}\mathrel{\overset{\mathrm{d}}{=}}\operatorname{Couple}_k(\mathbf{q}) Z = d Couple k ( q ) and Z ~ ′ = d Couple k ( q ′ ) \widetilde{\mathbf{Z}}'\mathrel{\overset{\mathrm{d}}{=}}\operatorname{Couple}_k(\mathbf{q}') Z ′ = d Couple k ( q ′ ) .
Furthermore, since P i = P i ′ P_i=P'_i P i = P i ′ for all i ≠ j i\ne j i = j , we have Z ~ i = Z ~ i ′ \widetilde{Z}_i=\widetilde{Z}'_i Z i = Z i ′ pointwise for all i ≠ j i\ne j i = j . Thus, the vectors ( Z ~ 1 , … , Z ~ k ) (\widetilde{Z}_1,\ldots,\widetilde{Z}_k) ( Z 1 , … , Z k ) and ( Z ~ 1 ′ , … , Z ~ k ′ ) (\widetilde{Z}'_1,\ldots,\widetilde{Z}'_k) ( Z 1 ′ , … , Z k ′ ) are neighboring. Let M ( Z ; B ) M(\mathbf{Z};B) M ( Z ; B ) denote the probability that StabilityBasedMode outputs an element of a measurable set B B B on input Z \mathbf{Z} Z . Since M M M is ( ε , δ ) (\varepsilon,\delta) ( ε , δ ) -DP and our coupled input vectors are adjacent a.s., M ( ( Z ~ 1 , … , Z ~ k ) ; B ) ≤ e ε M ( ( Z ~ 1 ′ , … , Z ~ k ′ ) ; B ) + δ M((\widetilde{Z}_1,\ldots,\widetilde{Z}_k);B)\le e^{\varepsilon}M((\widetilde{Z}'_1,\ldots,\widetilde{Z}'_k);B)+\delta M (( Z 1 , … , Z k ) ; B ) ≤ e ε M (( Z 1 ′ , … , Z k ′ ) ; B ) + δ . Taking the expectation over the global measure Γ \Gamma Γ yields:
Pr [ Aggregate ( q ) ∈ B ] = E Γ [ M ( ( Z ~ 1 , … , Z ~ k ) ; B ) ] ≤ ε E Γ [ M ( ( Z ~ 1 ′ , … , Z ~ k ′ ) ; B ) ] + δ = ε ℓ Pr [ Aggregate ( q ′ ) ∈ B ] + δ . \Pr[\operatorname{Aggregate}(\mathbf{q}) \in B] = \mathbb{E}_{\Gamma}[M((\widetilde{Z}_1,\ldots,\widetilde{Z}_k);B)]
\leq\varepsilon\mathbb{E}_{\Gamma}[M((\widetilde{Z}'_1,\ldots,\widetilde{Z}'_k);B)] + \delta
= \varepsilon^{\ell}\Pr[\operatorname{Aggregate}(\mathbf{q}') \in B] + \delta. Pr [ Aggregate ( q ) ∈ B ] = E Γ [ M (( Z 1 , … , Z k ) ; B )] ≤ ε E Γ [ M (( Z 1 ′ , … , Z k ′ ) ; B )] + δ = ε ℓ Pr [ Aggregate ( q ′ ) ∈ B ] + δ .
A symmetric argument holds for the reverse direction.
Tuple-Level Utility. We next show that if a large majority of the input distributions q i q_i q i are close to a target ground-truth distribution p p p in TV, then the output of Aggregate \operatorname{Aggregate} Aggregate matches an ideal draw p ℓ ∼ p ℓ p^{\ell} \sim p^{\ell} p ℓ ∼ p ℓ with high probability, and is thus close to p ℓ p^{\ell} p ℓ in TV.
As in the privacy analysis, the utility guarantee relies on constructing an analysis coupling from the grand coupling scheme Γ P \Gamma_{\mathcal{P}} Γ P . Specifically, for input distributions q = ( q 1 , … , q k ) ∈ F k \mathbf{q}=(q_1,\ldots,q_k)\in\mathcal{F}^k q = ( q 1 , … , q k ) ∈ F k and a target distribution p ∈ F p\in\mathcal{F} p ∈ F , we take a single draw ( Z P ) P ∈ P ∼ Γ P (Z_P)_{P\in\mathcal{P}}\sim\Gamma_{\mathcal{P}} ( Z P ) P ∈ P ∼ Γ P and define Z ~ i = Z q i ℓ \widetilde{Z}_i=Z_{q_i}^{\ell} Z i = Z q i ℓ for each i ∈ [ k ] i\in[k] i ∈ [ k ] , alongside an unobserved “ghost” sample Z ~ ∗ = Z p ℓ \widetilde{Z}_*=Z_{p^{\ell}} Z ∗ = Z p ℓ , indexed by the ground-truth product distribution p ℓ ∈ P p^{\ell}\in\mathcal{P} p ℓ ∈ P . Since L ( Couple k ( q ) ) \mathcal{L}(\operatorname{Couple}_k(\mathbf{q})) L ( Couple k ( q )) and L ( Couple 1 ( p ) ) \mathcal{L}(\operatorname{Couple}_1(p)) L ( Couple 1 ( p )) are simply a finite-dimensional marginal of the global measure Γ P \Gamma_{\mathcal{P}} Γ P , so our construction from the grand coupling scheme Γ P \Gamma_{\mathcal{P}} Γ P satisfies ( Z ~ 1 , … , Z ~ k ) ∼ L ( Couple k ( q ) ) (\widetilde{Z}_1,\ldots,\widetilde{Z}_k)\sim\mathcal{L}(\operatorname{Couple}_k(\mathbf{q})) ( Z 1 , … , Z k ) ∼ L ( Couple k ( q )) and Z ~ ∗ ∼ L ( Couple 1 ( p ) ) = p ℓ \widetilde{Z}_*\sim\mathcal{L}(\operatorname{Couple}_1(p))=p^{\ell} Z ∗ ∼ L ( Couple 1 ( p )) = p ℓ , by property G3 and G1.
We emphasize that Z ~ ∗ \widetilde{Z}_* Z ∗ is never computed by the algorithm, as it is adjoined purely for the analysis. For every accurate distribution q i q_i q i where TV ( q i , p ) ≤ η \operatorname{TV}(q_i,p)\leq\eta TV ( q i , p ) ≤ η , property G2 guarantees that Z ~ i = Z ~ ∗ \widetilde{Z}_i=\widetilde{Z}_* Z i = Z ∗ except with probability at most 2 ℓ η 2\ell\eta 2 ℓ η . Consequently, when 7 / 8 7/8 7/8 of the q i q_i q i are η \eta η -close to p p p , an overwhelming majority of the coupled samples Z ~ i \widetilde{Z}_i Z i collapse onto the exact same value Z ~ ∗ ∼ p ℓ \widetilde{Z}_*\sim p^{\ell} Z ∗ ∼ p ℓ pointwise, creating a large frequency gap required for StabilityBasedMode \operatorname{StabilityBasedMode} StabilityBasedMode to succeed.
Proposition B.2 (Utility of Aggregate \operatorname{Aggregate} Aggregate ). Fix p ∈ Q ⊆ F p\in\mathcal{Q}\subseteq\mathcal{F} p ∈ Q ⊆ F and a tuple q = ( q 1 , … , q k ) ∈ F k \mathbf{q}=(q_1,\ldots,q_k)\in\mathcal{F}^k q = ( q 1 , … , q k ) ∈ F k . Denote J = { j ∈ [ k ] : TV ( q j , p ) ≤ η } J=\{j\in[k]:\operatorname{TV}(q_j,p)\leq\eta\} J = { j ∈ [ k ] : TV ( q j , p ) ≤ η } for some fixed η \eta η . Suppose StabilityBasedMode \operatorname{StabilityBasedMode} StabilityBasedMode outputs the mode with probability at least 1 − β S B M 1-\beta_{\mathrm{SBM}} 1 − β SBM , if the frequency gap (between the mode and the second most frequently occurring element) is at least τ \tau τ . Then, for every c ∈ ( 0 , 1 ) c\in(0,1) c ∈ ( 0 , 1 ) such that 2 ( 1 − c ) ∣ J ∣ − k ≥ τ 2(1-c)|J|-k\geq\tau 2 ( 1 − c ) ∣ J ∣ − k ≥ τ ,
TV ( L ( Aggregate ( q ) ) , p ℓ ) ≤ 2 ℓ η c + β S B M \operatorname{TV}(\mathcal{L}(\operatorname{Aggregate}(\mathbf{q})),p^{\ell})\leq\frac{2\ell\eta}{c}+\beta_{\mathrm{SBM}} TV ( L ( Aggregate ( q )) , p ℓ ) ≤ c 2 ℓ η + β SBM
Proof. Fix a tuple q = ( q 1 , … , q k ) ∈ F k \mathbf{q}=(q_1,\ldots,q_k)\in\mathcal{F}^k q = ( q 1 , … , q k ) ∈ F k and a ground truth p ∈ Q p\in\mathcal{Q} p ∈ Q . Let P i = q i ℓ P_i=q_i^{\ell} P i = q i ℓ and P ∗ = p ℓ P_*=p^{\ell} P ∗ = p ℓ . We construct an analysis coupling using a single draw ( Z P ) P ∈ P ∼ Γ P (Z_P)_{P\in\mathcal{P}}\sim\Gamma_{\mathcal{P}} ( Z P ) P ∈ P ∼ Γ P from our grand coupling scheme. We evaluate the coordinates corresponding to the input distributions, setting Z ~ i = Z P i \widetilde{Z}_i=Z_{P_i} Z i = Z P i for i ∈ [ k ] i\in[k] i ∈ [ k ] , alongside an unobserved “ghost” sample Z ~ ∗ = Z P ∗ \widetilde{Z}_*=Z_{P_*} Z ∗ = Z P ∗ . By properties (G1) and (G3), ( Z ~ 1 , … , Z ~ k ) ∼ L ( Couple k ( q ) ) (\widetilde{Z}_1,\ldots,\widetilde{Z}_k)\sim\mathcal{L}(\operatorname{Couple}_k(\mathbf{q})) ( Z 1 , … , Z k ) ∼ L ( Couple k ( q )) and Z ~ ∗ ∼ p ℓ \widetilde{Z}_*\sim p^{\ell} Z ∗ ∼ p ℓ . Similar to the proof of Proposition B.1, we construct an analysis coupling using a single draw ( Z P ) P ∈ P ∼ Γ P (Z_P)_{P\in\mathcal{P}}\sim\Gamma_{\mathcal{P}} ( Z P ) P ∈ P ∼ Γ P from our grand coupling scheme. Specifically, we consider the coordinates corresponding to the input distributions, setting Z ~ i = Z P i \widetilde{Z}_i=Z_{P_i} Z i = Z P i for i ∈ [ k ] i\in[k] i ∈ [ k ] , alongside an unobserved “ghost” sample Z ~ ∗ = Z P ∗ \widetilde{Z}_*=Z_{P_*} Z ∗ = Z P ∗ . By properties G1 and G3, ( Z ~ 1 , … , Z ~ k ) ∼ L ( Couple k ( q ) ) (\widetilde{Z}_1,\ldots,\widetilde{Z}_k)\sim\mathcal{L}(\operatorname{Couple}_k(\mathbf{q})) ( Z 1 , … , Z k ) ∼ L ( Couple k ( q )) and Z ~ ∗ ∼ Couple 1 ( p ) = p ℓ \widetilde{Z}_*\sim\operatorname{Couple}_1(p)=p^{\ell} Z ∗ ∼ Couple 1 ( p ) = p ℓ .
By the disagreement bound (G2), we have Pr [ Z ~ i ≠ Z ~ ∗ ] ≤ 2 TV ( P i , P ∗ ) = 2 TV ( q i ℓ , p ℓ ) \Pr[\widetilde{Z}_i\neq\widetilde{Z}_*]\leq2\operatorname{TV}(P_i,P_*)=2\operatorname{TV}(q_i^{\ell},p^{\ell}) Pr [ Z i = Z ∗ ] ≤ 2 TV ( P i , P ∗ ) = 2 TV ( q i ℓ , p ℓ ) for all i ∈ [ k ] i\in[k] i ∈ [ k ] . For indices in our accurate set J = { j ∈ [ k ] : TV ( q j , p ) ≤ η } J=\{j\in[k]:\operatorname{TV}(q_j,p)\leq\eta\} J = { j ∈ [ k ] : TV ( q j , p ) ≤ η } , this implies Pr [ Z ~ j ≠ Z ~ ∗ ] ≤ 2 ℓ η \Pr[\widetilde{Z}_j\neq\widetilde{Z}_*]\leq2\ell\eta Pr [ Z j = Z ∗ ] ≤ 2 ℓ η .
Consider the indicator I j = 1 { Z ~ j ≠ Z ~ ∗ } I_j=1\{\widetilde{Z}_j\neq\widetilde{Z}_*\} I j = 1 { Z j = Z ∗ } for all j ∈ J j\in J j ∈ J . We have E [ ∑ j ∈ J I j ] ≤ 2 ℓ η ∣ J ∣ \mathbb{E}[\sum_{j\in J}I_j]\leq2\ell\eta|J| E [ ∑ j ∈ J I j ] ≤ 2 ℓ η ∣ J ∣ , and by Markov’s inequality, we have Pr [ ∑ j ∈ J I j > c ∣ J ∣ ] ≤ 2 ℓ η c \Pr[\sum_{j\in J}I_j>c|J|]\leq\frac{2\ell\eta}{c} Pr [ ∑ j ∈ J I j > c ∣ J ∣ ] ≤ c 2 ℓ η . On the event ∑ j ∈ J I j ≤ c ∣ J ∣ \sum_{j\in J}I_j\leq c|J| ∑ j ∈ J I j ≤ c ∣ J ∣ , there are at least ∣ J ∣ − ∑ j ∈ J I j ≥ ( 1 − c ) ∣ J ∣ |J|-\sum_{j\in J}I_j\geq(1-c)|J| ∣ J ∣ − ∑ j ∈ J I j ≥ ( 1 − c ) ∣ J ∣ coordinates of ( Z ~ 1 , … , Z ~ k ) (\widetilde{Z}_1,\ldots,\widetilde{Z}_k) ( Z 1 , … , Z k ) equal to Z ~ ∗ \widetilde{Z}_* Z ∗ . Since 2 ( 1 − c ) ∣ J ∣ − k ≥ τ > 0 2(1-c)|J|-k\geq\tau>0 2 ( 1 − c ) ∣ J ∣ − k ≥ τ > 0 , by our choice of c c c , it follows that ( 1 − c ) ∣ J ∣ > k / 2 (1-c)|J|>k/2 ( 1 − c ) ∣ J ∣ > k /2 , ensuring Z ~ ∗ \widetilde{Z}_* Z ∗ appears in a strict majority of the k k k coordinates. Consequently, Z ~ ∗ \widetilde{Z}_* Z ∗ is the unique mode, any competing element appears at most k − ( 1 − c ) ∣ J ∣ k-(1-c)|J| k − ( 1 − c ) ∣ J ∣ times, and the frequency gap satisfies, 2 ( 1 − c ) ∣ J ∣ − k ≥ τ 2(1-c)|J|-k\geq\tau 2 ( 1 − c ) ∣ J ∣ − k ≥ τ . Let Z ~ \widetilde{Z} Z be the output obtained by applying StabilityBasedMode on the analysis-coupled vector ( Z ~ 1 , … , Z ~ k ) (\widetilde{Z}_1,\ldots,\widetilde{Z}_k) ( Z 1 , … , Z k ) , as in Algorithm 1. Since ( Z ~ 1 , … , Z ~ k ) = d Couple k ( q ) (\widetilde{Z}_1,\ldots,\widetilde{Z}_k) \overset{\mathrm{d}}{=} \operatorname{Couple}_k(q) ( Z 1 , … , Z k ) = d Couple k ( q ) , it follows that Z ~ = d Aggregate ( q ) \widetilde{Z} \overset{\mathrm{d}}{=} \operatorname{Aggregate}(q) Z = d Aggregate ( q ) . Hence, ( Z ~ , Z ~ ∗ ) (\widetilde{Z},\widetilde{Z}_*) ( Z , Z ∗ ) is indeed a coupling of L ( Aggregate ( q ) ) \mathcal{L}(\operatorname{Aggregate}(q)) L ( Aggregate ( q )) and p ℓ p^\ell p ℓ .
By the accuracy guarantee of StabilityBasedMode, we have Pr [ Z ~ ≠ Z ~ ∗ ∣ ∑ j ∈ J I j ≤ c ∣ J ∣ ] ≤ β S B M \Pr[\widetilde{Z} \ne\widetilde{Z}_* \mid\sum_{j \in J} I_j \le c|J|] \le\beta_{\mathrm{SBM}} Pr [ Z = Z ∗ ∣ ∑ j ∈ J I j ≤ c ∣ J ∣ ] ≤ β SBM . Thus, Pr [ Z ~ ≠ Z ~ ∗ ] ≤ β S B M + 2 ℓ η c \Pr[\widetilde{Z} \ne\widetilde{Z}_*] \le\beta_{\mathrm{SBM}} + \frac{2\ell\eta}{c} Pr [ Z = Z ∗ ] ≤ β SBM + c 2 ℓ η . Additionally, TV ( L ( Aggregate ( q ) ) , p ℓ ) = TV ( L ( Z ~ ) , L ( Z ~ ∗ ) ) ≤ Pr [ Z ~ ≠ Z ~ ∗ ] ≤ β S B M + 2 ℓ η c \operatorname{TV}(\mathcal{L}(\operatorname{Aggregate}(q)),p^\ell) = \operatorname{TV}(\mathcal{L}(\widetilde{Z}),\mathcal{L}(\widetilde{Z}_*)) \le\Pr[\widetilde{Z} \ne\widetilde{Z}_*] \le\beta_{\mathrm{SBM}} + \frac{2\ell\eta}{c} TV ( L ( Aggregate ( q )) , p ℓ ) = TV ( L ( Z ) , L ( Z ∗ )) ≤ Pr [ Z = Z ∗ ] ≤ β SBM + c 2 ℓ η .
To complete the utility guarantee, we instantiate the mode mechanism step with the stability-based mode mechanism [46 Vad17] , whose privacy and accuracy properties are recalled below.
Fact B.3 (Stability-based mode, Proposition 3.4 from [Vad17]; [46 ] ). For every ε , δ ∈ ( 0 , 1 ) \varepsilon,\delta\in(0,1) ε , δ ∈ ( 0 , 1 ) and m ∈ N m \in\mathbb{N} m ∈ N , there exists an ( ε , δ ) (\varepsilon,\delta) ( ε , δ ) -DP mechanism M : X m → X M:\mathcal{X}^m \to\mathcal{X} M : X m → X such that if the difference in counts between the mode of x m x^m x m and the second most frequently occurring element is larger than 4 log ( 1 / δ ) / ε 4\log(1/\delta)/\varepsilon 4 log ( 1/ δ ) / ε , M ( x ) M(x) M ( x ) outputs the mode with probability at least 1 − δ 1-\delta 1 − δ .
We can now synthesize these components. By combining our tuple-level privacy guarantee (in Proposition B.1 ), the utility bound of our analysis coupling (Proposition B.2 ), and the accuracy of the mode mechanism (Fact B.3 ), we obtain the final privacy and utility guarantee for the Aggregate subroutine, stated below.
Corollary B.4. Let β , ε , δ ∈ ( 0 , 1 ) \beta,\varepsilon,\delta\in(0,1) β , ε , δ ∈ ( 0 , 1 ) , ℓ ∈ N + \ell\in\mathbb{N}^{+} ℓ ∈ N + be parameters. Set η = β 64 ℓ \eta= \frac{\beta}{64\ell} η = 64 ℓ β , δ 0 = min { δ , β / 4 } \delta_0 = \min\{\delta,\beta/4\} δ 0 = min { δ , β /4 } , and choose k k k such that k ≥ 8 log ( 4 / β ) + log ( 1 / δ ) ε k \ge8\frac{\log(4/\beta)+\log(1/\delta)}{\varepsilon} k ≥ 8 ε l o g ( 4/ β ) + l o g ( 1/ δ ) . Then Aggregate, with StabilityBasedMode instantiated using parameters ( ε , δ 0 ) (\varepsilon,\delta_0) ( ε , δ 0 ) as in Fact B.3 , is ( ε , δ 0 ) (\varepsilon,\delta_0) ( ε , δ 0 ) -DP. Moreover, for any p ∈ Q p \in\mathcal{Q} p ∈ Q and input tuple q ∈ F k q \in\mathcal{F}^k q ∈ F k , if ∣ { j ∈ [ k ] : TV ( q j , p ) ≤ η } ∣ ≥ 7 k / 8 |\{j \in[k] : \operatorname{TV}(q_j,p) \le\eta\}| \ge7k/8 ∣ { j ∈ [ k ] : TV ( q j , p ) ≤ η } ∣ ≥ 7 k /8 , then TV ( L ( Aggregate ( q ) ) , p ℓ ) ≤ β / 2 \operatorname{TV}(\mathcal{L}(\operatorname{Aggregate}(q)),p^\ell) \le\beta/2 TV ( L ( Aggregate ( q )) , p ℓ ) ≤ β /2 .
Proof. By Proposition B.1 , Aggregate is ( ε , δ 0 ) (\varepsilon,\delta_0) ( ε , δ 0 ) -DP, since StabilityBasedMode is ( ε , δ 0 ) (\varepsilon,\delta_0) ( ε , δ 0 ) -DP. For utility, take τ = 4 log ( 1 / δ 0 ) / ε \tau= 4\log(1/\delta_0)/\varepsilon τ = 4 log ( 1/ δ 0 ) / ε and β S B M = δ 0 \beta_{\mathrm{SBM}} = \delta_0 β SBM = δ 0 from Fact B.3 , and choose c = 1 / 8 c = 1/8 c = 1/8 . It is not hard to verify that 2 ( 1 − c ) ∣ J ∣ − k ≥ k / 2 ≥ τ 2(1-c)|J|-k \ge k/2 \ge\tau 2 ( 1 − c ) ∣ J ∣ − k ≥ k /2 ≥ τ . Proposition B.2 therefore gives a coupling ( Z ~ , Z ~ ∗ ) (\widetilde{Z},\widetilde{Z}_*) ( Z , Z ∗ ) such that Z ~ = d Aggregate ( q ) \widetilde{Z} \overset{\mathrm{d}}{=} \operatorname{Aggregate}(q) Z = d Aggregate ( q ) , Z ~ ∗ = d p ℓ \widetilde{Z}_* \overset{\mathrm{d}}{=} p^\ell Z ∗ = d p ℓ , and Pr [ Z ~ ≠ Z ~ ∗ ] ≤ β S B M + 2 ℓ η c = δ 0 + β / 4 ≤ β / 2 \Pr[\widetilde{Z} \ne\widetilde{Z}_*] \le\beta_{\mathrm{SBM}} + \frac{2\ell\eta}{c} = \delta_0 + \beta/4 \le\beta/2 Pr [ Z = Z ∗ ] ≤ β SBM + c 2 ℓ η = δ 0 + β /4 ≤ β /2 . Consequently, TV ( L ( Aggregate ( q ) ) , p ℓ ) ≤ β / 2 \operatorname{TV}(\mathcal{L}(\operatorname{Aggregate}(q)),p^\ell) \le\beta/2 TV ( L ( Aggregate ( q )) , p ℓ ) ≤ β /2 .
We establish the privacy and utility guarantees of Sampler (Algorithm 2), which takes a dataset S ∈ X n S \in\mathcal{X}^n S ∈ X n as input. Here, two datasets S , S ′ ∈ X n S,S' \in\mathcal{X}^n S , S ′ ∈ X n are adjacent if they differ in at most one record.
Proposition B.5 (Privacy of Sampler). If StabilityBasedMode in Aggregate is ( ε , δ ) (\varepsilon,\delta) ( ε , δ ) -DP, then Sampler is ( ε , δ ) (\varepsilon,\delta) ( ε , δ ) -DP. 12
Proof. Fix neighboring datasets S = ( X 1 , … , X n ) S=(X_1,\ldots,X_n) S = ( X 1 , … , X n ) and S ′ = ( X 1 ′ , … , X n ′ ) S'=(X'_1,\ldots,X'_n) S ′ = ( X 1 ′ , … , X n ′ ) such that X i = X i ′ X_i=X'_i X i = X i ′ for all i ≠ j i\ne j i = j .
We first couple the data-independent randomness used in the two executions on S S S and S ′ S' S ′ . Specifically, we draw a single permutation Π ∼ U ( S n ) \Pi\sim\mathcal{U}(S_n) Π ∼ U ( S n ) and a single collection of learner seeds U 1 , … , U k ∼ i . i . d . U ( [ 0 , 1 ] ) U_1,\ldots,U_k \overset{\mathrm{i.i.d.}}{\sim} \mathcal{U}([0,1]) U 1 , … , U k ∼ i.i.d. U ([ 0 , 1 ]) independently of data, and use the same realizations of them in both executions.
Now fix a realization ( π , u 1 , … , u k ) (\pi,u_1,\ldots,u_k) ( π , u 1 , … , u k ) . Denote the partitions as ( s i ) i (s_i)_i ( s i ) i and ( s i ′ ) i (s'_i)_i ( s i ′ ) i on the two executions. Let a a a be the block id containing the position π − 1 ( j ) \pi^{-1}(j) π − 1 ( j ) , i.e. the position occupied by the differing data coordinate under permutation. Since X i = X i ′ X_i = X_i' X i = X i ′ for all i ≠ j i \ne j i = j , it follows that s i = s i ′ s_i = s_i' s i = s i ′ for all i ≠ a i \ne a i = a . Therefore, with the same learner seed u i u_i u i used on s i s_i s i and s i ′ s_i' s i ′ , it follows that q i = L ( s i ; u i ) = L ( s i ′ ; u i ) = q i ′ q_i = L(s_i;u_i) = L(s_i';u_i) = q_i' q i = L ( s i ; u i ) = L ( s i ′ ; u i ) = q i ′ for all i ≠ a i \ne a i = a . Hence, for every fixed realization of the data-independent randomness, the induced tuples q = ( q 1 , … , q k ) \mathbf{q} = (q_1,\ldots,q_k) q = ( q 1 , … , q k ) and q ′ = ( q 1 ′ , … , q k ′ ) \mathbf{q}' = (q_1',\ldots,q_k') q ′ = ( q 1 ′ , … , q k ′ ) differ in at most one coordinate.
Let M ( q ; B ) M(\mathbf{q};B) M ( q ; B ) denote the probability that Aggregate ( q ) \operatorname{Aggregate}(\mathbf{q}) Aggregate ( q ) lies in a measurable set B ⊆ X ℓ B \subseteq\mathcal{X}^{\ell} B ⊆ X ℓ . Since StabilityBasedMode is ( ε , δ ) (\varepsilon,\delta) ( ε , δ ) -DP, by Proposition B.1 , Aggregate is ( ε , δ ) (\varepsilon,\delta) ( ε , δ ) -DP under change-by-one adjacency on input tuple. Hence, for every realization ( π , u 1 , … , u k ) (\pi,u_1,\ldots,u_k) ( π , u 1 , … , u k ) , M ( q ; B ) ≤ e ε M ( q ′ ; B ) + δ M(\mathbf{q};B) \le e^{\varepsilon}M(\mathbf{q}';B) + \delta M ( q ; B ) ≤ e ε M ( q ′ ; B ) + δ . Averaging over the shared data-independent randomness yields
Pr [ Sampler ( S ) ∈ B ] = E Π , ( U i ) i ∈ [ k ] [ M ( q ; B ) ] ≤ e ε E Π , ( U i ) i ∈ [ k ] [ M ( q ′ ; B ) ] + δ = e ε Pr [ Sampler ( S ′ ) ∈ B ] + δ . \Pr[\operatorname{Sampler}(S) \in B] = \mathbb{E}_{\Pi,(U_i)_{i\in[k]}}[M(\mathbf{q};B)]
\le e^{\varepsilon}\mathbb{E}_{\Pi,(U_i)_{i\in[k]}}[M(\mathbf{q}';B)] + \delta
= e^{\varepsilon}\Pr[\operatorname{Sampler}(S') \in B] + \delta. Pr [ Sampler ( S ) ∈ B ] = E Π , ( U i ) i ∈ [ k ] [ M ( q ; B )] ≤ e ε E Π , ( U i ) i ∈ [ k ] [ M ( q ′ ; B )] + δ = e ε Pr [ Sampler ( S ′ ) ∈ B ] + δ .
and similarly for the reverse direction. Therefore, Sampler is ( ε , δ ) (\varepsilon,\delta) ( ε , δ ) -DP. □
Proposition B.6 (Utility of Sampler). Fix a base learner L L L for Q Q Q . Let ε > 0 \varepsilon> 0 ε > 0 , β , δ ∈ ( 0 , 1 ) \beta,\delta\in(0,1) β , δ ∈ ( 0 , 1 ) , ℓ ∈ N + \ell\in\mathbb{N}^{+} ℓ ∈ N + be parameters. Set η = β 64 ℓ \eta= \frac{\beta}{64\ell} η = 64 ℓ β , δ 0 = min { δ , β / 4 } \delta_0 = \min\{\delta,\beta/4\} δ 0 = min { δ , β /4 } , and choose k k k such that k ≥ max { 8 β log ( 4 β δ ) , 128 log ( 4 β ) } k \ge\max\left\{\frac{8}{\beta}\log\left(\frac{4}{\beta\delta}\right),128\log\left(\frac{4}{\beta}\right)\right\} k ≥ max { β 8 log ( β δ 4 ) , 128 log ( β 4 ) } . Let b ∈ N + b \in\mathbb{N}^{+} b ∈ N + be a sample size such that, for every p ∈ Q p \in Q p ∈ Q , if S i ∼ p b S_i \sim p^b S i ∼ p b and U i ∼ U ( [ 0 , 1 ] ) U_i \sim\mathrm{U}([0,1]) U i ∼ U ([ 0 , 1 ]) independently, Pr [ TV ( L ( S i ; U i ) , p ) ≥ η ] ≤ 1 / 16 \Pr[\operatorname{TV}(L(S_i;U_i),p) \ge\eta] \le1/16 Pr [ TV ( L ( S i ; U i ) , p ) ≥ η ] ≤ 1/16 . Set n = k b n = kb n = k b . Then, for every p ∈ Q ⊆ F p \in Q \subseteq\mathcal{F} p ∈ Q ⊆ F and S ∼ p n S \sim p^n S ∼ p n , TV ( L ( Sampler ( S ) ) , p ℓ ) ≤ 3 β / 4 \operatorname{TV}(\mathcal{L}(\operatorname{Sampler}(S)),p^\ell) \le3\beta/4 TV ( L ( Sampler ( S )) , p ℓ ) ≤ 3 β /4 . Thus, Sampler is an ( ℓ , 3 β / 4 ) (\ell,3\beta/4) ( ℓ , 3 β /4 ) -approximate sampler for Q Q Q in the sense of Definition 1.4 .
Proof. Fix p ∈ Q p \in Q p ∈ Q . We first show that, with high probability, at least 7 k / 8 7k/8 7 k /8 of the learner outputs are η \eta η -close to p p p . Note that ( X π ( 1 ) , … , X π ( n ) ) ∼ p n (X_{\pi(1)},\ldots,X_{\pi(n)}) \sim p^n ( X π ( 1 ) , … , X π ( n ) ) ∼ p n , for any fixed permutation π \pi π , since S ∼ p n S \sim p^n S ∼ p n . Hence, conditional on Π = π \Pi= \pi Π = π , the blocks S 1 , … , S k S_1,\ldots,S_k S 1 , … , S k , where S i = ( X π ( ( i − 1 ) b + 1 ) , … , X π ( i b ) ) S_i = (X_{\pi((i-1)b+1)},\ldots,X_{\pi(ib)}) S i = ( X π (( i − 1 ) b + 1 ) , … , X π ( ib ) ) , are independent and S i ∼ p b S_i \sim p^b S i ∼ p b for every i i i . Moreover, the learner seeds U 1 , … , U k U_1,\ldots,U_k U 1 , … , U k remain independent conditional on Π = π \Pi= \pi Π = π , and are independent of the data S S S . Therefore, conditional on Π = π \Pi= \pi Π = π , ( S i , U i ) , … , ( S k , U k ) (S_i,U_i),\ldots,(S_k,U_k) ( S i , U i ) , … , ( S k , U k ) are independent, and thus, the learner outputs q i = L ( S i ; U i ) q_i = L(S_i;U_i) q i = L ( S i ; U i ) are independent.
Now define indicators I i = 1 { TV ( q i , p ) > η } I_i = 1\{\operatorname{TV}(q_i,p) > \eta\} I i = 1 { TV ( q i , p ) > η } for each i ∈ [ k ] i \in[k] i ∈ [ k ] . By the accuracy guarantee of the learner, Pr [ I i = 1 ∣ Π = π ] ≤ 1 / 16 \Pr[I_i = 1 \mid\Pi= \pi] \le1/16 Pr [ I i = 1 ∣ Π = π ] ≤ 1/16 for every i i i , and thus, I 1 , … , I k I_1,\ldots,I_k I 1 , … , I k are independent Bernoulli with means at most 1 / 16 1/16 1/16 , conditional on Π = π \Pi= \pi Π = π . By Hoeffding’s inequality, Pr [ ∑ i = 1 k I i ≥ k / 8 ∣ Π = π ] ≤ exp ( − 2 k ( 1 / 8 − 1 / 16 ) 2 ) = e − k / 128 ≤ β / 4 \Pr[\sum_{i=1}^{k} I_i \ge k/8 \mid\Pi= \pi] \le\exp(-2k(1/8-1/16)^2) = e^{-k/128} \le\beta/4 Pr [ ∑ i = 1 k I i ≥ k /8 ∣ Π = π ] ≤ exp ( − 2 k ( 1/8 − 1/16 ) 2 ) = e − k /128 ≤ β /4 . Since this holds uniformly for every realization π \pi π , so taking an expectation over Π \Pi Π yields Pr [ ∑ i = 1 k I i ≥ k / 8 ] ≤ β / 4 \Pr[\sum_{i=1}^{k} I_i \ge k/8] \le\beta/4 Pr [ ∑ i = 1 k I i ≥ k /8 ] ≤ β /4 , i.e. Pr [ E c ] ≤ β / 4 \Pr[E^c] \le\beta/4 Pr [ E c ] ≤ β /4 , where E = { ∣ { i ∈ [ k ] : TV ( q i , p ) ≤ η } ∣ ≥ 7 k / 8 } E = \{|\{i \in[k] : \operatorname{TV}(q_i,p) \le\eta\}| \ge7k/8\} E = { ∣ { i ∈ [ k ] : TV ( q i , p ) ≤ η } ∣ ≥ 7 k /8 } .
By Corollary B.4 , every realized q \mathbf{q} q that satisfies E E E , satisfies TV ( L ( Aggregate ( q ) ) , p ℓ ) ≤ β / 2 \operatorname{TV}(\mathcal{L}(\operatorname{Aggregate}(\mathbf{q})),p^\ell) \le\beta/2 TV ( L ( Aggregate ( q )) , p ℓ ) ≤ β /2 . Note that L ( Sampler ( S ) ∣ E ) = E [ L ( Aggregate ( q ) ) ∣ E ] \mathcal{L}(\operatorname{Sampler}(S)\mid E) = \mathbb{E}[\mathcal{L}(\operatorname{Aggregate}(\mathbf{q}))\mid E] L ( Sampler ( S ) ∣ E ) = E [ L ( Aggregate ( q )) ∣ E ] , so TV ( L ( Sampler ( S ) ∣ E ) , p ℓ ) ≤ β / 2 \operatorname{TV}(\mathcal{L}(\operatorname{Sampler}(S)\mid E),p^\ell) \le\beta/2 TV ( L ( Sampler ( S ) ∣ E ) , p ℓ ) ≤ β /2 . Thus,
TV ( L ( Sampler ( S ) ) , p ℓ ) ≤ Pr [ E ] TV ( L ( Sampler ( S ) ∣ E ) , p ℓ ) + Pr [ E c ] TV ( L ( Sampler ( S ) ∣ E c ) , p ℓ ) ≤ β / 2 + Pr [ E c ] ≤ β / 2 + β / 4 = 3 β / 4. \operatorname{TV}(\mathcal{L}(\operatorname{Sampler}(S)),p^\ell)
\le\Pr[E]\operatorname{TV}(\mathcal{L}(\operatorname{Sampler}(S)\mid E),p^\ell)
+ \Pr[E^c]\operatorname{TV}(\mathcal{L}(\operatorname{Sampler}(S)\mid E^c),p^\ell)
\le\beta/2 + \Pr[E^c] \le\beta/2 + \beta/4 = 3\beta/4. TV ( L ( Sampler ( S )) , p ℓ ) ≤ Pr [ E ] TV ( L ( Sampler ( S ) ∣ E ) , p ℓ ) + Pr [ E c ] TV ( L ( Sampler ( S ) ∣ E c ) , p ℓ ) ≤ β /2 + Pr [ E c ] ≤ β /2 + β /4 = 3 β /4.
Finally, by the coupling characterization of total variation distance, there exists a coupling ( Z ~ , Z ~ ∗ ) (\widetilde{Z},\widetilde{Z}_*) ( Z , Z ∗ ) such that Z ~ = d Sampler ( S ′ ) \widetilde{Z} \overset{\mathrm{d}}{=} \operatorname{Sampler}(S') Z = d Sampler ( S ′ ) , Z ~ ∗ = d p ℓ \widetilde{Z}_* \overset{\mathrm{d}}{=} p^\ell Z ∗ = d p ℓ , and Pr [ Z ~ ≠ Z ~ ∗ ] ≤ 3 β / 4 \Pr[\widetilde{Z} \ne\widetilde{Z}_*] \le3\beta/4 Pr [ Z = Z ∗ ] ≤ 3 β /4 . Thus, Sampler is an ( ℓ , 3 β / 4 ) (\ell,3\beta/4) ( ℓ , 3 β /4 ) -correlated sampler for Q Q Q . □
Combining the dataset-level privacy guarantee of Proposition B.5 with the utility guarantee of Proposition B.6 yields the formal guarantee for the end-to-end pipeline Sampler.
Corollary B.7. Under the same setting as B.6, Sampler is ( ε , δ 0 ) (\varepsilon,\delta_0) ( ε , δ 0 ) -DP ( ℓ , 3 β / 4 ) (\ell,3\beta/4) ( ℓ , 3 β /4 ) -approximate sampler for Q Q Q in the sense of Definition 1.4 .
Agnostic private distribution learning
In this section, we extend our main result beyond the realizable setting. We show that if a class of distributions is agnostic learnable, then it is also agnostic private learnable. The following is the definition of agnostic learnability.
Definition C.1 (C C C -agnostic learnability). Let C ≥ 1 C \ge1 C ≥ 1 and let Q ⊆ Δ ( X ) Q \subseteq\Delta(\mathcal{X}) Q ⊆ Δ ( X ) be nonempty. We say that Q Q Q is C C C -agnostically learnable with sample complexity m : ( 0 , 1 ) 2 → N ≥ 1 m:(0,1)^2 \to\mathbb{N}_{\ge1} m : ( 0 , 1 ) 2 → N ≥ 1 if there exists a randomized learner L : X ∗ → Δ ( X ) L:\mathcal{X}^{*} \to\Delta(\mathcal{X}) L : X ∗ → Δ ( X ) such that, for every α , β ∈ ( 0 , 1 ) \alpha,\beta\in(0,1) α , β ∈ ( 0 , 1 ) and p ∈ Δ ( X ) p \in\Delta(\mathcal{X}) p ∈ Δ ( X ) , given a sample set S ∼ p m ( α , β ) S \sim p^{m(\alpha,\beta)} S ∼ p m ( α , β ) , the learner’s output p ^ = L ( S ) \hat{p}=L(S) p ^ = L ( S ) satisfies TV ( p , p ^ ) ≤ C OPT ( p ) + α \operatorname{TV}(p,\hat{p}) \le C\operatorname{OPT}(p)+\alpha TV ( p , p ^ ) ≤ C OPT ( p ) + α with probability at least ≥ 1 − β \ge1-\beta ≥ 1 − β , where the probability is over the samples and the learner’s internal randomness, and OPT ( p ) : = inf q ∈ Q TV ( p , q ) \operatorname{OPT}(p):=\inf_{q\in Q}\operatorname{TV}(p,q) OPT ( p ) := inf q ∈ Q TV ( p , q ) .
Theorem C.2 (General agnostic private reduction). Let X \mathcal{X} X be standard Borel and C ≥ 1 C \ge1 C ≥ 1 . If Q ⊆ Δ ( X ) Q \subseteq\Delta(\mathcal{X}) Q ⊆ Δ ( X ) is C C C -agnostically learnable with sample complexity m ( α , β ) m(\alpha,\beta) m ( α , β ) , then, for every ε , δ ∈ ( 0 , 1 ) \varepsilon,\delta\in(0,1) ε , δ ∈ ( 0 , 1 ) , it is also ( 8 C 2 + 9 C ) (8C^2+9C) ( 8 C 2 + 9 C ) -agnostically learnable under ( ε , δ ) (\varepsilon,\delta) ( ε , δ ) -differential privacy, with sample complexity
n 0 ( α , β , ε , δ ) = O ( m ( α 16 ( C + 3 ) , 1 16 ) m ( 2 α C + 3 , β ) 1 ℓ log ( 4 ( C + 3 ) α δ ) ) n_0(\alpha,\beta,\varepsilon,\delta)=O\left(m\left(\frac{\alpha}{16(C+3)},\frac{1}{16}\right)m\left(\frac{2\alpha}{C+3},\beta\right)^{\frac{1}{\ell}}\log\left(\frac{4(C+3)}{\alpha\delta}\right)\right) n 0 ( α , β , ε , δ ) = O ( m ( 16 ( C + 3 ) α , 16 1 ) m ( C + 3 2 α , β ) ℓ 1 log ( α δ 4 ( C + 3 ) ) ) for every α , β ∈ ( 0 , 1 ) \alpha,\beta\in(0,1) α , β ∈ ( 0 , 1 ) . The reduction preserves properness.
Proof. Let L L L be a C C C -agnostic learner witnessing Definition C.1 with sample complexity m m m . Set
a = α C + 3 , b = m ( a 16 , 1 16 ) , T = m ( 2 a , β ) , δ 0 = min { δ , a 4 } , and k = ⌈ max { 16 ε log ( 1 δ 0 ) , 128 log ( 4 a ) } ⌉ . a=\frac{\alpha}{C+3},\quad b=m\left(\frac{a}{16},\frac{1}{16}\right),\quad T=m(2a,\beta),\quad\delta_0=\min\left\{\delta,\frac{a}{4}\right\},\quad\text{and}\quad k=\left\lceil\max\left\{\frac{16}{\varepsilon}\log\left(\frac{1}{\delta_0}\right),128\log\left(\frac{4}{a}\right)\right\}\right\rceil. a = C + 3 α , b = m ( 16 a , 16 1 ) , T = m ( 2 a , β ) , δ 0 = min { δ , 4 a } , and k = ⌈ max { ε 16 log ( δ 0 1 ) , 128 log ( a 4 ) } ⌉ .
Let n 0 = T k b n_0=Tkb n 0 = T k b .
We use Algorithm 7 with the parameters chosen above. Each call to Sampler invokes Algorithm 2 on a batch S j ∈ X k b S_j \in\mathcal{X}^{kb} S j ∈ X k b , using k k k blocks of size b b b and base learner L b a s e L_{\mathrm{base}} L base , and returns one private sample Z j ∈ X Z_j \in\mathcal{X} Z j ∈ X . We instantiate Sampler with ambient family F = Δ ( X ) \mathcal{F}=\Delta(\mathcal{X}) F = Δ ( X ) , the coupling implementation from Algorithm 6, and privacy parameters ( ε , δ 0 ) (\varepsilon,\delta_0) ( ε , δ 0 ) . All invocations use fresh randomness, independent of the outer permutation and of one another. The final invocation of L L L receives ( Z 1 , … , Z T ) (Z_1,\ldots,Z_T) ( Z 1 , … , Z T ) and returns q ^ \hat{q} q ^ using independent randomness.
For inputs larger than n 0 n_0 n 0 , we use only the first n 0 n_0 n 0 observations. For inputs smaller than n 0 n_0 n 0 , we return a fixed element of Q Q Q .
Privacy. By Proposition B.5 , each invocation of Sampler is ( ε , δ 0 ) (\varepsilon,\delta_0) ( ε , δ 0 ) -DP. Conditioned on the data-independent outer permutation, the invocations use disjoint sets of records. Parallel composition (Lemma 2.5 ) therefore implies that ( Z 1 , … , Z T ) (Z_1,\ldots,Z_T) ( Z 1 , … , Z T ) is ( ε , δ 0 ) (\varepsilon,\delta_0) ( ε , δ 0 ) -DP. Averaging over the permutation preserves this guarantee. The final estimate is post-processing, so by Proposition 2.4 the learner is ( ε , δ ) (\varepsilon,\delta) ( ε , δ ) -DP because δ 0 ≤ δ \delta_0 \le\delta δ 0 ≤ δ . The fixed output on insufficient data is also private. If the assumed learnability is proper, choose L L L proper; the final estimate and the fixed output then both belong to Q Q Q .
Utility. Fix p ∈ Δ ( X ) p \in\Delta(\mathcal{X}) p ∈ Δ ( X ) , write OPT : = inf q ∈ Q TV ( p , q ) \operatorname{OPT}:=\inf_{q\in Q}\operatorname{TV}(p,q) OPT := inf q ∈ Q TV ( p , q ) , and consider one invocation of Sampler on S 0 ∼ p k b S_0 \sim p^{kb} S 0 ∼ p k b . Let q 1 , … , q k q_1,\ldots,q_k q 1 , … , q k be its base learner outputs and set η = C OPT + a 16 \eta=C\operatorname{OPT}+\frac{a}{16} η = C OPT + 16 a . By the agnostic guarantee of L L L and the choice of b b b , Pr [ TV ( q i , p ) ≤ η ] ≥ 15 16 \Pr[\operatorname{TV}(q_i,p)\le\eta]\ge\frac{15}{16} Pr [ TV ( q i , p ) ≤ η ] ≥ 16 15 for every i ∈ [ k ] i\in[k] i ∈ [ k ] .
Let J = { i ∈ [ k ] : TV ( q i , p ) ≤ η } J=\{i\in[k]:\operatorname{TV}(q_i,p)\le\eta\} J = { i ∈ [ k ] : TV ( q i , p ) ≤ η } and E = { ∣ J ∣ ≥ 7 k 8 } E=\{|J|\ge\frac{7k}{8}\} E = { ∣ J ∣ ≥ 8 7 k } . Independence of the blocks and learner randomness implies that the events { TV ( q i , p ) > η } \{\operatorname{TV}(q_i,p)>\eta\} { TV ( q i , p ) > η } are independent, each with probability at most 1 16 \frac{1}{16} 16 1 . Thus, Hoeffding’s inequality gives Pr [ E c ] ≤ exp ( − 2 k ( 1 8 − 1 16 ) 2 ) = e − k / 128 ≤ a 4 \Pr[E^c]\le\exp\left(-2k\left(\frac{1}{8}-\frac{1}{16}\right)^2\right)=e^{-k/128}\le\frac{a}{4} Pr [ E c ] ≤ exp ( − 2 k ( 8 1 − 16 1 ) 2 ) = e − k /128 ≤ 4 a .
By Fact B.3 , the mode mechanism instantiated with ( ε , δ 0 ) (\varepsilon,\delta_0) ( ε , δ 0 ) has τ = 4 ε log ( 1 δ 0 ) \tau=\frac{4}{\varepsilon}\log\left(\frac{1}{\delta_0}\right) τ = ε 4 log ( δ 0 1 ) and β S B M = δ 0 \beta_{\mathrm{SBM}}=\delta_0 β SBM = δ 0 .
On E E E , taking c = 1 4 c=\frac{1}{4} c = 4 1 , our choice of k k k ensures 2 ( 1 − c ) ∣ J ∣ − k ≥ 5 k 16 ≥ k 4 ≥ τ 2(1-c)|J|-k\ge\frac{5k}{16}\ge\frac{k}{4}\ge\tau 2 ( 1 − c ) ∣ J ∣ − k ≥ 16 5 k ≥ 4 k ≥ τ .
For each tuple q \mathbf{q} q , let μ q : = L ( Aggregate ( q ) ) \mu_{\mathbf{q}}:=\mathcal{L}(\operatorname{Aggregate}(\mathbf{q})) μ q := L ( Aggregate ( q )) . By Proposition B.2 with ℓ = 1 \ell=1 ℓ = 1 , on E E E we have TV ( μ q , p ) ≤ 8 η + δ 0 \operatorname{TV}(\mu_{\mathbf{q}},p)\le8\eta+\delta_0 TV ( μ q , p ) ≤ 8 η + δ 0 . This holds with probability at least 1 − e − k / 128 ≥ 1 − a 4 1-e^{-k/128}\ge1-\frac{a}{4} 1 − e − k /128 ≥ 1 − 4 a over the block data and base learner randomness.
Let R p : = L ( Sampler ( S 0 ) ) R_p:=\mathcal{L}(\operatorname{Sampler}(S_0)) R p := L ( Sampler ( S 0 )) , where the law includes both the randomness of S 0 ∼ p k b S_0\sim p^{kb} S 0 ∼ p k b and the sampler’s internal randomness. By Algorithm 2 and the law of total probability, for every measurable B ⊆ X B\subseteq\mathcal{X} B ⊆ X , R p ( B ) = E q [ μ q ( B ) ] R_p(B)=\mathbb{E}_{\mathbf{q}}[\mu_{\mathbf{q}}(B)] R p ( B ) = E q [ μ q ( B )] . Consequently,
TV ( R p , p ) = sup B ∣ E q [ μ q ( B ) − p ( B ) ] ∣ ≤ E q [ sup B ∣ μ q ( B ) − p ( B ) ∣ ] = E q [ TV ( μ q , p ) ] ≤ ( 8 η + δ 0 ) Pr [ E ] + Pr [ E c ] ≤ 8 η + δ 0 + e − k / 128 ≤ 8 C OPT + a 2 + a 4 + a 4 = 8 C OPT + a . \begin{aligned}
\operatorname{TV}(R_p,p)&=\sup_B\left|\mathbb{E}_{\mathbf{q}}[\mu_{\mathbf{q}}(B)-p(B)]\right|\\
&\le\mathbb{E}_{\mathbf{q}}\left[\sup_B|\mu_{\mathbf{q}}(B)-p(B)|\right]\\
&=\mathbb{E}_{\mathbf{q}}[\operatorname{TV}(\mu_{\mathbf{q}},p)]\\
&\le(8\eta+\delta_0)\Pr[E]+\Pr[E^c]\\
&\le8\eta+\delta_0+e^{-k/128}\\
&\le8C\operatorname{OPT}+\frac{a}{2}+\frac{a}{4}+\frac{a}{4}\\
&=8C\operatorname{OPT}+a.
\end{aligned} TV ( R p , p ) = B sup ∣ E q [ μ q ( B ) − p ( B )] ∣ ≤ E q [ B sup ∣ μ q ( B ) − p ( B ) ∣ ] = E q [ TV ( μ q , p )] ≤ ( 8 η + δ 0 ) Pr [ E ] + Pr [ E c ] ≤ 8 η + δ 0 + e − k /128 ≤ 8 C OPT + 2 a + 4 a + 4 a = 8 C OPT + a .
The data-independent permutation preserves the joint distribution p T k b p^{Tkb} p T k b , so the batches S 1 , … , S T S_1,\ldots,S_T S 1 , … , S T are independent, each with distribution p k b p^{kb} p k b . Each invocation applies the same Sampler, with the same parameters and independent randomness. Hence, for every j ∈ [ T ] j\in[T] j ∈ [ T ] , L ( Z j ) = R p \mathcal{L}(Z_j)=R_p L ( Z j ) = R p . Since each Z j Z_j Z j depends only on S j S_j S j and the randomness of its invocation, the outputs are independent. Therefore, ( Z 1 , … , Z T ) ∼ R p T (Z_1,\ldots,Z_T)\sim R_p^T ( Z 1 , … , Z T ) ∼ R p T .
Applying the C C C -agnostic guarantee of L L L with T = m ( 2 a , β ) T=m(2a,\beta) T = m ( 2 a , β ) samples from R p R_p R p gives
TV ( q ^ , R p ) ≤ C inf q ∈ Q TV ( q , R p ) + 2 a \operatorname{TV}(\widehat{q},R_p)\le C\inf_{q\in\mathcal{Q}}\operatorname{TV}(q,R_p)+2a TV ( q , R p ) ≤ C q ∈ Q inf TV ( q , R p ) + 2 a
with probability at least 1 − β 1-\beta 1 − β . By the triangle inequality, inf q ∈ Q TV ( q , R p ) ≤ OPT + TV ( p , R p ) \inf_{q\in\mathcal{Q}}\operatorname{TV}(q,R_p)\le\operatorname{OPT}+\operatorname{TV}(p,R_p) inf q ∈ Q TV ( q , R p ) ≤ OPT + TV ( p , R p ) . Thus, on the same event,
TV ( q ^ , p ) ≤ TV ( q ^ , R p ) + TV ( R p , p ) ≤ C OPT + ( C + 1 ) TV ( R p , p ) + 2 a ≤ C OPT + ( C + 1 ) ( 8 C OPT + a ) + 2 a = ( 8 C 2 + 9 C ) OPT + ( C + 3 ) a = ( 8 C 2 + 9 C ) OPT + α . \begin{aligned}
\operatorname{TV}(\widehat{q},p)&\le\operatorname{TV}(\widehat{q},R_p)+\operatorname{TV}(R_p,p)\\
&\le C\operatorname{OPT}+(C+1)\operatorname{TV}(R_p,p)+2a\\
&\le C\operatorname{OPT}+(C+1)(8C\operatorname{OPT}+a)+2a\\
&=(8C^2+9C)\operatorname{OPT}+(C+3)a\\
&=(8C^2+9C)\operatorname{OPT}+\alpha.
\end{aligned} TV ( q , p ) ≤ TV ( q , R p ) + TV ( R p , p ) ≤ C OPT + ( C + 1 ) TV ( R p , p ) + 2 a ≤ C OPT + ( C + 1 ) ( 8 C OPT + a ) + 2 a = ( 8 C 2 + 9 C ) OPT + ( C + 3 ) a = ( 8 C 2 + 9 C ) OPT + α .
Finally, log ( 1 δ 0 ) = max { log ( 1 δ ) , log ( 4 a ) } ≤ log ( 4 a δ ) \log\left(\frac{1}{\delta_0}\right)=\max\left\{\log\left(\frac{1}{\delta}\right),\log\left(\frac{4}{a}\right)\right\}\le\log\left(\frac{4}{a\delta}\right) log ( δ 0 1 ) = max { log ( δ 1 ) , log ( a 4 ) } ≤ log ( a δ 4 ) . Since ε ∈ ( 0 , 1 ) \varepsilon\in(0,1) ε ∈ ( 0 , 1 ) , k = O ( 1 ε log ( 4 a δ ) ) k=O\left(\frac{1}{\varepsilon}\log\left(\frac{4}{a\delta}\right)\right) k = O ( ε 1 log ( a δ 4 ) ) . Therefore,
n 0 = T k b = O ( m ( a 16 , 1 16 ) m ( 2 a , β ) 1 ε log ( 4 a δ ) ) = O ( m ( α 16 ( C + 3 ) , 1 16 ) m ( 2 α C + 3 , β ) 1 ε log ( 4 ( C + 3 ) α δ ) ) , \begin{aligned}
n_0=Tkb=O\left(m\left(\frac{a}{16},\frac{1}{16}\right)m(2a,\beta)\frac{1}{\varepsilon}\log\left(\frac{4}{a\delta}\right)\right) \\
&=O\left(m\left(\frac{\alpha}{16(C+3)},\frac{1}{16}\right)m\left(\frac{2\alpha}{C+3},\beta\right)\frac{1}{\varepsilon}\log\left(\frac{4(C+3)}{\alpha\delta}\right)\right),
\end{aligned} n 0 = T k b = O ( m ( 16 a , 16 1 ) m ( 2 a , β ) ε 1 log ( a δ 4 ) ) = O ( m ( 16 ( C + 3 ) α , 16 1 ) m ( C + 3 2 α , β ) ε 1 log ( α δ 4 ( C + 3 ) ) ) ,
which proves the stated sample bound.
Algorithm 7 AgnosticPrivateLearner
Input: Dataset $S=(X_1,\ldots,X_n)$ with $n=Tkb$, a non-private $C$-agnostic learner $L$, and parameters $a,b,k,T,\beta,\varepsilon,\delta_0$.
Output: An estimate $\widehat{q}\in\Delta(\mathcal{X})$.
1: Define $L_{\mathrm{base}}(D):=L\left(D;\frac{a}{16},\frac{1}{16}\right)$.
2: Draw a data-independent permutation $\Pi\sim\mathop{\mathrm{U}}(S_n)$.
3: for each $j\in[T]$ do
4: $S_j\leftarrow(X_{\Pi((j-1)kb+1)},\ldots,X_{\Pi(jkb)})$.
5: $Z_j\leftarrow\operatorname{Sampler}(S_j;\ell=1,k,b,L_{\mathrm{base}},\varepsilon,\delta_0)$.
6: end for
7: $\widehat{q}\leftarrow L((Z_1,\ldots,Z_T);2a,\beta)$.
8: return $\widehat{q}$.
Next, we present a generic agnostic private learner based on the non-private Minimum distance estimator of [26 ] . This automatically gives an agnostic private learner for every countable class whose Yatracos class has finite VC dimension.
Definition C.3 (Yatracos class). Let Q ⊆ Δ ( X ) \mathcal{Q}\subseteq\Delta(\mathcal{X}) Q ⊆ Δ ( X ) be a non-empty class of distributions. For every ordered pair ( q , q ′ ) ∈ Q (q,q')\in\mathcal{Q} ( q , q ′ ) ∈ Q , fix a measurable set A q , q ′ ⊆ X A_{q,q'}\subseteq\mathcal{X} A q , q ′ ⊆ X satisfying q ( A q , q ′ ) − q ′ ( A q , q ′ ) = TV ( q , q ′ ) q(A_{q,q'})-q'(A_{q,q'})=\operatorname{TV}(q,q') q ( A q , q ′ ) − q ′ ( A q , q ′ ) = TV ( q , q ′ ) .13 The corresponding Yatracos class is defined as:
Yat ( Q ) : = { A q , q ′ : q , q ′ ∈ Q } . \operatorname{Yat}(\mathcal{Q}):=\{A_{q,q'}:q,q'\in\mathcal{Q}\}. Yat ( Q ) := { A q , q ′ : q , q ′ ∈ Q } .
Theorem C.4 (Minimum-distance estimator, [26 ] ). Let C > 0 C>0 C > 0 be a sufficiently large universal constant and α , β ∈ ( 0 , 1 ) \alpha,\beta\in(0,1) α , β ∈ ( 0 , 1 ) . Let Q ⊆ Δ ( X ) \mathcal{Q}\subseteq\Delta(\mathcal{X}) Q ⊆ Δ ( X ) be a nonempty countable14 class of distributions with d = VC ( Yat ( Q ) ) < ∞ d=\operatorname{VC}(\operatorname{Yat}(\mathcal{Q}))<\infty d = VC ( Yat ( Q )) < ∞ . There exists an algorithm, called MDE, that, for every p ∈ Δ ( X ) p\in\Delta(\mathcal{X}) p ∈ Δ ( X ) , given a dataset S ∼ p n S\sim p^n S ∼ p n of size n ≥ C ( d + log ( 2 / β ) ) / α 2 n\ge C(d+\log(2/\beta))/\alpha^2 n ≥ C ( d + log ( 2/ β )) / α 2 , returns q ^ ∈ Q \widehat{q}\in\mathcal{Q} q ∈ Q satisfying
Pr [ TV ( q ^ , p ) ≤ 3 OPT + α ] ≥ 1 − β , \Pr[\operatorname{TV}(\widehat{q},p)\le3\operatorname{OPT}+\alpha]\ge1-\beta, Pr [ TV ( q , p ) ≤ 3 OPT + α ] ≥ 1 − β , where OPT : = inf q ∈ Q TV ( p , q ) \operatorname{OPT}:=\inf_{q\in\mathcal{Q}}\operatorname{TV}(p,q) OPT := inf q ∈ Q TV ( p , q ) .
The following corollary gives a generic agnostic private learner for any countable class with finite VC ( Yat ( Q ) ) \operatorname{VC}(\operatorname{Yat}(\mathcal{Q})) VC ( Yat ( Q )) .
Corollary C.5 (Agnostic private learning Yatracos classes). Let C > 0 C>0 C > 0 be a sufficiently large universal constant and α , β , ε , δ ∈ ( 0 , 1 ) \alpha,\beta,\varepsilon,\delta\in(0,1) α , β , ε , δ ∈ ( 0 , 1 ) . Let Q ⊆ Δ ( X ) \mathcal{Q}\subseteq\Delta(\mathcal{X}) Q ⊆ Δ ( X ) be a nonempty countable class of distributions with d = VC ( Yat ( Q ) ) < ∞ d=\operatorname{VC}(\operatorname{Yat}(\mathcal{Q}))<\infty d = VC ( Yat ( Q )) < ∞ . There exists an ( ε , δ ) (\varepsilon,\delta) ( ε , δ ) -differentially private algorithm that, for every p ∈ Δ ( X ) p\in\Delta(\mathcal{X}) p ∈ Δ ( X ) , given a dataset S ∼ p n S\sim p^n S ∼ p n of size n ≥ C ( d + 1 ) ( d + log ( 2 / β ) ) ε α 4 log ( 2 α δ ) n\ge C\frac{(d+1)(d+\log(2/\beta))}{\varepsilon\alpha^4}\log\left(\frac{2}{\alpha\delta}\right) n ≥ C ε α 4 ( d + 1 ) ( d + l o g ( 2/ β )) log ( α δ 2 ) , returns q ^ ∈ Q \widehat{q}\in\mathcal{Q} q ∈ Q satisfying
Pr [ TV ( q ^ , p ) ≤ 99 OPT + α ] ≥ 1 − β , \Pr[\operatorname{TV}(\widehat{q},p)\le99\operatorname{OPT}+\alpha]\ge1-\beta, Pr [ TV ( q , p ) ≤ 99 OPT + α ] ≥ 1 − β , where OPT : = inf q ∈ Q TV ( p , q ) \operatorname{OPT}:=\inf_{q\in\mathcal{Q}}\operatorname{TV}(p,q) OPT := inf q ∈ Q TV ( p , q ) .
:::
Proof. In Algorithm 7, take L ( D ; t , ρ ) : = MDE ( D ; t , ρ ) L(D;t,\rho) := \operatorname{MDE}(D;t,\rho) L ( D ; t , ρ ) := MDE ( D ; t , ρ ) . By Theorem C.4 , this is a proper 3-agnostic learner for Q \mathcal{Q} Q with sample complexity m ( t , ρ ) = ⌈ C 0 d + log ( 2 / ρ ) t 2 ⌉ m(t,\rho) = \left\lceil C_0\frac{d+\log(2/\rho)}{t^2}\right\rceil m ( t , ρ ) = ⌈ C 0 t 2 d + l o g ( 2/ ρ ) ⌉ , where C 0 C_0 C 0 is a universal constant. Theorem C.2 therefore makes Algorithm 7 ( ε , δ ) (\varepsilon,\delta) ( ε , δ ) -DP and guarantees q ^ ∈ Q \widehat{q} \in\mathcal{Q} q ∈ Q with TV ( q ^ , p ) ≤ 99 OPT + α \operatorname{TV}(\widehat{q},p) \le99\operatorname{OPT}+\alpha TV ( q , p ) ≤ 99 OPT + α with probability at least 1 − β 1-\beta 1 − β .
Its sample bound is n 0 = O ( m ( α 96 , 1 16 ) m ( α 3 , β ) 1 ε log ( 24 α δ ) ) n_0 = O\left(m\left(\frac{\alpha}{96},\frac{1}{16}\right)m\left(\frac{\alpha}{3},\beta\right)\frac{1}{\varepsilon}\log\left(\frac{24}{\alpha\delta}\right)\right) n 0 = O ( m ( 96 α , 16 1 ) m ( 3 α , β ) ε 1 log ( α δ 24 ) ) . Since m ( α 96 , 1 16 ) = O ( d + 1 α 2 ) m\left(\frac{\alpha}{96},\frac{1}{16}\right) = O\left(\frac{d+1}{\alpha^2}\right) m ( 96 α , 16 1 ) = O ( α 2 d + 1 ) and m ( α 3 , β ) = O ( d + log ( 2 / β ) α 2 ) m\left(\frac{\alpha}{3},\beta\right) = O\left(\frac{d+\log(2/\beta)}{\alpha^2}\right) m ( 3 α , β ) = O ( α 2 d + l o g ( 2/ β ) ) , this becomes n 0 = O ( ( d + 1 ) ( d + log ( 2 / β ) ) ε α 4 log ( 2 α δ ) ) n_0 = O\left(\frac{(d+1)(d+\log(2/\beta))}{\varepsilon\alpha^4}\log\left(\frac{2}{\alpha\delta}\right)\right) n 0 = O ( ε α 4 ( d + 1 ) ( d + l o g ( 2/ β )) log ( α δ 2 ) ) , which proves the claimed guarantee. □ \square □
For potentially infinite classes, Bun et al. [17 Theorem 4] give a private hypothesis selection guarantee depending on d = VC ( Yat ( Q ) ) d = \operatorname{VC}(\operatorname{Yat}(\mathcal{Q})) d = VC ( Yat ( Q )) and a locally small cover. An α \alpha α -cover H ⊆ Q \mathcal{H} \subseteq\mathcal{Q} H ⊆ Q contains an element within TV \operatorname{TV} TV distance α \alpha α of every q ∈ Q q \in\mathcal{Q} q ∈ Q , and is ( k , r ) (k,r) ( k , r ) -locally small if sup h ∈ H ∣ { h ′ ∈ H : TV ( h , h ′ ) ≤ r } ∣ ≤ k \sup_{h\in\mathcal{H}}|\{h' \in\mathcal{H} : \operatorname{TV}(h,h') \le r\}| \le k sup h ∈ H ∣ { h ′ ∈ H : TV ( h , h ′ ) ≤ r } ∣ ≤ k . Applied to a ( k , 8 α ) (k,8\alpha) ( k , 8 α ) -locally small α \alpha α -cover, their ( ε , δ ) (\varepsilon,\delta) ( ε , δ ) -private algorithm uses O ( d + log ( 1 / β ) α 2 + log ( k / β ) + log ( 1 / δ ) ε ) O\left(\frac{d+\log(1/\beta)}{\alpha^2}+\frac{\log(k/\beta)+\log(1/\delta)}{\varepsilon}\right) O ( α 2 d + l o g ( 1/ β ) + ε l o g ( k / β ) + l o g ( 1/ δ ) ) samples and achieves the 7-agnostic guarantee TV ( q ^ , p ) ≤ 7 α \operatorname{TV}(\widehat{q},p) \le7\alpha TV ( q , p ) ≤ 7 α with probability at least 1 − β 1-\beta 1 − β , provided TV ( p , h ) ≤ α \operatorname{TV}(p,h) \le\alpha TV ( p , h ) ≤ α for some h ∈ H h \in\mathcal{H} h ∈ H . This guarantee requires a supplied approximation bound. For countable classes, Corollary C.5 removes both this requirement and the locally small cover assumption: finite d d d alone suffices for a 99-agnostic private learner, guaranteeing TV ( q ^ , p ) ≤ 99 OPT + α \operatorname{TV}(\widehat{q},p) \le99\operatorname{OPT}+\alpha TV ( q , p ) ≤ 99 OPT + α with probability at least 1 − β 1-\beta 1 − β for every p p p . This gives a qualitative improvement in the assumptions, at the cost of a larger sample complexity and approximation constant.
Omitted proof from Section 6
Here, we formally prove the guarantee for the boosted learner introduced in § 5.1 . Its core idea is confidence boosting: amplifying a learner’s constant success probability to 1 − β 1-\beta 1 − β through independent repetition and robust aggregation.
Proof of Theorem 6.1. Set α 0 = α 3 \alpha_0 = \frac{\alpha}{3} α 0 = 3 α , β 0 = 1 8 \beta_0 = \frac{1}{8} β 0 = 8 1 , δ 0 = min { δ , β 0 / 4 } \delta_0 = \min\{\delta,\beta_0/4\} δ 0 = min { δ , β 0 /4 } , and T = ⌈ 128 ln ( 1 / β ) ⌉ T = \lceil128\ln(1/\beta)\rceil T = ⌈ 128 ln ( 1/ β )⌉ . Fixing β 0 \beta_0 β 0 keeps the block size c c c independent of the target failure probability β \beta β . We amplify confidence by increasing the number of independent runs T T T . By Theorem 5.1 , each private learner is ( ε , δ 0 ) (\varepsilon,\delta_0) ( ε , δ 0 ) -DP. Since the runs use disjoint blocks and independent randomness, parallel composition (Lemma 2.5 ) gives the same privacy guarantee for the vector of estimates. Selecting p ^ j \widehat{p}_j p j uses only these estimates, so post-processing preserves privacy (Proposition 2.4 ). Since δ 0 ≤ δ \delta_0 \le\delta δ 0 ≤ δ , the overall procedure is ( ε , δ ) (\varepsilon,\delta) ( ε , δ ) -DP.
For utility, Theorem 5.1 provides a block size c = O ( m ( β 0 64 m ( α 0 , β 0 / 4 ) , 1 16 ) 1 ε log ( 4 β 0 δ 0 ) ) c = O\left(m\left(\frac{\beta_0}{64m(\alpha_0,\beta_0/4)},\frac{1}{16}\right)\frac{1}{\varepsilon}\log\left(\frac{4}{\beta_0\delta_0}\right)\right) c = O ( m ( 64 m ( α 0 , β 0 /4 ) β 0 , 16 1 ) ε 1 log ( β 0 δ 0 4 ) ) such that each estimate satisfies TV ( p ^ t , p ) < α 0 \operatorname{TV}(\widehat{p}_t,p) < \alpha_0 TV ( p t , p ) < α 0 with probability at least 1 − β 0 1-\beta_0 1 − β 0 . Fix p ∈ Q p \in\mathcal{Q} p ∈ Q and let T = { t ∈ [ T ] : TV ( p ^ t , p ) < α 0 } \mathcal{T} = \{t \in[T] : \operatorname{TV}(\widehat{p}_t,p) < \alpha_0\} T = { t ∈ [ T ] : TV ( p t , p ) < α 0 } . The estimates are independent because the input is i.i.d., the blocks are disjoint, and the runs use independent randomness. Hoeffding’s inequality therefore gives Pr [ ∣ T ∣ ≤ T 2 ] ≤ exp ( − 2 T ( 1 2 − β 0 ) 2 ) ≤ β \Pr\left[|\mathcal{T}| \le\frac{T}{2}\right] \le\exp\left(-2T\left(\frac{1}{2}-\beta_0\right)^2\right) \le\beta Pr [ ∣ T ∣ ≤ 2 T ] ≤ exp ( − 2 T ( 2 1 − β 0 ) 2 ) ≤ β . On the event ∣ T ∣ > T / 2 |\mathcal{T}| > T/2 ∣ T ∣ > T /2 , any two estimates indexed by T \mathcal{T} T are within 2 α 0 2\alpha_0 2 α 0 of one another. Thus, N t ≥ ∣ T ∣ N_t \ge|\mathcal{T}| N t ≥ ∣ T ∣ for every t ∈ T t \in\mathcal{T} t ∈ T , and the maximizing index j j j satisfies N j ≥ ∣ T ∣ > T / 2 N_j \ge|\mathcal{T}| > T/2 N j ≥ ∣ T ∣ > T /2 . The estimates counted by N j N_j N j must include some i ∈ T i \in\mathcal{T} i ∈ T . Consequently, TV ( p ^ j , p ) ≤ TV ( p ^ j , p ^ i ) + TV ( p ^ i , p ) < 3 α 0 = α \operatorname{TV}(\widehat{p}_j,p) \le\operatorname{TV}(\widehat{p}_j,\widehat{p}_i)+\operatorname{TV}(\widehat{p}_i,p) < 3\alpha_0 = \alpha TV ( p j , p ) ≤ TV ( p j , p i ) + TV ( p i , p ) < 3 α 0 = α . This proves the accuracy guarantee. Finally, n = T c n = Tc n = T c . Since β 0 = 1 / 8 \beta_0 = 1/8 β 0 = 1/8 and δ 0 = min { δ , 1 32 } \delta_0 = \min\{\delta,\frac{1}{32}\} δ 0 = min { δ , 32 1 } , we have log ( 4 β 0 δ 0 ) = O ( log ( 4 δ ) ) \log\left(\frac{4}{\beta_0\delta_0}\right) = O\left(\log\left(\frac{4}{\delta}\right)\right) log ( β 0 δ 0 4 ) = O ( log ( δ 4 ) ) and T = O ( log ( 4 β ) ) T = O\left(\log\left(\frac{4}{\beta}\right)\right) T = O ( log ( β 4 ) ) . Substituting these choices gives the stated bound.
Finally, if m ( α , β ) = poly ( 1 α , log ( 1 β ) ) m(\alpha,\beta) = \operatorname{poly}\left(\frac{1}{\alpha},\log\left(\frac{1}{\beta}\right)\right) m ( α , β ) = poly ( α 1 , log ( β 1 ) ) , this bound is poly ( 1 α , log ( 1 β ) , 1 ε , log ( 1 δ ) ) \operatorname{poly}\left(\frac{1}{\alpha},\log\left(\frac{1}{\beta}\right),\frac{1}{\varepsilon},\log\left(\frac{1}{\delta}\right)\right) poly ( α 1 , log ( β 1 ) , ε 1 , log ( δ 1 ) ) . □ \square □