Scaled Sparse Linear Regression
Tingni Sun, Cun-Hui Zhang
Introduction
This paper concerns the simultaneous estimation of the regression coefficients and noise level in a high-dimensional linear model. High-dimensional data analysis is a topic of great current interest due to the growth of applications where the number of unknowns far exceeds the number of data points. Among statistical models arising from such applications, linear regression is one of the best understood. Penalization, convex minimization and thresholding methods have been proposed, tested with real and simulated data, and proved to control errors in prediction, estimation and variable selection under various sets of regularity conditions. These methods typically require an appropriate penalty or threshold level. A larger penalty level may lead to a simple model with large bias, while a smaller penalty level may lead to a complex noisy model due to overfitting. Scale-invariance considerations and existing theory suggest that the penalty level should be proportional to the noise level of the regression model. In the absence of knowledge of the latter level, cross-validation is commonly used to determine the former. However, cross-validation is computationally costly and theoretically poorly understood, especially for the purpose of variable selection and the estimation of regression coefficients. The penalty level selected by cross-validation is called the prediction-oracle in Meinshausen & Bühlmann 2006, who gave an example to show that the prediction-oracle solution does not lead to consistent model selection for the lasso.
Estimation of the noise level in high-dimensional regression is interesting in its own right. Examples include quality control in manufacturing and risk management in finance.
An iterative algorithm
where is a vector of regression coefficients. Let the penalty be standardized to , where . A vector is a critical point of the penalized loss (1) if and only if
If the penalized loss (1) is convex in , then (2) is the Karush–Kuhn–Tucker condition for its minimization.
Given a penalty function , one still has to choose a penalty level to arrive at a solution of (2). Such a choice may depend on the purpose of estimation, since variable selection may require a larger than does prediction. However, scale-invariance considerations and theoretical results suggest using a penalty level proportional to the noise level . This motivates a scaled penalized least squares estimator as a numerical equilibrium in the following iterative algorithm:
The first step of our implementation is the computation of a solution path of (2) beginning from for . For quadratic spline penalties with knots, Zhang 2010 developed an algorithm to compute a linear spline path of solutions of (2) to cover the entire range of . This extends the least angle regression solution or lasso path (Osborne et al. 2000a; Osborne et al. 2000b; Efron et al. 2004) from and includes the minimax concave penalty for and the smoothly clipped absolute deviation penalty (Fan & Li 2001) for . An R package named plus is available for computing the solution paths for these penalties.
The second step of our implementation is the iteration (3) along the solution path computed in the first step. That is to use the already computed
in (3). For the scaled lasso, we use in (3) and in (1) and (2). For the scaled minimax concave penalized selection, we use and the minimax concave penalty , where regularizes the maximum concavity of the penalty. When , it becomes the scaled lasso. The algorithm (3) can be easily implemented once a solution path is computed.
Antoniadis 2010 suggested this jointly convex loss function as a way of extending Huber’s robust regression method to high dimensions. For and with fixed , , so that in (4) minimizes over . For fixed , in (3) minimizes over . During the revision of this paper, we learned that She & Owen 2011 have considered penalizing Huber’s concomitant loss function for outlier detection in linear regression. We summarize some properties of the algorithm (3) with (4) in the following proposition.
Let be a solution path of (2) with . The penalized loss function (5) is jointly convex in and the algorithm (3) with (4) converges to
The resulting estimators and are scale equivariant in in the sense that and . Moreover,
Since (5) is not strictly convex, the joint estimator may not be unique for some data . However, since (5) is strictly convex in , is always unique in (6) and the uniqueness of follows from that of the lasso estimator at ; is unique when the second part of (2) is strict in the sense of not hitting when , which holds almost everywhere in for . See, for example, Zhang 2010.
Let . For , (7) implies that
While the present paper continues our earlier work (Sun & Zhang 2010) by providing further theoretical and numerical justifications for (3) and (4), the estimator has appeared in different forms. In addition to (5) and (6) of Antoniadis 2010, (8) appeared in Zhang 2010. While this paper was in revision, a reviewer called our attention to Belloni et al. 2011, who focused on studying in an equivalent form as square-root lasso. We note that (3) and (4) allow concave penalties and degrees of freedom adjustments as in Zhang 2010.
Theoretical results
Let be a vector of true regression coefficients. An expert with oracular knowledge of would estimate the noise level by the oracle estimator
where , the compatibility factor (van de Geer & Bühlmann 2009), is defined as
with the cone . Since the prediction error bound is valid for all and , is related to its minimum over all and at the oracle scale :
In particular, if with and , then
Theorem 3.1 extends to the scaled lasso a unification of prediction oracle inequalities for a fixed penalty. With , (13) gives , or
For fixed penalty , the upper bound has been previously established for different and , with possibly different constant factors. Examples include (Greenshtein & Ritov 2004; Greenshtein 2006), with (van de Geer & Bühlmann 2009), and (Koltchinskii et al. 2011). In (10), the coefficient for is 1 as in Koltchinskii et al. 2011.
if with . This allows to have many small elements, as in Zhang & Huang 2008, Zhang 2009 and Ye & Zhang 2010. The bound improves upon its earlier version in van de Geer & Bühlmann 2009 by a constant factor .
Let and be as in Theorem 3.1. Set . (i) The following inequalities hold when ,
(ii) Let . For all and ,
If with and , then
Since with , the rate in (18) is essentially the square of that in (13), in view of (12). It follows that the scaled lasso provides a faster convergence rate than does the penalized maximum likelihood estimator for the estimation of the noise level (Städler et al. 2010; Sun & Zhang 2010). In particular, (18) implies that
with , when can be treated as a constant. The bounds in (20) and its general version (18) lead to the asymptotic normality (19) under proper assumptions. Thus, statistical inference about is justified with the scaled lasso in certain large--smaller- cases, for example, when under the compatibility condition (van de Geer & Bühlmann 2009).
The quantities in (22) are used in the uniform uncertainty principle (Candes & Tao 2007) and the sparse Riesz condition (Zhang & Huang 2008). We note that is the minimum eigenvalue of among , is the corresponding maximum eigenvalue, and is the maximum operator norm of size off-diagonal sub-blocks of the Gram matrix .
Suppose . Then, Theorem 3.2 holds with replaced by , and for ,
for all , where . In particular, for and ,
The proofs of Theorems 1 and 2 are based on a basic inequality
as a consequence of the Karush–Kuhn–Tucker conditions (2). The version of (25) with is well-known (van de Geer & Bühlmann 2009) and controls for sparse . When , (25) controls the excess for sparse by the same argument. The general is taken in Theorem 1, while is taken in Theorem 2. In both cases, (25) provides the cone condition in (11) and (21). This is used to derive upper and lower bounds for (7), the derivative of the profile loss function with respect to , within a neighborhood of . The bounds for the minimizer then follow from the joint convexity of the penalized loss (5).
2 Estimation after model selection
We have proved that without requiring the knowledge of , the scaled lasso enjoys prediction and estimation properties comparable to the best known theoretical results for known , and the scaled lasso estimate of enjoys consistency and asymptotic normality properties under proper conditions. However, the lasso estimator may have substantial bias (Fan & Peng 2004; Zhang 2010), and its bias is significant in our own simulation experiments. Although the smoothly clipped absolute deviation and minimax concave penalized selectors were introduced to remove the bias of the lasso (Fan & Li 2001; Zhang 2010), a theoretical study of their scaled version (3) is beyond the scope of this paper. In this subsection, we present theoretical results for another bias removing method: least squares estimation after model selection.
Given an estimator of the coefficient vector , the least squares estimator of and the corresponding estimator of the noise level in the model selected by are
where . Alternatively, we may use to estimate the noise level. However, since the effect of this degrees of freedom adjustment is of smaller order than our error bound, we will focus on the simpler (27).
In addition to the compatibility factor in (11), we use sparse eigenvalues to study the least squares estimation after the scaled lasso selection. Let be the smallest eigenvalue of a matrix and the largest . For models , define
as the sparse lower eigenvalue of the Gram matrix for models containing and the sparse upper eigenvalue for models disjoint with . Let and . The following theorem provides prediction and estimation error bounds for (27) after the scaled lasso selection, along with an upper bound for the false positive , a key element in our study.
Let be the scaled lasso estimator in (6) and the least squares estimator (27) in model . Let and be as in Theorem 3.2 and be an integer satisfying . If , then
with and , and
Moreover, in addition to the probability bound for in Theorem 3.2 (ii), for all integers ,
For Gaussian design matrices, the sparse eigenvalues and can be treated as constants when is small and the eigenvalues of the expected Gram matrix are uniformly bounded away from zero and infinity (Zhang & Huang 2008). Since and , they can be treated as constants in the same sense in Theorem 3.4. Thus, for sufficiently small , we may take an of the same order as . In this case, the difference between and the scaled lasso estimator is of no greater order than the difference between and the estimation target . Consequently,
As we have mentioned earlier, the key element in our analysis of (27) is the bound in (28). Since this is a weaker assertion than variable consistency , the conditions of Theorem 3.4 on the design matrix is of a weaker form than the irrepresentability condition for variable selection consistency (Meinshausen & Bühlmann 2006; Zhao & Yu 2006). In Zhang & Huang 2008 and Zhang 2010, upper bounds for the false positive were obtained under a sparse Riesz condition on and .
Numerical results
The top section of Table 3.2 presents our simulation results, while the bottom section includes the simulation results of Fan et al. 2012 for several joint estimators of using cross-validation, without repeating their experiment. In addition to the bias and the standard error of the ratios for the five original estimators and for the least squares estimation after model selection, we report the average model size and the relative frequency of sure screening, , as in Fan et al. 2012, where is the selected model.
Without post processing, the scaled minimax concave penalized selector with the universal penalty level clearly outperforms other procedures in this example. However, the results of the least squares estimation after model selection at penalty level are nearly identical to the top performer for all five methods. In view of the results in average model size and sure screening proportion, the success of post processing at is clearly due to the success of model selection. The five methods select too few variables at the larger penalty level and too many at the smaller , both leading to substantial bias in the estimation of for . For , selecting a slightly smaller model does not harm so much since a substantial portion of the effect of the missing variables is explained by the selected variables correlated to them. The minimax concave penalized selector is nearly unbiased in this example, so that it does not need post processing. Cross-validation methods select about 30 variables when the true model size is 3. This over selection is probably the reason for the large bias for most cross-validation methods and large standard error for all of them.
This experiment has the same setting as in the simulation study in Sun & Zhang 2010, where the scaled lasso and the scaled minimax concave penalized selection are called the naive estimators. We provide the description of the simulation settings in Sun & Zhang 2010 in our notation as follows: , the are normalized columns from a Gaussian random matrix with independent and identically distributed rows and correlation between the -th and -th entries within each row, for the minimax concave penalty and smoothly clipped absolute deviation penalty, the nonzero are composed of five blocks of centered at random multiples of 25, sets , and is a vector of independent and identically distributed variables. Thus, the true noise level is . We set for low correlation between design vectors and for high correlation.
2 Real data example
We study a data set containing 18976 probes for 120 rats, which is reported in Scheetz et al. 2006. Our goal is to find probes that are related to that of gene TRIM32, which has been found to cause Bardet–Biedl syndrome, a genetically heterogeneous disease of multiple organ systems including the retina. We consider linear regression with the probe from TRIM32, 1389163_at, as the response variable. As in Huang et al. 2008, we focus on 3000 probes with the largest variances among the 18975 covariates and consider two approaches. The first approach is to regress on these probes. The second approach is to regress on the 200 probes among the 3000 with the largest marginal correlation coefficients with TRIM32. For the cross-validation lasso, we randomly partition the data 1000 times, each with a training set of size 80 and a validation set of size 40. For each partition, the penalty level is selected by minimizing the prediction mean squared error in the validation set. Then we compute the lasso estimator with all 120 observations at the penalty level equal to the median of the selected penalty levels with the 1000 random partitions. Since cross-validation tends to choose a larger model, we also consider an adjusted version using the cross-validated error of the least squares estimator after the lasso selection. For the minimax concave penalty, we set , where is the 95% quantile of .
Table 4.2 shows the probe sets identified by four methods: the cross-validation lasso, its adjusted version, the scaled lasso at at universal penalty level , and the minimax concave penalized selection at the same penalty level. We apply stability selection (Meinshausen & Buhlmann 2010) to check the reliability of selection. Let be independent variables with and
where is the penalty level chosen by individual methods. Stability selection selects variables with nonzero estimated over 50 times in 100 replications. We observe that the scaled minimax concave penalized selector produces most sparse and most stable selection, followed by the adjusted cross-validation, the scaled lasso and then the plain cross-validation. The selection results are consistent among the four methods in the sense that the selected models are almost nested. Since the model size is between 6 and 8 by stability selection in all 8 cases and by the scaled minimax concave penalized selection for both and , these two methods provide most consistent results. The scaled lasso and the adjusted cross-validation yield identical lasso and stability selections for and identical stability selection for .
We also compare the prediction performance of the scaled lasso with that of the lasso with the best fixed penalty level. We compute the scaled estimators in 1000 replications. In each replication, the dataset is split at random into a training set with 80 observations and a test set with 40 observations. The prediction mean squared error is computed within the test set, while the scaled estimators and the lasso estimator with fixed penalty level are computed based on the training set. Figure 1 demonstrates that in prediction, the scaled lasso with chosen as performs almost as well as the lasso with the optimal fixed .
In addition, we compare the prediction performance of all the estimators mentioned in this section. In each replication, we compute the penalized maximum likelihood estimator, its bias-correction, and scaled penalization methods based on the training set of 80 observations. For cross-validation, the training set of 80 observations is further partitioned at random 100 times into two groups of sizes 60 and 20, and a penalty level is selected by minimizing the estimated loss in the smaller group for the lasso estimator based on the larger group. This selected penalty level is then used for the lasso with the entire training set. Thus, the cross-validation lasso is also based on the training set with 80 observations. For the penalty level selected by the adjusted cross-validation, two estimators are considered: the lasso estimator and the least squares estimator after the lasso selection. In Table 4.2, we present the medians of the prediction mean squared error and the selected model size in the 200 replications. The scaled lasso has comparable prediction performance as cross-validation. Again, Table 4.2 suggests that original cross-validation tends to choose larger models, while adjusted cross-validation leads to results comparable with the scaled lasso.
Discussion
In the proof of our theoretical results for the scaled lasso, we use oracle inequalities for fixed penalty which unify and somewhat sharpen existing results. We now present this result. Define
as a sharper version of in (10).
with in (12). Moreover, in the same event and with in (16),
The interpretations of (32) and (33) are given in (15) and (17), along with their relationship to several existing results. We note here that the condition for (15) and (17), weaker than the parallel condition on the restricted eigenvalue (Bickel et al. 2009), can be slightly weakened by using in (21) (Ye & Zhang 2010).
Acknowledgement
This research was supported by the National Science Foundation and the National Security Agency. We thank Jian Huang for sharing the gene expression data, and reviewers for valuable suggestions.
Appendix
Here we prove Proposition 2.1, Theorem 5.1, Theorem 3.1, Theorem 3.2 and then Theorem 3.4.
(i) Since is a solution of (2) at ,
for all . Since is unchanged in a neighborhood of , for . Thus,
(ii) The convergence of (3) and (4) follows from the joint convexity of . The scale invariance follows from , where expresses the dependence of (5) on the data .
(i) Let . Since and for , the inner product of and the Karush–Kuhn–Tucker condition (2) yield
Since , this gives the basic inequality (25). Let . Since , is no greater than with . Thus, (25) implies
with . For and , (34) directly yields . For general , we want to prove
It suffices to consider . In this case, by (34), so that by (11)
Let and . It follows from (34) and (35) that . For such , . This gives .
For , it suffices to consider the case , where the cone condition holds for . Now, satisfies . The maximum of , attained at , is
(ii) Let and . It follows from (34) with that
It suffices to consider . In this case
Thus, , or equivalently . It follows from (11) that , so that
Let and . Write (37) as . Subject to this inequality, the maximum of is . This maximum, attained at , is . Thus,
This gives for .
Assume without loss of generality. Consider and the penalty level for the lasso. Since and , the Cauchy–Schwarz inequality and (32) imply
Since for , the derivative (7) of the loss with satisfies
This implies by the strict convexity of the profile loss (5) in . For , by (10) and (12), so that at ,
This implies by the strict convexity of (5) in . Thus, the first part of (13) holds. Moreover,
The proof of Theorem 2 requires the following lemma.
Let have the t-distribution with degrees of freedom. Then, there exists such that for all
Let . Since has the -distribution,
where as .
We need to express as a function of at in the proof. Define
We have , and .
(i) Consider . Let and . Since , the Karush–Kuhn–Tucker condition (2) gives
as lower and upper bounds for . This is a key point in the proof.
For , , so that (33) in Theorem 5.1 implies . It follows (39) that for ,
due to . As in the proof of Theorem 3.1, we find by (7) and the strict convexity of (5) in .
Now we prove that . For , by (16). Thus, since and , for , (39) and (33) imply that
It follows that by convexity.
Since , . This completes the proof of (18).
Since for any ,
Since , this bounds the tail probability of by the union bound. Since follows the distribution, converges to in distribution, which then implies (19) by (18) under .
Let . It follows from the proof of Theorem 3.2 (i) that , so that . By (2), for . Let with . Since is the upper sparse eigenvalue, . By the basic inequality (25) with , and is in the cone . Thus, since by (11), . It follows that . Since all of size have size , does not have a subset of size . This gives the first inequality in (28).
Let be the orthogonal projection to the linear span of . By the definition of and the prediction error bound in Theorem 5.1,
This gives the second inequality in (28). The prediction error bound follows from