Robust and Differentially Private Mean Estimation
Xiyang Liu, Weihao Kong, Sham Kakade, Sewoong Oh
Introduction
When releasing database statistics on a collection of entries from individuals, we would ideally like to make it impossible to reverse-engineer each individual’s potentially sensitive information. Privacy-preserving techniques add just enough randomness tailored to the statistical task to guarantee protection. At the same time, it is becoming increasingly common to apply such techniques to databases collected from multiple sources, not all of which can be trusted. Emerging data access frameworks, such as federated analyses across users’ devices or data silos , make it easier to temper with such collected datasets, leaving private statistical analyses vulnerable to a malicious corruption of a fraction of the data.
Differential privacy has emerged as a widely accepted de facto measure of privacy, which is now a standard in releasing the statistics of the U.S. Census data statistics and also deployed in real-world commercial systems . A statistical analysis is said to be differentially private (DP) if the likelihood of the (randomized) outcome does not change significantly when a single arbitrary entry is added/removed (formally defined in §1.2). This provides a strong privacy guarantee: even a powerful adversary who knows all the other entries in the database cannot confidently identify whether a particular individual is participating in the database based on the outcome of the analysis. This ensures plausible deniability, central to protecting an individual’s privacy.
In this paper, we focus on one of the most canonical problems in statistics: estimating the mean of a distribution from i.i.d. samples. For distributions with unbounded support, such as sub-Gaussian and heavy-tailed distributions, fundamental trade-offs between accuracy, sample size, and privacy have only recently been identified and efficient private estimators proposed. However, these approaches are brittle when a fraction of the data is corrupted, posing a real threat, referred to as data poisoning attacks . In defense of such attacks, robust (but not necessarily private) statistics has emerged as a popular setting of recent algorithmic and mathematical breakthroughs .
One might be misled into thinking that privacy ensures robustness since DP guarantees that a single outlier cannot change the estimation too much. This intuition is true only in a low dimension; each sample has to be an obvious outlier to significantly change the mean. However, in a high dimension, each corrupted data point can look perfectly uncorrupted but still shift the mean significant when colluding together (e.g., see Fig. 1). Focusing on the canonical problem of mean estimation, we introduce novel algorithms that achieve robustness and privacy simultaneously even when a fraction of data is corrupted arbitrarily. For such algorithms, there is a fundamental question of interest: do we need more samples to make private mean estimation also robust against adversarial corruption?
Algorithm 2 is -DP. When fraction of the data is arbitrarily corrupted from samples from a -dimensional sub-Gaussian distribution with mean and an identity sub-Gaussian parameter, if then Algorithm 2 achieves w.h.p.
PRIME is -DP and under the assumption of Thm.1, if , achieves w.h.p.
Heavy-tailed distributions. When samples are drawn from a distribution with a bounded covariance, parameters of Algorithm 2 can be modified to nearly match the optimal sample complexity of (non-robust) private mean estimation in Table 2. This algorithm also matches the fundamental limit on the accuracy of (non-private) robust estimation, which in this case is .
The proposed PRIME-ht for covariance bounded distributions achieve computational efficiency at the cost of an extra factor of in sample size. This bottleneck is also due to DP PCA, and it remains open whether this gap can be closed by an efficient estimator.
PRIME-ht is -DP and if achieves w.h.p. under the assumptions of Thm. 3.
We introduce PRIME which simultaneously achieves -DP and robustness against -fraction of corruption. A major challenge in making a standard filter-based robust estimation algorithm (e.g., ) private is the high sensitivity of the filtered set that we pass from one iteration to the next. We propose a new framework which makes private only the statistics of the set, hence significantly reducing the sensitivity. Our major innovation is a tight analysis of the end-to-end sensitivity of this multiple interactive accesses to the database. This is critical in achieving robustness while preserving privacy and is also of independent interest in making general iterative filtering algorithms private.
The classical filter approach (see, e.g. ) needs to access the database times, which brings an extra factor in the sample complexity due to DP composition. In order to reduce the iteration complexity, following the approach in , we propose filtering multiple directions simultaneously using a new score based on the matrix multiplicative weights (MMW). In order to privatize the MMW filter, our major innovation is a novel adaptive filtering algorithm DPthreshold() that outputs a single private threshold which guarantees sufficient progress at every iteration. This brings the number of database accesses from to .
One downside of PRIME is that it requires an extra factor in the sample complexity, compared to known lower bounds for (non-robust) DP mean estimation. To investigate whether this is also necessary, we propose a sample optimal exponential time robust mean estimation algorithm in §4 and prove that there is no extra statistical cost to jointly requiring privacy and robustness. Our major technical innovations is in using resilience property of the dataset to not only find robust mean (which is the typical use case of resilience) but also bound sensitivity of that robust mean.
2 Preliminary on differential privacy (DP)
DP is a formal metric for measuring privacy leakage when a dataset is accessed with a query .
Let be a random vector with entries i.i.d. sampled from Laplace distribution with pdf . Let denote a Gaussian random vector with mean and covariance .
We use these output perturbation mechanisms along with the exponential mechanism as building blocks. Appendix A provides detailed survey of privacy and robust estimation.
3 Problem formulation
Similarly, we consider the same problem for heavy-tailed distributions with a bounded covariance. We present the assumption and main results for covariance bounded distributions in Appendix B.
Outline. We present PRIME for sub-Gaussian distribution in §2, and present theoretical analysis in §3. We then introduce an exponential time algorithm with near optimal guarantee in §4. Due to space constraints, analogous results for heavy-tailed distributions are presented in Appendix B.
PRIME: efficient algorithm for robust and DP mean estimation
In order to describe the proposed algorithm PRIME, we need to first describe a standard (non-private) iterative filtering algorithm for robust mean estimation.
Non-private robust mean estimation approaches recursively apply the following filter, whose framework is first proposed in . Given a dataset , the current set of data points is updated starting with . At each step, the following filter (Algorithm 1 in ) attempts to detect the corrupted data points and remove them.
Compute the top eigenvector of the covariance of the current data set ;
Compute scores for all data points : ;
Draw a random threshold: ;
Remove outliers from defined as is in the largest -tail of and , where
This is repeated until the empirical covariance is sufficiently small and the empirical mean is output. At a high level, the correctness of this algorithm relies on the key observation that the -fraction of adversarial corruption can not significantly change the mean of the dataset without introducing large eigenvalues in the empirical covariance. Therefore, the algorithm finds top eigenvector of the empirical covariance in step , and tries to correct the empirical covariance by removing corrupted data points. Each data point is assigned a score in step 2 which indicates the “badness” of the data points, and a threshold in step 3 is carefully designed such that step 4 guarantees to remove more corrupted data points than good data points (in expectation). This guarantees the following bound achieving the near-optimal sample complexity shown in the second row of Table 1. A formal description of this algorithm is in Algorithm 4 in Appendix C.
Under assumption 1, the above filtering algorithm achieves accuracy w.p. if .
Challenges in making robust mean estimation private. To get a DP and robust mean, a naive attempt is to apply a standard output perturbation mechanism to . However, this is obviously challenging since the end-to-end sensitivity is intractable. The standard recipe to circumvent this is to make the current “state” private at every iteration. Once is private (hence, public knowledge), making the next “state” private is simpler. We only need to analyze the sensitivity of a single step and apply some output perturbation mechanism with . End-to-end privacy is guaranteed by accounting for all these ’s using the advanced composition . This recipe has been quite successful, for example, in training neural networks with (stochastic) gradient descent , where the current state can be the optimization variable . However, for the above (non-private) filtering algorithm, this standard recipe fails, since the state is a set and has large sensitivity. Changing a single data point in can significantly alter which (and how many) samples are filtered out.
2 A new framework for private iterative filtering
Instead of making the (highly sensitive) itself private, we propose a new framework which makes private only the statistics of : the mean and the top principal direction . There are two versions of this algorithm, which output the exactly same with the exactly same privacy guarantees, but are written from two different perspectives. We present here the interactive version from the perspective of an analyst accessing the dataset via DP queries ( and ), because this version makes clear the inner operations of each private mechanisms, hence making the sensitivity analysis transparent, checking the correctness of privacy guarantees easy, and tracking privacy accountant simple. In practice, one should implement the centralized version (Algorithm 7 in Appendix D), which is significantly more efficient.
We give a high-level explanation of each step of Algorithm 1 here and give the formal definitions of all the queries in Appendix D. First, returns (the parameters of) a hypercube that is guaranteed to include all uncorrupted samples while preserving privacy. This is achieved by running coordinate-wise private histograms and selecting as the center of the largest bin for the -th coordinate. Since covariance is , returns a fixed . Such an adaptive estimate of the support is critical in tightly bounding the sensitivity of all subsequent queries, which operate on the clipped dataset; all data points are projected as in all the queries that follow. With clipping, a single data point can now change at most by .
The subsequent steps perform the non-private filtering algorithm of §2.1, but with private statistics and . As the set changes over time, we lower bound its size (which we choose to be ) to upper bound the sensitivity of other queries and .
Recall that two datasets are neighboring, i.e., , iff . This lemma implies that if two datasets are neighboring, then they are still neighboring after filtering with the same parameters, no matter how many times we filter them. Hence, this lemma allows us to use the standard output-perturbation mechanisms with -DP. Advanced composition ensures that end-to-end guarantee of such queries is -DP. Together with -DP budget used in , this satisfied the target privacy. Analyzing the utility of this algorithm, we get the following guarantee.
Algorithm 1 is -DP. Under Assumption 1, there exists a universal constant such that if and then Algorithm 1 achieves with probability .
The first term in the sample complexity is optimal (cf. Table 1), but there is a factor of gap in the second term. This is due to the fact that we need to run iterations in the worst-case. Such numerous accesses to the database result in large noise to be added at each iteration, requiring large sample size to combat that extra noise. We introduce PRIME to reduce the number of iterations to and significantly reduce the sample complexity.
3 PRIME: novel robust and private mean estimator
Algorithm 1 (specifically Filter() in Algorithm 1) accesses the database times. This is necessary for two reasons. First, the filter checks only one direction at each iteration. In the worst case, the corrupted samples can be scattered in orthogonal directions such that the filter needs to be repeated times. Secondly, even if the corrupted samples are clustered together in one direction, the filter still needs to be repeated times. This is because we had to use a large (random) threshold of to make the threshold data-independent so that we can keep the sensitivity of Filter() low, which results in slow progress. We propose filtering multiple directions simultaneously using a new score based on the matrix multiplicative weights. Central to this approach is a novel adaptive filtering algorithm DPthreshold() that guarantees sufficient decrease in the total score at every iteration.
for some choice of . If we set the number of iterations to one, a choice of recovers the previous score that relied on the top singular vector from §2.1 and a choice of gives a simple norm based score . An appropriate choice of smoothly interpolates between these two extremes, which ensures that iterations are sufficient for the spectral norm of the covariance to decrease strictly by a constant factor. This guarantees that after epochs, we sufficiently decrease the covariance to ensure that the empirical mean is accurate enough. Critical in achieving this gain is our carefully designed filtering algorithm DPthreshold that uses the privately computed MMW-based scores using Gaussian mechanism on the covariance matrices as shown in Algorithm 11 in Appendix E.
Novelty. The corresponding non-private filtering of [36, Algorithm 9] for robust mean estimation takes advantage of an adaptive threshold, but filters out each sample independently resulting in a prohibitively large sensitivity; the coupling between each sample and the randomness used to filter it can change widely between two neighboring datasets. On the other hand, Algorithm 1 (i.e., Filter() in Algorithm 6) takes advantage of jointly filtering all points above a single threshold with a single randomness , but the non-adaptive (and hence large) choice of the range results in a large number of iterations because each filtering only decrease the score by little. To sufficiently reduce the total score while maintaining a small sensitivity, we introduce a filter with a single and adaptive threshold.
Algorithm. Our goal here is to privately find a single scalar such that when a randomized filter is applied on the scores with a (random) threshold (with drawn uniform in $\rho=\max_{j\in S_{t}}\tau_{j}$. While this guarantees the filter removes more corrupted samples than good samples, it does not make sufficient progress in reducing the total score of the samples.
Ideally, we want the thresholding to decrease the total score by a constant multiplicative factor, which will in the end allow the algorithm to terminate within logarithmic iterations. To this end, we propose a new scheme of using the largest such that the following inequality holds:
We use a private histogram of the scores to approximate this threshold. Similar to , we use geometrically increasing bin sizes such that we use only bins while achieving a preferred multiplicative error in our quantization. At each epoch and iteration , we run DPthreshold sketched in the following to approximate followed by a random filter. Step 3 replaces the non-private condition in Eq. (1). A complete description is provided in Algorithm 11.
Privately compute scores for all data points ;
Analyses of PRIME
Building on the framework of Algorithm 1, PRIME (Algorithm 9) replaces the score with the MMW-based score presented in §2.3.1 and the filter with the adaptive DPthreshold. This reduces the number of iterations to achieving the following bound.
PRIME is -differentially private. Under Assumption 1 there exists a universal constant such that if and , then PRIME achieves with probability .
A proof is provided in Appendix F. The notation hides logarithmic terms in , , and . To achieve an error of , the first term is necessary even if there is no corruption. The accuracy of matches the lower bound shown in for any polynomial time statistical query algorithm, and it nearly matches the information theoretical lower bound on robust estimation of . On the other hand, the second term of has an extra factor of compared to the optimal one achieved by exponential time Algorithm 2. It is an open question if this gap can be closed by a polynomial time algorithm.
The bottleneck is the private matrix multiplicative weights. Such spectral analyses are crucial in filter-based robust estimators. Even for a special case of privately computing the top principal component, the best polynomial time algorithm requires samples , and this sample complexity is also necessary as shown in [39, Corollary 25].
To boost the success probability to for some small , we need an extra factor in the sample complexity to make sure the dataset satisfies the regularity condition with probability . Then we can run PRIME times and choose the output of a run that satisfies and at termination.
Numerical experiments support our theoretical claims. The left figure with is in the large regime where the DP Mean error is dominates by and PRIME error by . Hence, PRIME error is constant whereas DP Mean error increases with the dimension . The second figure with is in the small regime when DP Mean error consists of and PRIME is dominated by . Both increase with the dimension , and the gap can be made large by increasing . The right figure with is when DP Mean error is dominated by and PRIME by when . Below this threshold, which happens in this example around , the added noise in the private mechanism starts to dominate with decreasing . Both algorithms have respective thresholds below which the error increases with decreasing . This threshold is larger for PRIME because it uses the privacy budget to perform multiple operations and hence the noise added to the final output is larger compared to DP Mean. Below this threshold, which can be easily determined based on the known parameters , we should either collect more data (which will decrease the threshold) or give up filtering and spend all privacy budget on and the empirical mean (which will reduce the error). Details of the experiments are in Appendix L.
Exponential time algorithm with near-optimal sample complexity
Novelty. An existing exponential time algorithm for robust and private mean estimation in strictly requires the uncorrupted samples to be drawn from a Gaussian distribution. We also provide a similar algorithm based on private Tukey median in Appendix I and its analysis in Appendix J. In this section, we introduce a novel estimator that achieves near-optimal guarantees for more general sub-Gaussian distributions (and also covariance bounded distributions) but takes an exponential run-time. Its innovation is in leveraging on the resilience property of well-behaved distributions not only to estimate the mean robustly (which is the standard use of the property) but also to adaptively bound the sensitivity of the estimator, thus achieving optimal privacy-accuracy tradeoff.
Algorithm. As data is corrupted, we define as a surrogate for resilience of the uncorrupted part of the set. If indeed consists of a fraction of independent samples from the promised class of distributions, the goodness score will be close to the resilience property of the good data.
For , let us define
Algorithm 2 first checks if the resilience matches that of the promised distribution. The data is pre-processed with to ensure we can check privately. Once resilience is cleared, we can safely use the exponential mechanism based on the score function in Definition 4.3 to select an approximate robust mean privately. The choice of the sensitivity critically relies on the fact that resilient datasets have small sensitivity of . Without the resilience check, the sensitivity is resulting in an extra factor of in the sample complexity.
We propose the score function in the following definition, which is a robust estimator of the distance between the mean and the candidate .
Analysis. For any direction , the truncated mean estimator provides a robust estimation of the true mean along the direction , thus the distance can be simply defined by taking the maximum over all directions . We show the sensitivity of this simple estimator is bounded by the resilience property divided by , which is once the resilience check is passed. This leads to the following near-optimal sample complexity. We provide a proof in Appendix H.2.
Algorithm 2 is -DP. Under Assumption 1, this algorithm achieves with probability if
Run-time. Computing exactly can take operations. The exponential mechanism implemented with -covering for and a constant covering for can take operations.
Conclusion
There are several directions for improving our results further and applying the framework to solve other problems. PRIME provides a new design principle for private and robust estimation. This can be more broadly applied to fundamental statistical analyses such as robust covariance estimation robust PCA , and robust linear regression .
PRIME could be improved in a few directions. First, the sample complexity of in Theorem 6 is suboptimal in the second term. Improving the factor requires bypassing differentially private singular value decomposition, which seems to be a challenging task. However, it might be possible to separate the factor from the rest of the terms and get an additive error of the form . This requires using Laplace mechanism in private MMW (line 10 Algortihm 10). Secondly, the time complexity of PRIME is dominated by computation time of the matrix exponential in (line 10 Algortihm 10). Total number of operations scale as . One might hope to achieve time complexity using approximate computations of ’s using techniques from . This does not improve the sample complexity, as the number of times the dataset is accessed remains the same. Finally, for (non-robust) private mean estimation, CoinPress provides a practical improvement in the small sample regime by progressively refining the search space . The same principle could be applied to PRIME to design a robust version of CoinPress. One important question remains open; how are differential privacy and robust statistics fundamentally related? We believe our exponential time algorithm hints on a fundamental connection between robust statistics of a data projected onto one-dimensional subspace and sensitivity of resulting score function for the exponential mechanism. It is an interesting direction to pursue this connection further to design novel algorithms that bridge privacy and robustness.
Acknowledgement
Sham Kakade acknowledges funding from the National Science Foundation under award CCF-1703574. Sewoong Oh acknowledges funding from Google faculty research award, NSF grants IIS-1929955, CCF-1705007, CNS-2002664, CCF 2019844 as a part of Institute for Foundation of Machine Learning, and CNS-2112471 as a part of Institute for Future Edge Networks and Distributed Intelligence.
References
Appendix
In an attempt to design efficient algorithms for robust and private mean estimation, proposed an algorithm with a mis-calculated sensitivity, which can result in violating the privacy guarantee. This can be corrected by pre-processing with our approach of checking the resilience (as in Algorithm 2), but this requires a run-time exponential in the dimension.
However, under -corruption, shows that achieving an error better than under -th moment bound is as computationally hard as the small-set expansion problem, even without requiring DP. Hence, under the assumption of , no polynomial-time algorithm exists that can outperform our PRIME-ht even if we have stronger assumptions of -th moment bound. On the other hand, there exists an exponential time algorithm for non-private robust mean estimation that achieves . Combining it with the bound of , an interesting open question is whether there is an (exponential time) algorithm that achieves with sample complexity under -corruption and -DP.
Robust estimation. Designing robust estimators under the presence of outliers has been considered by statistics community since 1960s . Recently, give the first polynomial time algorithm for mean and covariance estimation with no (or very weak) dependency on the dimensionality in the estimation error. Since then, there has been a flurry of research on robust estimation problems, including mean estimation , covariance estimation , linear regression and sparse regression , principal component analysis , mixture models and list-decodable learning . See for a survey of recent work.
One line of work that is particularly related to our algorithm PRIME is , which leverage the ideas from matrix multiplicative weight and fast SDP solver to achieve faster, sometimes nearly linear time, algorithms for mean and covariance estimation. In PRIME, we use a matrix multiplicative weight approach similar to to reduce the iteration complexity to logarithmic, which enables us to achieve the dependency in the sample complexity.
The concept of resilience is introduced in as a sufficient condition such that learning in the presence of adversarial corruption is information-theoretically possible. The idea of resilience is later generalized in for a wider range of adversarial corruption models. While there exists simple exponential time robust estimation algorithm under resilience condition, it is challenging to achieve differential privacy due to high sensitivity. We propose a novel approach to leverage the resilience property in our exponential time algorithm for sub-gaussian and heavy-tailed distributions.
Appendix B Main results under heavy-tailed distributions
We consider distributions with bounded covariance as defined as follows.
Under these assumptions, Algorithm 2 achieves near optimal guarantees but takes exponential time. The dominant term in the sample complexity cannot be improved as it matches that of the optimal non-robust private estimation . The accuracy cannot be improved as it matches that of the optimal non-private robust estimation . We provide a proof in Appendix H.1.
Algorithm 2 is -differentially private. Under Assumption 2, if
this algorithm achieves with probability .
PRIME-ht is -differentially private. Under Assumption 2 there exists a universal constant such that if , and , then PRIME-ht achieves with probability . The notation hides logarithmic terms in , and .
Remark 1. To boost the success probability to for some small , we will randomly split the data into subsets of equal sizes, and run Algorithm 3 to obtain a mean estimation from each of the subset. Then we can apply multivariate “mean-of-means” type estimator to get with probability . This is efficient as we only have trials and run-time of mean-of-means is dominated by the time it takes to find all pairwise distances, which is only . There are pairs, and for each pair we compute the distance between means in operations.
Appendix C Background on (non-private) robust mean estimation
The following tie-breaking rule is not essential for robust estimation, but is critical for proving differential privacy, as shown later in Appendix F.1.
Given a set of scalar values for a subset , define the sorted list of such that for all . When there is a tie such that , it is broken by . Further ties are broken by comparing the remaining entries of and , in an increasing order of the coordinate. If ,then the tie is broken arbitrarily. We define to be the set of largest valued samples.
With this definition of -tail, we can now provide a complete description of the robust mean estimation that achieves the guarantee provided in Proposition 2.1.
Appendix D A new framework for private iterative filtering
We provide complete descriptions of all algorithms used in private iterative filtering. We present the interactive version first, followed by the centralized version.
Adaptive estimation of the range of the dataset is essential in computing private statistics of data. We use the following algorithm proposed in . It computes a private histogram of a set of 1-dimensional points and select the largest bin as the one potentially containing the mean of the data. Note that does not need not be chosen adaptively to include all the uncorrupted data with a high probability.
The following guarantee (and the algorithm description) is used in the analysis (and the implementation) of the query .
This is an intermediate result in the proof of Lemma 2.3 in . Note that, conceptually, we are applying the private histogram algorithm to an infinite number of bins in the intervals each of length . This is possible because the algorithm only changes the bins that are occupied by at least on sample. Practically, we only need to add noise to those bins that are occupied, and hence we limit the range from to without loss of generality and without any changes to the privacy guarantee of the algorithm.
D.2 Centralized version of the algorithm
In practice, one should run the centralized version of the private iterative filtering, in order to avoid multiple redundant computations of the interactive version. The main difference is that the redundant filtering repeated every time a query is called in the interactive version is now merged into a single run. The resulting estimation and the privacy loss are exactly the same.
First, introduced in , returns a hypercube that is guaranteed to include all uncorrupted samples, while preserving privacy. It is followed by a private filtering DPfilter in Algorithm 8.
D.3 The analysis of private iterative filtering (Algorithms 1 and 7) and a proof of Theorem 5
, introduced in , returns a hypercube that is guaranteed to include all uncorrupted samples, while preserving privacy. In the following lemma, we show that is also robust to adversarial corruption. Such adaptive bounding of the support is critical in privacy analysis of the subsequent steps. We clip all data points by projecting all the points with to lie inside the hypercube and pass them to DPfilter for filtering. The algorithm and a proof are provided in §D.3.1.
(Algorithm 5) is -differentially private. Under Assumption 1, returns such that if and , then all uncorrupted samples in are in with probability .
In DPfilter, we make only the mean and the top principal direction private to decrease sensitivity. The analysis is now more challenging since depends on all past iterates and internal randomness . To decrease the sensitivity, we modify the filter in line 8 to use the maximum support (which is data independent) instead of the maximum contribution (which is data dependent and sensitive). While one data point can significantly change and the output of one step of the filter in Algorithm 4, the sensitivity of the proposed filter is bounded conditioned on all past , as we show in the following lemma. This follows from the fact that conditioned on , the proposed filter is a contraction. We provide a proof in Appendix D.3.3 and Appendix D.3.4. Putting together Lemmas D.2 and D.3, we get the desired result in Theorem 5.
DPfilter is -differentially private. Under the hypotheses of Theorem 5, DPfilter achieves with probability , if and is large enough such that the original uncorrupted samples are inside the hypercube .
Differential privacy guarantee. To achieve end-to-end target privacy guarantee, Algorithm 7 separates the privacy budget into two. The ()-DP guarantee of follows from Lemma D.2. The ()-DP guarantee of DPfilter follows from Lemma D.3.
Accuracy. From Lemma D.2 is guaranteed to return a hypercube that includes all clean data in the dataset. It follows from Lemma D.3 that when , we have .
Assuming that , we have that with probability , . Using the assumption that , since . This implies that with probability , the algorithm choose the bin from , which means the estimate . By the tail bound of sub-Gaussian distribution and a union bound over , we have that with probability , for all and , .
Proof of Lemma 2.2. We only need to show that one step of the proposed filter is a contraction. To this end, we only need to show contraction for two datasets at distance 1, i.e., . For fixed and , we apply filter to set of scalars and , whose distance is also one. If the entries that are different (say and ) are both below the subset of the top points (as in Definition C.1), then the same set of points will be removed for both and the distance is preserved . If they are both above the top subset, then either both are removed, one of them is removed, or both remain. The rest of the points that are removed coincide in both sets. Hence, . If is below and is above the top subset of respective datasets, then either is not removed (in which case ) or is removed (in which case and the distance remains one).
Note that when there are ties, it is critical to resolve them in a consistent manner in both datasets and . The tie breaking rule of Definition C.1 is critical in sorting those samples with the same score ’s in a consistent manner.
Proof of Lemma F.1. The analysis of contraction of the filtering step in DPMMWfilter is analogous to that of private iterative filtering in Lemma 2.2.
We explicitly write out how many times we access the database and how much privacy is lost each time in an interactive version of DPfilter in Algorithm 1, which performs the same operations as DPfilter. In order to apply Lemma G.13, we cap at 0.9 in initializing . We call , , and times, each with guarantee. In total this accounts for privacy loss, using Lemma G.13 and our choice of and .
The following theorem analyzing DPfilter implies the desired Lemma D.3 when the good set is -subgaussian good, which follows from G.3 and the assumption that .
To prove this theorem, we use the following lemma to first show that we do not remove too many uncorrupted samples. The upper bound on the accuracy follows immediately from Lemma G.7 and the stopping criteria of the algorithm.
If , and , then there exists constant such that for each iteration , with probability , we have Eq. (4) holds. If this condition holds, we have
Now we bound the number of iterations under the conditions of Lemma D.5. Let . Since Eq. (5), we have
Let be the stopping time. We know . By Wald’s equation, we have
Assuming we have for some sufficiently large, it suffices to show
Lemma G.6 shows that the magnitude of the largest eigenvalue of is positive since the magnitudes negative eigenvalues are all less than . So we have
where the first inequality follows from Lemma D.6, and the second inequality follows from our choice of large constant . The next lemma regularity conditions for ’s for each iteration is satisfied.
If , then there exists a large constant such that, with probability , we have
Thus, by combining with Lemma D.5, we have
By our choice of sample complexity , with probability , we have , (Lemma D.6), and simultaneously hold before stopping.
We first consider the upper bound of the good points.
where the is implied by the fact that for any vector , we have , follows from Lemma G.7 and follows from our choice of large constant .
Since , we know , so we have for ,
where follows from Lemma G.6, and follows from Lemma G.7, and follows from our choice of large constant .
where the last inequality follows from Lemma G.6, which shows that the magnitude of the largest eigenvalue of must be positive. ∎
Appendix E PRIME: efficient algorithm for private and robust mean estimation
We provide our main algorithms, Algorithm 9 and Algorithm 10, in Appendix E.1 and the corresponding proof in Appendix F. We provide our novel DPthreshold and its anlysis in Appendix E.2.
We define as the original set of clean samples (as defined in Assumption 1 and 2) and as the set of corrupted samples that replace of the clean samples. The (rescaled) covariance is denoted by , where denotes the mean.
E.2 Algorithm and analysis of DPthreshold
Algorithm DPthreshold() running on a dataset is -DP. Define . If ’s satisfy
and , then DPthreshold outputs a threshold such that with probability ,
E.3 Proof of Lemma E.1
1. Threshold sufficiently reduces the total score.
Let be the threshold picked by the algorithm. Let denote the minimum value of the interval of the bin that belongs to. It holds that
2. Threshold removes more bad data points than good data points.
Define to be the threshold such that . Suppose , because , . Trivially due to the fact that . Then we have the threshold picked by the algorithm , which implies . Suppose , since , we have
where (a) holds by Lemma E.3, and (b) holds since . If , the statement of the Lemma E.3 directly implies Equation (13).
Since , it holds
Assuming that the conditions in Lemma E.2 holds, and for any such that
First we show an upper bound on :
Then we show an lower bound on :
Combing the lower bound and the upper bound yields the desired statement ∎
Appendix F The analysis of PRIME and the proof of Theorem 6
Let be the end-to-end target privacy guarantee. The ()-DP guarantee of follows from Lemma D.2. We are left to show that DPMMWfilter in Algorithm 10 satisfy -DP. To this end, we explicitly write out how many times we access the database and how much privacy is lost each time in an interactive version of DPMMWfilter in Algorithm 13, which performs the same operations as DPMMWfilter.
In order to apply Lemma G.13, we cap at 0.9 in initializing . We call and times, each with guarantee. In total this accounts for privacy loss. The rest of the mechanisms are called times ( and each call two DP mechanisms internally), each with guarantee. In total this accounts for privacy loss. Altogether, this is within the privacy budget of .
This is a powerful tool for designing private mechanisms, as it guarantees that we can safely simulate the filtering process with privatized parameters and preserve the neighborhood of the dataset; if are neighboring (i.e., ) then so are the filtered pair and (i.e., ). Note that in all the interactive mechanisms in Algorithm 12, the noise we need to add is proportional to the set sensitivity of Filter defined as . If the repeated application of the Filter is not a contraction in , this results in a sensitivity blow-up. Fortunately, the above lemma ensures contraction of the filtering, proving that . Hence, it is sufficient for us to prove privacy for two neighboring filtered sets (as opposed to proving privacy for two neighboring original datasets before filtering ).
In , satisfy -DP as the sensitivity is (Definition 1.2) and we add . The release of also satisfy -DP as the sensitivity is , assuming as ensured by the stopping criteria, and we add . Note that in the outer loop call of , we only release once in the end, and hence we count as one access. On the other hand, in the inner loop, we use both and from so we count it as two accesses.
In , is -DP as the sensitivity is , and we add . is -DP as the sensitivity is and we add . This is made formal in the following theorem with a proof. in Appendix F.1.1. This algorithm is identical to the MOD-SULQ algorithm introduced in and analyzed in [18, Theorem 5], up to the choice of the noise variance. But a tighter analysis improves over the MOD-SULQ analysis from by a factor of in the variance of added Gaussian noise as noted in .
with for and for .
In , the differential privacy follows from that of DPthreshold proved in Lemma E.1.
For any fixed unit vector , we have
where is CDF of standard Gaussian. According to Gaussian mechanism, if , we have .
F.2 Proof of part 2 of Theorem 6 on accuracy
The accuracy of PRIME follows from the fact that returns a hypercube that contains all the clean data with high probability (Lemma D.2) and that DPMMWfilter achieves the desired accuracy (Theorem 11) if the original uncorrupted dataset is -subgaussian good. is -subgaussian good if we have as shown in Lemma G.3. We present the proof of Theorem 11 below.
Moreover, each epoch runs for at most iterations.
In epochs, following Lemma F.3 guarantees that we find a candidate set of samples with . We provide proof of Lemma F.3 in the Appendix F.3.
F.3 Proof of Lemma F.3
Lemma F.3 is a combination of Lemma F.4 and Lemma F.5. We state the technical lemmas and subsequently provide the proofs.
For an epoch and for all if Lemma F.4 holds, , and , then we have with probability .
To prove that we make progress for each iteration, we first show our dataset satisfies regularity conditions in Eqs. (14) and (15) that we need for DPthreshold. Following Lemma F.6 implies with probability , our scores satisfies the regularity conditions needed in Lemma E.1.
For each epoch and iteration , under the hypotheses of Lemma F.4, with probability , we have
where .
Then by Lemma E.1 our DPthreshold gives us a threshold such that
Conditioned on the hypotheses and the claims of Lemma E.1, according to our filter rule from Algorithm 10, we have
where follows from our assumption on and stopping criteria. Rearranging the terms completes the proof. ∎
First of all, Lemma G.9, Lemma G.10 and Lemma G.11 gives us following Lemma F.7, which basically shows with enough samples, we can make sure the noises added for privacy guarantees are small enough with probability .
For , if and then we have with probability , following conditions simultaneously hold:
Now under above conditions, since , we have . Using the fact that , we also have
Thus, from the first and the second claims in Lemma F.7, we have
For an epoch and an iteration , since , we have
where follows from the fact that for any vector , we have , follows from Lemma G.4, follows from Lemma G.7, follows from our choice of large constant , and in the last inequality we used Eq. (16).
where follows from Lemma G.4, follows from Lemma G.5 and Lemma G.7 and follows from our choice of large constant .
Under the conditions of Lemma F.7, we have picked large enough such that with probability , we have
Since , we have . Combining the above inequality and the fifth claim of Lemma F.7 together, we have
By Lemma G.1, we have . By our choice of , we have and . Therefore, by Lemma G.14, we have
where follows from our choice of and . By Lemma G.6, for , we have
By Lemma G.6, we have . Also, we know . Then we have
where the last inequality follows from our assumption that , and conditions of Lemma F.7 hold and we have . ∎
Appendix G Technical lemmas
If , then .
and .
for any subset so that , we have
A set of i.i.d. samples from an identity covariance sub-Gaussian distribution of size is -subgaussian good with respect to with probability .
For any subset such that , we have
For any subset such that , we have
G.2 Auxiliary Lemmas on Laplace and Gaussian mechanism
Let be arbitrary. For , the Gaussian Mechanism with parameter is -differentially private.
For , an end-to-end guarantee of -differential privacy is satisfied if a dataset is accessed times, each with a -differential private mechanism.
G.3 Analysis of ‖M(St(s))−𝐈‖2\|M(S_{t}^{(s)})-\mathbf{I}\|_{2} shrinking
For any symmetric matrix , we let denote .
and satisfies for all , then for all , , it holds that
Rearranging terms, and taking a supremum over , we obtain that
Appendix H Exponential time DP robust mean estimation of sub-Gaussian and heavy tailed distributions (Algorithm 2)
In this section, we give a self-contained proof of the privacy and utility of our exponential time robust mean estimation algorithm for sub-Gaussian and heavy tailed distributions. The proof relies on the resilience property of the uncorrupted data as shown in the following lemmas.
for all sets of size at least .
Let be a set of i.i.d. points from a sub-Gaussian distribution with a parameter . Given that , is -resilient around its mean with probability .
Let be a set of i.i.d. samples drawn from distribution whose mean and covariance are respectively, and that . Given that , there exists a constant that only depends on such that is -resilient around with probability .
Lemma K.1 ensures that returns samples in a bounded support of Euclidean distance with where samples are uncorrupted ( is corrupted by adversary and can be corrupted by the pre-processing step). For a -resilient dataset, we first show that is robust against corruption.
Let be the set of -corrupted data. Given that , with probability , .
This follows immediately by selecting to be the uncorrupted fraction of the dataset and applying ()-resilience. After pre-processing, we have that , and then clearly has sensitivity .
Given that , is -differentially private. Further, with probability , .
In the algorithm, we first compute . If , we stop and output . Otherwise, we use exponential mechanism with score function to find an estimate . We prove the privacy guarantee of our algorithm as follows.
Algorithm 2 is -differentially private if .
We consider neighboring datasets , under the following two scenario
Given that , for any neighboring dataset , .
For an -corrupted dataset , Algorithm 2 achieves with probability , if .
We use the following lemma showing that is a good approximation of .
Let be the set of -corrupted data. Given that , with probability ,
This implies that the exponential mechanism achieves the following bounds.
where denotes the normalizing factor for the exponential mechanism and is the volume of a ball of radius in dimensions. It follows that
for .
Since , define as the minimizing subset in Definition 4.2 such that
By this definition of and Lemma H.1,
This implies the sensitivity of is bounded by :
First we show . Notice that , and . By the ()-resilience property, we have , and . Since , by the -resilience property,
Since , are the largest and smallest points respectively and , we get
Combining and we get
where holds by the definition of the distance :
H.2 Case of sub-Gaussian distributions and a proof of Theorem 7
Th proof is analogous to the previous section, we only state the lemmas that differ. returns a hypercube that includes all uncorrupted data points with a high probability.
Let be the set of -corrupted data. Given that , with probability , .
Algorithm 2 is -differentially private if .
Given that , for any neighboring dataset , .
Let be the set of -corrupted data. Given that , with probability ,
This implies the following utility bound.
For an -corrupted dataset , Algorithm 2 achieves with probability , if .
Appendix I Background on exponential time approaches for Gaussian distributions
In this section, we provide a background on exponential time algorithms that achieve optimal guarantees but only applies to and heavily relies on the assumption that samples are drawn from a Gaussian distribution. In §4, we introduce a novel exponential time approach that seamlessly generalizes to both sub-Gaussian and covariance-bounded distributions.
We introduce Algorithm 14, achieving the optimal sample complexity of (Theorem 12). The main idea is to find an approximate Tukey median (which is known to be a robust estimate of the mean ), using the exponential mechanism of to preserve privacy.
For a dataset of i.i.d. samples from a -dimensional Gaussian distribution , an adversary corrupts an fraction of the samples as defined in Assumption 1. Then, any in the Tukey median set of a corrupted dataset satisfies with probability at least if .
where is the sensitivity of (from Definition 1.2) and ensures normalization to one. This mechanism is -differentially private, since and .
The sampled from the distribution (19) is -differentially private.
Under the hypotheses of Corollary I.1, there exists a universal constant such that if , and , then Algorithm 14 is -differentially private and achieves with probability .
Appendix J Proof of Theorem 12 on the accuracy of the exponential mechanism for Tukey median
First, the -differential privacy guarantee of private Tukey median follows as a corollary of Proposition I.2, by noting that sensitivity of is one, where is a dataset of size . This follows from the fact that for any fixed and , is the number of samples on one side of the hyperplane, which can change at most by one if we change one sample in .
Note that this is the standard definition of Tukey depth. First we show that for large enough, the Tueky depth for the empirical distribution is close to that of the true distribution. We provide proofs of the following lemmas later in this section.
The proof of Lemma J.1can be found in §J.1. This allows us to use the known Tukey depths of a Gaussian distribution to bound the Tukey depths of the corrupted empirical one. We use this to show that there is a strict separation between the Tueky depth of a point in and a point in . The proof of Lemma J.2 can be found in §J.2.
Define , and assume . Given that , with probability ,
This implies that most of the probability mass of the exponential mechanism is concentrated inside a ball of radius around the true mean . Hence, with high probability, the exponential mechanism outputs an approximate mean that is close to the true one. The following lemma finishes the proof the the desired claim, whose proof can be found in §J.3.
by letting . We conclude the proof since
J.2 Proof of Lemma J.2
Then Lemma J.1 implies that with probability
where (a) holds since , and it is easy to verify that (b) holds for . The second claim holds since
where holds by setting .
J.3 Proof of Lemma J.3
using the fact that and that , and
where is an absolute constant. If we set , we get that
which implies that with probability at least , .
Appendix K The algorithmic details and the analysis of PRIME-ht for covariance bounded distributions
We provide the algorithm and the analysis for the range estimation query , and then prove the result on analyzing PRIME-ht.
is -differentially private. Under Assumption 2 and for , if , returns a ball of radius centered at that includes uncorrupted samples where with probability .
We first show that applying the private histogram to each coordinate provides a robust estimate of the range, but with a constant probability 0.9.
Under the -corruption model of Assumption 2, if , for , in Algorithm 5 with a choice of and returns intervals of size such that with probability 0.9 for each .
Following [54, Algorithm 2], we partition the dataset into subsets of an equal size and apply the median-of-means approach. Applying Lemma K.2, it is ensured (e.g., by [54, Lemma A.4]) that more than half of the partitions satisfy that the center of the interval is within 240 away from , with probability . Therefore the median of those centers is within from the true mean in each coordinate. This requires the total sample size larger only by a factor of .
To choose a radius ball around this estimated mean that includes fraction of the points, we choose . Since for , this implies that we can choose -ball around the estimated mean with .
K.2 Proof of Theorem 9
The proof of the privacy guarantee of Algorithm 16 follows analogously from the proof of the privacy of PRIME and is omitted here. The accuracy guarantee follows form the following theorem and Lemma K.1.
Moreover, each epoch runs for at most iterations.
Algorithm 16 is a similar matrix multiplicative weights based filter algorithm for distributions with bounded covariance. Similarly, we first state following Lemma K.3 and prove Theorem 13 given Lemma K.3
Lemma K.3 is a combination of Lemma K.4, Lemma K.5 and Lemma K.6. We state the technical lemmas and subsequently provide the proofs.
For each epoch and iteration , under the hypotheses of Lemma K.3 then with probability , we have
where .
For epoch , suppose for where , if Lemma K.5 holds, , and , then we have with probability .
By Lemma G.9, Lemma G.10 and Lemma G.11, we can pick such that with probability , following conditions simultaneously hold:
.
where follows from the fact that for any vector , we have , follows from -goodness of , follows from Lemma K.11 and follows from our choice of large constant and sample complexity .
Lemma K.4 implies with probability , our scores satisfies the condition in Eq. (20). Then by Lemma K.7 our DPthreshold-ht gives us a threshold such that
According to our filter rule from Algorithm 17, we have
At the same time, Lemma K.7 gives us a such that with probability , we have
where follows from our assumption that .
We pick large enough such that with probability ,
By Lemma G.1, we have . by our choice of , we have and . Therefore, by Lemma G.14 we have
where follows from our choice of , , and .
Algorithm DPthreshold-ht() running on a dataset is -DP. Define . If ’s satisfy
Let be the threshold picked by the algorithm. Let denote the minimum value of the interval of the bin that belongs to. It holds that
Define to be the threshold such that . Suppose , we have because , . Then the threshold picked by the algorithm , which implies . Suppose , since
where (a) holds by Lemma K.8, and (b) holds since . If , the statement of the Lemma K.8 directly implies Equation (21).
Assuming that the condition in Eq.(20) holds, then for any such that
First we show an upper bound on :
Then we show an lower bound on :
Combing the lower bound and the upper bound yields the desired statement ∎
.
Let be an -corrupted bounded covariance dataset under Assumption 2. If is -good with respect to , then for any such that , we have
Appendix L Experiments
We evaluate PRIME and compare with a DP mean estimator of on synthetic dataset in Figure 1 and Figure 2, which consists of samples from . The main focus of this evaluation is to compare the estimation error and demonstrate the robustness of PRIME under differential privacy guarantees. Our choice of experimental settings and hyper parameters are as follows: , , , , .
Figure 2 shows additional experiments including the regime where we do not have enough number of samples. When , the utility guarantee (Theorem 5) does not hold. The noise we add on the final output becomes large as decreases and dominates the estimation error. The DP Mean has lower error compared to PRIME when is small because PRIME spends some privacy budget to perform operations other than those in DP Mean in the Algorithm 10. In practice, we can check whether there are enough number of samples based on known parameters , and choose to use DP Mean (or adjust how the privacy budget is distributed in PRIME).
Our implementation is based on Python with basic Numpy library. We run on a 2018 Macbook Pro machine. For each choice of in our settings, it takes less than minutes and PRIME stops after at most epochs. We have attached our code as supplementary materials.