Differentially Private Model Selection with Penalized and Constrained Likelihood
Jing Lei, Anne-Sophie Charest, Aleksandra Slavkovic, Adam Smith, Stephen Fienberg
Introduction
In data privacy research, the goal of data analysis is to provide accurate and useful statistical inference while preventing individual records being identified. Such privacy protection is crucial in many statistical applications, namely for the analysis of census and survey data, medical and clinical studies, genetics data, and web user data collected on the internet. In statistics, the treatment of confidential data has a long history under the name of “Statistical Disclosure Control” or “Statistical Disclosure Limitation”; see for example Dalenius 1977, Rubin 1993, Willenborg & De Waal 1996, Fienberg & Slavković 2010 and Hundepool et al. 2012. A long lasting challenge is to rigorously quantify the disclosure protection offered by privacy-preserving data analysis procedures.
The notion of differential privacy has been introduced with the same objective in theoretical computer science by Dwork 2006 and Dwork et al. 2006. The general idea of differential privacy is to require for the outcome of a randomized data analysis procedure not to change much for small perturbations of the input data, so that one can not infer from the output the presence or absence of any individual in the input data set, or infer some of the person’s characteristics. This requirement can be rigorously quantified, and does not depend on assumptions regarding the resources of the intruder, including any access to auxiliary information. Thus, differential privacy compares very favorably to measures of disclosure risk commonly used in the statistical disclosure control literature as it is more encompassing and a worst case definition, but at the same time it has been criticized as too stringent from the perspective of achieving needed statistical data utility; e.g., see Fienberg et al. 2010; Karwa & Slavkovic 2012.
Over the last decade, there has been a rapid development of differentially private algorithms and procedures in both the computer science and statistics literature. For example, the main focus in the computer science literature was on designing differentially private mechanisms and efficient algorithms for private data release, e.g., the Laplace noise perturbation mechanism (Dwork et al. 2006), the exponential mechanism (McSherry & Talwar 2007), releasing contingency tables (Barak et al. 2007), and boosting Hardt et al. 2010. On the statistics side, research efforts include designing consistent and efficient differentially private point estimators (Dwork & Lei 2009; Smith 2011; Chaudhuri et al. 2011; Lei 2011; Bassily et al. 2014; Karwa & Slavković 2016), non-parametric density estimation (Wasserman & Zhou 2010), hypothesis testing (Fienberg et al. 2011; Johnson & Shmatikov 2013; Uhler et al. 2013; Yu et al. 2014; Karwa et al. 2014; Solea 2014; Dwork et al. 2015; Sheffet 2015; Wang et al. 2015; Gaboardi et al. 2016), and statistical lower bounds (Chaudhuri & Hsu 2011; Duchi et al. 2013); there is also a large literature on private PAC learning, which echoes the concerns of statistical estimation in the context of classification (Kasiviswanathan et al. 2011; Beimel et al. 2010; Beimel et al. 2014; Beimel et al. 2015; Bun et al. 2015; Karwa et al. 2015).
In this paper we consider statistical model selection under the constraint of differential privacy. Despite the fast development in combining statistical theory and methodology with differential privacy, model selection under privacy constraints has not been well understood. The problem of differentially private model selection is motivated by practical concerns: when private data analysis procedures are needed, it is rarely known which model is most appropriate for the data. When releasing the whole dataset is impractical due to privacy concerns and releasing point estimates for pre-specified models has limited utility, an appropriate compromise is to first identify the best model and then obtain and release consistent point estimates for this model. Both tasks need to be performed under privacy constraints, and thus we need methods for differentially private model selection.
In particular, we focus on the classical linear regression model selection. In particular, we aim to provide insights for the following two questions. (i) Is it theoretically possible to do model selection with differential privacy under the classical conditions? (ii) What new practical concerns arise in model selection when differential privacy is required?
The remainder of the paper is organized as follows: In section 2, we review the definition and interpretation of differential privacy, as well as two general methods to create differentially private algorithms. Section 3 details the two proposed differentially private model selection procedures, with proofs of their privacy guarantee, and notes on the choice of the tuning parameters. Statements and proofs of the utility guarantee of the two algorithms are given in section 4. Empirical results, including a simulation study and two real data examples are reported in section 5. Section 6 provides a brief discussion. All proofs and technical details are collected in section 7.
Differential privacy
Differential privacy requires that the output of a procedure is not drastically altered under small perturbations of the input data set, such that an attacker, regardless of his auxiliary information and computing power, can hardly recover the presence/absence of a particular individual in the data set. The notion of differential privacy is a property of that data analysis procedure, rather than of the output obtained.
To formalize, consider a data set consisting of data points in sample space . A data analysis procedure , possibly randomized, maps the data set , together with a random input , to , an output space. Here we assume that is a measurable space and is a measurable function. Whenever it is not confusing, we will use to denote the random variable .
We can now state formally the property of differential privacy:
Given a privacy parameter, , the procedure satisfies -differential privacy if
where we define for convenience.
In the above notation, denotes the probability with respect to , which is the source of randomness in the data analysis procedure. Thus, the definition does not impose any conditions on the distribution of — the privacy is required to hold for all pairs of adjacent data sets. This stringent condition means that procedures which satisfy definition 1 provide very strong privacy guarantees, even against adversaries who have considerable partial information about the data set (Ganta et al. 2008). In order to satisfy this definition, any nonconstant procedure must be randomized.
The -differential privacy is a strong requirement, as it takes supremum over all possible neighboring data sets of size . A mild relaxation is the -differential privacy.
Given , , a procedure satisfies -differential privacy if, for all measurable and all neighboring data sets , ,
Here the requirement of the original -differential privacy is relaxed so that the distribution of only needs to be dominated by that of outside of a set with probability no more than .
In our discussion of statistical applications we will generally focus on data sets consisting of a sequence of independent random samples from an underlying distribution, and the corresponding probability will be denoted . Note that differential privacy definition has no such assumption. We will use to denote the overall randomness due to both the data and random mechanism in the analysis.
2 Statistical interpretation of differential privacy
Privacy protection due to differential privacy can be interpreted from a Bayesian perspective, as in Abowd & Vilhuber 2008; Kasiviswanathan & Smith 2008. Suppose one has a prior distribution of the input data set and then gets to observe the random output . Denote by and the marginal prior and posterior of , for the th entry in . Then if the procedure is differentially-private and the prior is a product measure on the entries of , we have that , thus limiting the information regarding gained from the output . A related hypothesis testing interpretation is given in Theorem 2.4 of Wasserman & Zhou 2010.
One may also intuitively interpret differential privacy as a specific notion of robustness. It requires that the distribution of is not changed too much if is perturbed in only one entry. However, the definition of differential privacy is also a worst case definition, where the probability ratio needs to be uniformly bounded over all pairs of adjacent input data sets. A key ingredient to designing differentially private statistical procedures is to bridge the gap between worst case privacy guarantee and the average case statistical utility. It turns out that robustness and regularization are the most relevant structures to explore, as in Dwork & Lei 2009, Chaudhuri et al. 2011 and Smith & Thakurta 2013.
3 Designing differentially private algorithms
For any statistical task which we want to carry, a specific randomized procedure must be designed to take as input a database and return an element of the output space while satisfying differential privacy. There exists a few approaches which are generic enough to be adaptable to various tasks, and which are often used as building blocks for more complicated procedures. We present two of these methods, which we use later in the construction of our procedure. The first adds random noise to a non-private output and the second samples randomly from a set of candidate outputs.
If , then it is easy to check (Dwork et al. 2006) that
satisfies -differential privacy, where is a standard double exponential random variable with density function (also known as the Laplace distribution).
Then satisfies -differential privacy.
When is finite, one may also use additive noise to approximately maximize over . Let
be the privatized score, where is an independent draw from the standard double exponential distribution. Then
satisfies -differential privacy and usually offers similar performance as the exponential mechanism.
In many statistical problems the sample space is not compact and hence for many statistics such as the sample mean. A general strategy is to show that one can add much less noise for most average case data sets with (, )-differential privacy (Dwork & Lei 2009). These methods often involves the notion of local sensitivity of a deterministic procedure :
If is finite and public, then one can show that adding noise to as in (2) with replaced by also gives -differential privacy. Unfortunately, depends on the data set and may contain sensitive information. However, there is a generic scheme based on this idea with valid privacy guarantee under the following two general conditions on the deterministic procedure .
For all there exists a real-valued function and a randomized procedure , which is -differentially private if and is assumed to be non-private.
Given and , there exists an -differentially private mapping satisfying for all .
An example of is the local sensitivity of some procedure , and is the noisy version as in (2) calibrated to a upper bound of the local sensitivity.
Under the above two assumptions, for any , and , satisfies -differential privacy.
Differentially Private Model Selection Procedures
Popular methods of model selection for linear regression include information criteria such as AIC (Akaike 1974) and BIC (Schwarz et al. 1978), cross-validation (Picard & Cook 1984), and the more recent penalized least squares (Tibshirani 1996; Fan & Li 2001, e.g.,). In this paper we focus on penalized least squares and penalized profile likelihood estimators, both being variants of the classical approach based on information criteria.
Given a parameter and an observed data set, the log likelihood is, ignoring constant terms,
When is known, we aim to find the that maximizes the log likelihood, which leads to model selection with the least squares. Without loss of generality, we assume . We then define
In the more realistic setting that is unknown, we can maximize over the nuisance parameter to perform model selection with the profile likelihood. Ignoring constant terms, we obtain the profile log-likelihood for :
Maximizing this over all in , we get
where is the amount of penalty on the model complexity. The best known and most widely used examples are the AIC (), and BIC ().
In the case of known , the penalized minimization becomes
It is worth noting that the penalized least square estimator, with appropriate choice of , can also be used in the case of unknown .
2 Towards Differentially Private Model Selection
To perform model selection in a differentially-private manner, we propose to apply the exponential mechanism or noisy minimization to the minimization problems (10) and (11), with (here in the general definition is replaced by ) given by the penalized likelihood
in the case of penalized log profile likelihood (10), or
in the case of penalized least squares (11).
A key step in this approach is to evaluate and control the sensitivities of and as functions of for any .
To simplify the notation and concentrate on the main idea, we will assume in the sequel that the data entries are bounded or standardized:
where is a known number that can grow with . Boundedness is typically required for differentially private data analysis. Standard methods for finding the range of the data set in a privacy-preserving manner include those given in Dwork & Lei 2009 and Smith 2011.
To bound the sensitivity of either of them, we must bound the possible parameter vectors we consider. We thus consider constrained versions of the two score functions. Given , we define
The sensitivity of the least squares loss is now easy to bound:
The proof is both short and elementary, and hence omitted.
In the case of the constrained profile likelihood, we can not bound the global sensitivity, but we can find a bound on the local sensitivity:
2.2 Algorithms
Lemma 3.1 implies that the penalized constrained lease squares
can easily be minimized in an -differentially private manner. The complete algorithm is given below.
Next, Lemma 3.2 gives an upper bound of the local sensitivity of . Now we apply the generic scheme of designing -differentially private algorithms described in section 2.3.
where is a standard Laplace random variable.
Then we can construct the final estimators using either exponential mechanism or the noisy minimization, both satisfying -differential privacy. The complete algorithm is given in Algorithm 2.
3 Choosing the tuning parameters
However, our numerical experiments in section 5 show that the performance of our proposed algorithms is sensitive to the choice of these tuning parameters. We acknowledge that fully data-driven and privacy-preserving methods for choosing these tuning parameters remain a challenging and important open problem, and is beyond the scope of this paper. Data-driven choice of penalty parameter is a hard problem even without privacy constraint. Here we provide some heuristics on potential solutions.
Utility Analysis
In privacy-preserving data analysis, the privacy shall be protected for any possible input data set. In other words, the privacy guarantee needs to cover the worst case and must be established with no distributional conditions on the data set. In the previous sections, our differentially private procedure only requires the data to be bounded, which can be verified or enforced easily in practice. On the other hand, the statistical utility (for example, consistency, rate of convergence) is usually based on common statistical assumptions on the data. To facilitate discussion, we first introduce some notation and assumptions.
Let be the design matrix, and the vector of ’s. For any , let be the design matrix consisting of the columns in . Then is a vector that is on the entries in and zero elsewhere. It is the ordinary least square estimate under model . The sample covariance is .
We state below five assumptions required for our utility results, and discuss their practical significance.
Our first assumption is a sparse linear model with Gaussian noise.
(Linear model with Gaussian noise) The data entries are generated from the linear regression model (4) with a regression coefficient vector with and . The noise are iid Gaussian with mean zero and variance .
Next we assume that the number of candidate models grows polynomially with sample size . This is usually the case when we only search over sparse models and the total number of variables grows polynomially in .
(Candidate models) The set of candidate models contains the true model , and has cardinality no more than for some positive number which is allowed to grow with . The largest candidate model has variables.
The worst-case sensitivity of the log likelihood is hard to control because the design matrix may be poorly conditioned. Thus we need to add some singular value condition on the design matrix.
(Design matrix) The design matrix is fixed, and the sample covariance satisfies the sparse eigenvalue condition:
Assumption A2 excludes the situation of linear dependence between columns of and . A similar condition has been considered in the literature on model selection consistency using information criteria (Nishii 1984). The condition stated here is for fixed design matrix , but it holds with high probability for many random designs. Assumption A2 also implies an upper bound of the range of . Such a boundedness property will help control the sensitivity of the log likelihood.
Moreover, as we did in section 3, we assume boundedness of the data entries, which are typically needed for developing differentially private procedures.
(Boundedness) Each entry of is bounded by , and each entry of is bounded by .
Finally, we assume that the sample size is large enough, when compared to other quantities in the analysis such as the noise variance, and the inverse of privacy parameter .
(Sample size) The sample size is large enough so that equations (18), (19), and (20) hold.
1 Utility of Penalized Constrained Least Squares
We first give a utility result for the noisy penalized constrained least squares estimator (Algorithm 1).
Assume A0-A4 hold. If satisfies
where for some , and
then the PCLS estimator given by Algorithm 1 with privacy parameter and penalty parameter satisfies
Recall that stands for taking probability over both the randomness in and in the generation of Laplace random variables (denoted by ) in the algorithm. Here we are assuming a fixed design matrix, so the randomness in is equivalent to the randomness of the additive noise .
2 Utility of Penalized Constrained Profile Likelihood
Now we provide utility result for the noisy penalized profile likelihood estimator.
Assume A0-A4 hold. If satisfies
where for some , and
then for any constant and large enough as quantified in Equations 18, 19 and 20, the selected model given by the noisy penalized constrained profile likelihood (Algorithm 2) satisfies
Empirical Results
We have shown in the last section that the two proposed differentially-private model selection procedures are consistent under similar conditions as the corresponding non-private algorithms. We now provide some empirical results for model selection via penalized constrained least squares, to illustrate both the utility of the algorithms and impact of tuning parameter selection as discussed in section 3.3.
In the simulations, we consider a small sample size () and a moderately large sample size (), as well as , and . For each set of parameters, we sample data sets from the true model, apply Algorithm 1 to each of them, and note the proportion of times we correctly identify the correct model. We set to be the maximum observed value of in the data set.
Figure 2 shows the results for case where . The effect of , , and are very similar in this case, but the choice of and are more important to achieve good utility. The sensitivity of the procedure to choices of and thus depends on the structure of the true parameter . Note however that for a proper choice for the parameters leads to completely accurate model selection with , and even very accurate for with .
2 Application to Real Data Sets
We first illustrate the results of model selection via penalized constrained least squares on a small data set of observations, then on a much larger one with hundreds of thousands observations.
The prostate data set contains several clinical measures for men with prostate cancer. In this paper, our goal is to predict the level of a prostate specific antigen using five continuous variables: the volume of the cancer, the weight of the prostate, the age of the patient, the capsular penetration and the benign prostatic hyperplasia amount. Except for age, all variables are taken on the scale. We also rescale all of the variables to take values between and , which could be done in a differentially private way.
We consider all possible main effects models for the model selection procedure, for a total of models to choose from. Model selection based on the penalized maximum likelihood, without the constraint of differential privacy, selects the model with only two variables in addition to the intercept: the volume of the cancer, and the weight of the prostate. Model selection is then performed using Algorithm 1 for various choices of , and . We set , obtained with the non-private algorithm, and we set to the maximum value of .
Figure 3 shows the proportion of times that the private and non-private procedures identify the correct model, for independent replications. With such a small data set, the choice of and are crucial: even without differential privacy, using too large a value of does not identify the correct model. Due to the small sample size, the utility of the private procedure also decreases quite rapidly with decreasing which should offer more privacy.
Since the constrained optimization can be implemented based only on sufficient statistics, the private procedure scales very well for much larger data sets. The housing data set contains several variables measured on houses sold in San Francisco Bay Area between 2003 and 2006. In addition to the price of the sale, the data include the year of the transaction (an ordinal variable with 4 levels), the latitude and longitude of the house, the county in which it is located (a categorical variable with 9 levels), and a continuous measure of its size. We preprocess the data to remove houses with price outside the range to thousand dollars, and size larger than sqft. We also combine some of the small counties into a new indicator variable. All predictors are also scaled so that they take values in $235,76013$ variables.
We consider again models with main effects only, for a total models to choose the best model from. Since the optimal model is not as clear as with the previous example, we do not directly compare the private procedure with a gold standard, but rather with its non-private counterpart. An algorithm which offers differential privacy but recovers the results of a non-private algorithm is successful. As above, we apply the model selection procedures with various values of , and , and use the maximum value of as .
Figure 4 shows the agreement between the private and non-private procedure for various tuning parameters. With this large data set, the results are very positive: even for we recover the same model with the differentially-private procedure as without the privacy requirement for chosen large enough. Note also that, as expected, the scale of is much larger than for the smaller Housing example.
Discussion
With modern data acquisition and storage techniques allowing the collection and analysis of huge amounts of personal information in a multitude of formats, protecting individual privacy inevitably becomes of crucial concern for modern data analysis. Although differential privacy offers mathematically strong and elegant privacy guarantee, the nature of such a conservative constraint in the context of everyday statistical analysis remains unclear. Previous works on statistical analysis with differential privacy mainly focused on simple statistical queries, such as location and scale statistics, regression coefficients, and simple hypothesis testing, and network analysis. In this paper we considered the more challenging problem of model selection in the classical setting. We showed that standard techniques for differentially private data analysis can be combined with known statistical tools such as penalized least square or information criteria to construct privacy-preserving model selection procedures with strong utility. We proposed two algorithms for this task, and proved privacy and utility results for each of them.
Our procedures feature a double-regularization, as they include both constrained estimation in the fitting step, and penalization in the model comparison step. Thus the method involves two tuning parameters. A key observation, illustrated in section 5, is that although we have proven good large-sample properties for a wide range of penalty parameter and , the practical performance is sensitive to the particular choices for these parameters. In other words, these tuning parameters play a very important and unique role in designing differentially private procedure with good practical performance. While in low-dimensional settings, the regularization parameter is not needed for statistical inference without privacy constraints, when privacy is a concern, a carefully chosen amount of regularization can lead to stable, low-sensitivity estimators even in the worst case. Appropriate choice of both tuning parameters thus reflects the need for some additional information in the data that will be useful for differentially private procedures, but not necessary in traditional inference methods. It will be an interesting future topic to give a general characterization of such privacy related quantities, and to develop differentially private methods to estimate these quantities.
Appendix: Proof details
Given a candidate model , the log likelihood is, ignoring constant terms,
Let be the constrained least square estimator with input data set . By boundedness of , and we have for all such that . Then using the fact that for , we have
Next we prove our main utility theorems. The proofs of Theorem 4.1 and Theorem 4.2 rely on the following result, which is a consequence of simple linear algebra.
We prove for noisy minimization. The argument can be adapted simply to cover the exponential mechanism.
Then according to Proposition 7.1, we can work directly with the unconstrained version.
For , using Claim 2 in Section 7 we have with probability at least . Thus with same probability we have
For , we have by Claim 3 in Section 7, with probability at least ,
Thus conditioning on , which has probability at least , we have
Consider events – as defined in Section 7. We focus on the event , which has probability at least .
In the case of , applying the fact that for to
where the last step follows from the fact that we are not in the event defined in Section 7.
On the event considered, we also have by Claim 5. Then we can bound the error probability by
2 Further proof details
Here we give details and summarize the multiple “with high probability” statements in the utility analysis. Recall that
We define the following events and give the corresponding probability bounds.
Claim 2.
Because , is a projection operator of dimension . Using a tail probability bound for Gaussian random variables, we have, for all ,
The desired result follows from union bound. ∎
For , let , , . Define
Denote the vector that agrees with on and 0 elsewhere.
where is the linear subspace spanned by .
Let . Then the second term in the above equation is distributed as . By the eigenvalue condition, we have , where is the minimum sparse eigenvalue and is a lower bound of the signal level. To see this, observe that and
where the first inequality uses the sparse eigenvalue condition.
Note that the dimension of is at most . Let . Using (18) we know that . Then with probability at least , we have
where the last inequality uses eq. (18). The claim follows from union bound. ∎
Claim 4.
Claim 5. . On , we have, using eq. (18)–(20),