Uniform Asymptotic Inference and the Bootstrap After Model Selection
Ryan J. Tibshirani, Alessandro Rinaldo, Robert Tibshirani, Larry Wasserman
Introduction
There has been a recent surge of work on conducting formally valid inference in a regression setting after a model selection event has occurred, see Berk et al. 2013; Lockhart et al. 2014; Tibshirani et al. 2016; Lee et al. 2016; Fithian et al. 2014; Bachoc et al. 2014, just to name a few. Our interest in this paper stems in particular from the work of Tibshirani et al. 2016, who presented a method to produce valid p-values and confidence intervals for adaptively fitted coefficients from any given step of a sequential regression procedure like forward stepwise regression (FS), least angle regression (LAR), or the lasso (the lasso is meant to be thought of as tracing out a sequence of models along its solution path, as the penalty parameter descends from to ). These authors use a statistic that is carefully crafted to be pivotal after conditioning on the model selection event. This idea is not specific to the sequential regression setting, and is an example of a broader framework that we might call selective pivotal inference, applicable in many other settings, as in, e.g., Taylor et al. 2016; Lee et al. 2016; Lee & Taylor 2014; Loftus & Taylor 2014; Reid et al. 2017; Choi et al. 2014; Fithian et al. 2014; Hyun et al. 2016.
A high-level description of the selective pivotal inference framework for sequential regression is as follows (details are provided in Section 2). FS, LAR, or the lasso is run for some number of steps , and a model is selected, call it . For FS and LAR, this model will always have active variables, and for the lasso, it will have at most , as variables can be added to or deleted from the active set at each step. We specify a linear contrast of the mean of interest, e.g., one giving the coefficient of a variable of interest in the model at step , in the regression of onto the active variables. By assuming normal errors in (1), and examining the distribution of conditional on having selected model , which we denote by , we can construct a confidence interval satisfying
for a given . The interpretation: if we were to repeatedly draw from (1) and run FS, LAR, or the lasso for steps, and only pay attention to cases in which we selected model , then among these cases, the constructed intervals contain with frequency tending to .
The above is a conditional perspective of the selective pivotal inference framework for FS, LAR, and lasso. An unconditional or marginal point of view is also possible, which we now describe. For each possible selected model , a constrast vector is specified, and the contrast is considered when model is selected, . To be concrete, we can again think of a setup such that gives the coefficient of a variable in the model at step , in the projection of onto the active set. Confidence intervals are then constructed in exactly the same manner as above (without change), and conditional coverage over all models implies the following unconditional property for ,
The interpretation is different: if we were to repeatedly draw from (1) and run FS, LAR, or lasso for steps, and construct confidence intervals , then these intervals contain their respective targets with frequency approaching . Notice that, by construction, the target itself may change each time we draw , though it is the same for all that give rise to the same selected model. In terms of the setting for regression contrasts described above, each time we draw and carry out the inferential procedure, the interval covers the coefficient of a possibly different variable in the active model, in the projection of onto the active variables. Figure 1 demonstrates this point.
(The above inequalities, as in and , are meant to be interpreted componentwise; we are also implicitly assuming that the limiting distribution is continuous, otherwise the above inner supremum should be restricted to continuity points of .) This is much stronger than the notion of pointwise convergence in distribution, which only requires that
for a particular sequence of distributions , .
A recent article by Kasy 2015 emphasizes the importance of uniformity in asymptotic approximations. This authors points out that a uniform version of the continuous mapping theorem follows directly from a standard proof of the continuous mapping theorem (e.g., see Theorem 2.3 in van der Vaart 1998).
Kasy 2015 also remarks that the central limit theorem for triangular arrays, specifically the Lindeberg-Feller central limit theorem (e.g., Proposition 2.27 in van der Vaart 1998) naturally extends to the uniform case. The logic is, roughly speaking: uniform convergence in (2) is equivalent to pointwise convergence over all sequences of distributions , , and triangular arrays, by design, can have a different distribution assigned to each row. Therefore if the Lindeberg condition holds for any possible sequence, then so does the convergence to normality.
where does not depend on the sequence , . Then converges in distribution to , uniformly with respect to .
In our work, a motivating reason for the study of uniform convergence is the associated property of uniform validity of asymptotic confidence intervals. That is, if depends on a parameter of the distribution , but does not, then we can consider any confidence set built from a probability rectangle of ,
and the uniform convergence of to , really just by rearranging its definition in (2), implies
Meanwhile, pointwise convergence as in (3) only implies
for a particular sequence , . For a confidence set satisfying (4), and a given tolerance , there exists a sample size such that the coverage is guaranteed to be at least , for , no matter the underlying distribution (over the class of distributions in question). Note that this is not necessarily true for a pointwise confidence set as in (5), as the required sample size here could depend on the particular distribution under consideration.
2 Summary of main results
An overview of our main contributions is as follows.
We establish that TG statistics for typical inferences along the FS, LAR, and lasso paths only depend on the data through and (Lemmas 3, 4, and 5 in Section 3), which is important since these two quantities have asymptotic limits in a standard low-dimensional asymptotic setup.
Placing mild constraints on the mean and error distribution in (1), and treating the dimension as fixed, we prove that the TG test statistic is asymptotically pivotal, converging to (the standard uniform distribution), when evaluated at the true population value for its pivot argument. We show that this holds uniformly over a wide class of distributions for the errors, without any real restrictions on the predictors (first part of Theorem 7 in Section 4).
The resulting confidence intervals are therefore asymptotically uniformly valid, over the same class of distributions (second part of Theorem 7 in Section 4).
The above asymptotic results assume that the error variance is known, so for unknown, we propose a plug-in approach that replaces in the TG statistic with a simple estimate, and alternatively, an efficient bootstrap approach. Both allow for conservative asymptotic inference (Theorem 11 in Section 5).
We present detailed numerical experiments that support the asymptotic validity of the TG p-values and confidence intervals for inference in low-dimensional regression problems that have nonnormal errors (Section 6). Our experiments reveal that the plug-in and bootstrap versions also show good performance, and the bootstrap method can often deliver substantially shorter intervals than those based directly on the TG statistic.
Our experiments also also suggest that the TG test statistic (and plug-in, bootstrap variants) may be asymptotically valid in even broader settings not covered by our theory, e.g., problems with heteroskedastic errors and (some) high-dimensional problems.
We prove that TG statistic does not exhibit a general uniform convergence to when the dimension is allowed to increase (Theorem 12 in Section 7).
3 Related work
A recent paper by Tian & Taylor 2017 is very related to our work here. These authors examine the asymptotic distribution of the TG statistic under nonnormal errors. Their main result proves that the TG statistic is asymptotically pivotal, under some restrictions on the model selection events in question. We view their work as providing a complementary perspective to our own: they consider a setting where the dimension grows, but place strong regularity conditions on the selected models; we adopt a more basic setting with fixed, and prove more broad uniformly valid convergence results for the TG pivot, free of regularity conditions.
In a sequence of papers, Leeb & Potscher 2003; Leeb & Potscher 2006; Leeb & Potscher 2008 prove that in a classical regression setting, it is impossible to find the distribution of a post-selection estimator of the underlying coefficients, even asymptotically. Specifically, they prove for an estimate of some underlying coefficient vector , any quantity of the form , for a linear transform , cannot be used for inference after model selection. Though can be made to be pivotal or at least asymptotically pivotal (once is chosen once appropriately), this is no longer true in the presence of selection, even if the dimension is fixed and the sample size approaches . Furthermore, they show that there is no uniformly consistent estimate of the distribution of (either conditionally or unconditionally), which makes unsuitable for inference. This fact is essentially a manifestation of the well-known Hodges phenomenon. The selective pivotal inference framework, and hence our paper, circumvents this problem as we do not claim (nor attempt) to estimate the distribution of , and instead make inferences using an entirely different pivot that is constructed via a careful conditioning scheme.
4 Notation
Selective inference
In this section, we review the selective pivotal inference framework for sequential regression procedures. We present interpretations for the inferences from both conditional and unconditional perpsectives, in Sections 2.2 and 2.5, respectively. The other subsections provide the necessary details for understanding the framework, beginning with the selection events encountered along the FS, LAR, and lasso paths.
The active sets are nested across steps, , as FS selects one variable to add to the active set at each step. However, the sign vectors are not, since these are determined by least squares on the active variables at each step. Hence, as defined, the number of possible models after steps of FS is
Moreover, the corresponding partition elements , in (6) are all convex cones. The proof of this fact is not difficult, and requires only a slight modification of the arguments in Tibshirani et al. 2016, given in Appendix A.1 for completeness. The result is easily seen for : after one step of FS, assuming without a loss of generality that have unit norm, we can express, e.g.,
2 Inference after selection
and therefore is a test for the significance of the th normalized coefficient in the linear projection of onto , written as for short. (Though the normalization in the denominator is irrelevant for this significance test, it acts as a key scaling factor for the asymptotics in Section 4.) The idea of using a projection parameter for inference, , has also appeared in, e.g., Berk et al. 2013; Wasserman 2014; Lee et al. 2016. Here is now a summary of the testing framework.
Assume i.i.d. errors in (1). Under the null hypothesis, the TG statistic has a standard uniform distribution, over draws of that land in . Mathematically, this is the property
for all . The probability above is taken over an arbitrary mean parameter for which (in fact, the TG statistic is constructed so that the law of only depends on through , so this is unambiguous). In order for (8) to hold, of course, and cannot be random, i.e., they cannot depend on , though they can be functions of .
Thus serves as a valid p-value (with exact finite sample size) for testing the null hypothesis , conditional on .
A confidence interval is obtained by inverting the test in (8). Given a desired confidence level , we define to be the set of all values such that . Then, by construction, the property in (8) (which we reiterate, assumes i.i.d. errors) translates into
The interpretation of the above statement is straightforward: the random interval contains the fixed parameter with probability , conditional on .
3 The truncated Gaussian pivot
We now describe the truncated Gaussian (TG) pivotal quantity in detail. As defined in Section 2.1, if we write for the selected model from the given algorithm (FS, LAR, or lasso), run for steps on , then is a convex cone, for any fixed achieveable model . Hence
for a fixed matrix (here the inequality is meant to be interpreted componentwise). Now to define the pivot for testing , several preliminary quantities must be introduced:
This has the following property, as stated in (8): when is drawn from (1) with i.i.d. errors, and , the pivot is uniformly distributed conditional on . See Lemmas 1 and 2 in Tibshirani et al. 2016 for a proof of this result.
4 P-values and confidence intervals
A statistic aligned to have power against the two-sided alternative is simply given by . For purely testing purposes, we find the one-sided p-values discussed above to be more natural, and hence these will serve as our default. On the other hand, for constructing confidence intervals, we prefer to invert the two-sided statistics, since these lead to two-sided intervals. As
the previously described confidence interval in (9) is just given by inverting the two-sided pivot.
To summarize: the default in this work, as with Tibshirani et al. 2016, is to consider one-sided hypothesis tests, but two-sided intervals. These are just two slightly different uses of the same pivot.
5 Inference after selection, revisited
We have portrayed selective pivotal inference, in sequential regression procedures, as a method for producing conditional p-values and intervals. An unconditional interpretation of this framework is also possible, which we describe here.
where denotes the indicator function for the partition element (and is as before, defined in (10)). The unconditional statistic can be used as follows: if a response is drawn from (1), then we can form to test the hypothesis .
Assume that the errors in (1) are i.i.d. . Then under the proper hypothesis, by summing up the conditional property in (8) across partition elements, we have
for all . The assertion above holds for a parameter such that , which we use as shorthand for for all . Note that this full specification, across all , is critical in order to apply the relevant null probability within each partition element (giving rise to the equality in (12)).
Therefore serves as a valid p-value (with exact finite sample size)—but for testing what null hypothesis? Formally, it is attached to , an exhaustive specification of , over all , but in truth, carries no information about models other than the selected one, . For this reason, we actually consider to be a p-value for the random null hypothesis . This is made more precise through confidence intervals.
A confidence interval is obtained by inverting the test in (12). But the TG statistic at ,
only depends on through . Thus, given a desired confidence level , let us define to be the set of such that , and to be the set of such that . Then we can see that
so the confidence interval is effectively infinite with respect to the values , , and inverting the test in (12) yields
The above expression says that the random interval traps the random parameter with probability , and thus, this supports the interpretation of as the null hypothesis underlying the unconditional TG statistic.
The pivotal property in (12) is derived under the distributional assumption that , i.e., for all , which may seem unnatural, as the catalog of pivot value can be large (e.g., on the order of after steps of FS), and so this is condition on possibly many contrasts of . However, it is worth emphasizing that the unconditional testing property in (12) is really only useful in that it allows us to formulate the unconditional confidence interval property in (13), which is a more natural statement about coverage of a single (random) parameter. When viewing selective inference from an unconditional perpsective, we find it more natural to place the focus on confidence intervals rather than hypothesis testing; in many ways, we find the former the more natural of the two perspectives, unconditionally. Tibshirani et al. 2016 in fact suggest separate nomenclature for the unconditional case, referring to the property in (13) as that of a selection interval (rather than confidence interval), to emphasize that this interval covers a moving target.
The master statistic
Given a response and predictors , our description thus far of the selected model , statistics and , etc., has ignored the role of . This was done for simplicity. The theory to come in Section 4 will consider to be nonrandom, but asymptotically must (of course) grow with , and so it will help to be precise about the dependence of the selected model and statistics on . We will denote these quantities by , , and to emphasize this dependence. We define
a -dimensional quantity that we will call the master statistic. As its name might suggest, this plays an important role: all normalized coefficients from regressing onto subsets of the variables can be written in terms of . That is, for an arbitrary set , the th normalized coefficient from the regression of onto is
which only depends on through . The same dependence is true, it turns out, for the selected models from FS, LAR, and the lasso. We defer the proof of the next lemma, as with all proofs in this paper, until the appendix.
For each of the FS, LAR, and lasso procedures, run for steps on data , the selected model only depends on through , the master statistic.
In more detail, for any fixed , the matrix such that can be written as , where depends only on . Hence
This lemma asserts that the master statistic governs model selection, as performed by FS, LAR, and the lasso. It is also central to TG pivot for these procedures. Denoting , the statistic in (10) only depends on through three quantities:
The third quantity is always a function of , by Lemma 3. When is chosen so that is a normalized coefficient in the regression of onto a subset of the variables in , the first two quantities are also functions of . Thus, in this case, the TG pivot only depends on through the master statistic ; in fact, it is continuous at any point such that is nonsingular and does not lie on the boundary of a model selection event.
Fix any model , and suppose that is chosen so that is a normalized coefficient from projecting onto a subset of the variables in . Then the TG statistic only depends on by means of , so that we may write
Further, the function is continous at any point such that is nonsingular and .
Finally, we show that the conditional pivotal property of the TG statistic in (8) can be phrased entirely in terms of the master statistic.
Assume the conditions of Lemma 4, and additionally that is drawn from (1). Construct the master statistic . Then there is a function such that
Thus if the errors in (1) are i.i.d. , then the conditional pivotal property (8) of the TG statistic can be reexpressed as
Equipped with the last two lemmas, asymptotic theory for the TG test, when is fixed, is not far off. Under weak conditions on the data model in (1), the central limit theorem tells us that converges weakly to a normal random variable. With converging to a deterministic matrix, the continuous mapping theorem will then provide the appropriate asymptotic limit for the statistic . This is made more precise next.
Asymptotic theory
We specify the class of distributions that we will be working with for in (1). Let be a fixed, known constant. First we define a set of error distributions
As grows, we allow the underlying mean to change, but we place a restriction on this parameter so that it has an appropriate asymptotic limit. Specifically, we consider a class of sequences of mean parameters such that has an asymptotic limit lying in some compact set, with uniform convergence to this limit. Formally, write (in a slight abuse of notation) to denote a sequence of mean parameters in , and let denote the set of limit points of . Then, for some constant , we require of the class ,
2 Uniform convergence results
We begin with a result on the uniform convergence of (the random part of) the master statistic to a normal distribution, both marginally and conditionally.
Assume that has asymptotic covariance matrix , as in (14), and satisfies the normalization condition in (15). Let , this class as defined in (16), for a sequence of mean parameters , as defined in (17). Denote as . Then converges in distribution to , uniformly over , and uniformly over all . That is,
This lemma, combined with Lemmas 4 and 5 of the last section, leads us to uniform asymptotic theory for the TG test. We remind the reader that , the number of steps, is to be considered fixed in the next result (as it is throughout the paper).
Assume the conditions of Lemma 6. Suppose FS, LAR, or the lasso is run for steps on . Below we describe the conditional and unconditional asymptotic results separately.
(a, Markovic) Fix any model . Let be a vector such that gives a normalized coefficient in the projection of onto some subset of the variables in , and let be an arbitrary pivot value. Then under , the conditional TG statistic converges in distribution to , uniformly over , and over . That is,
Moreover, if we define to be the set of such that , then is an asymptotically uniformly valid confidence interval for . That is,
(b) Let be a catalog of vectors such that each yields a normalized coefficient in the projection of onto a subset of the variables in , for , and be a catalog of pivot values. Then under , the same results as in part (a) hold marginally. That is,
and for defined to be the set of such that ,
An initial version of this work contained only the unconditional result in part (b) of the theorem. Jelena Markovic pointed out that the conditional result in part (a) should also be possible, and thus this conditional result should also be attributed to her. Between the initial and the current version of this paper, in addition to revising Theorem 7, we have also revised Theorems 11 and 12 to include the appropriate conditional results.
Unknown σ2\sigma^{2} and the bootstrap
The results of the previous section assumed that the error variance in the model (1) was known. Here we consider two strategies when is unknown. The first plugs a (rather naive) estimate of into the usual TG statistic. The second is a computationally efficient bootstrap method. Both, as we will show, yield asymptotically conservative p-values. (In practice, the bootstrap often gives shorter confidence intervals than those based on the TG pivot; see Section 6.)
Given a model , contrast vector , and pivot value , consider the TG statistic . Let us abbreviate
where the latter two functions are as defined in Section 2.3. In this notation, we can succintly write the TG statistic as
When is unknown, we propose a simple plug-in approach that replaces with , where
the sample variance of (here denotes the sample mean), and is a fixed constant. To be explicit, we consider the modified TG statistic
The scaling factor facilitates our theoretical study of the above plug-in statistic, and practically, we have found that ignoring it (i.e., setting ) works perfectly well, though a choice of, say, seems to have a minor effect anyway.
When the mean of is nonzero, the sample variance is generally too large as an estimate of . As we will show, the modified statistic in (19) thus yields asymptotically conservative p-values. Residual based estimates of are not as useful in our setting because they depend more heavily on the linearity of the underlying regression model, and they suffer practically when is close to (see also the discussion at the start of Section 6).
2 An efficient bootstrap approach
As an alternative to the plug-in method of the last subsection, we investigate a highly efficient bootstrap scheme that does not rely on knowledge of . Our general framework so far treats as fixed, and for our bootstrap strategy to respect this assumption, we cannot use, say, the pairs bootstrap, and must perform sampling with respect to only. The residual bootstrap is ruled out since we do not assume that the mean follows a linear model in . This leaves us to consider simple bootstrap sampling of the components of . This is somewhat nonstandard, as the components of in (1) are not i.i.d., but it provides a mechanism for provably conservative asymptotic inference, and it is what makes our approach so computationally efficient.
where the probability on the right-hand side is taken with (and thus ) treated as fixed, and with denoting a random variable. The main idea is now to approximate the truncated normal distribution underlying the TG statistic with an appropriate one from bootstrap samples,
where is a constant as before, and for a small constant . Again, we have found that ignoring the scaling factor (i.e., setting ) works just fine in practice, though a choice like does not cause major differences anyway. On the contrary, a nonzero choice of the padding factor like does play an important practical role, since the bootstrap probabilities in the numerator and denominator in (20) can sometimes be zero.
Lastly, it is worth emphasizing that practical estimation of the bootstrap probabilities appearing in (20) is quite an easy computational task, because the regression procedure in question, be it FS, LAR, or the lasso, need not be rerun beyond its initial run on the observed . After this initial run, we can just save the realized quantities , and then draw, say, bootstrap samples in order to estimate the probabilities in (20). This is not at all computationally expensive. Moreover, to estimate (20) over multiple trial values of (so that we can invert these bootstrap p-values for a bootstrap confidence interval), only a single common set of bootstrap samples is needed, since we can just shift appropriately for each bootstrap sample .
3 Asymptotic theory for unknown σ2\sigma^{2}
Treating the dimension as fixed, we will assume the previous limiting conditions (14), (15) on the matrix , and additionally, that
Assume that satisfies (14), (15), (21). If is any vector such that gives a normalized regression coefficient from projecting onto some subset of the variables in , then
We specify assumptions on the distribution of in (1) that are similar to (but slightly stronger than) those in Section 4.1. For constants , we define a set of error distributions
where as before, denotes the distribution of , for . We define a class of sequences of mean parameters that satisfies, as before,
for a constant , where recall denotes the set of limit points in ; also, for each , at each , we require
for constants , where . Note that the assumptions , with and , are not much stronger than our assumptions in Section 4.1: we require the existence of two more moments for the error distribution, and place an additional weak condition on the growth of (components of) . These conditions are sufficient to prove the following helpful lemma.
Assume that satisfies (14), (15). Let , where this class is as defined in (22), and let , where this class is as in (23), (24). Then for any fixed , and ,
In words, the event has probability tending to 1 conditional on , uniformly over , and over . Furthermore, denoting the sample third moment of as
we have that for any , there exists such that for sufficiently large ,
The last two lemmas allow us to tie the distribution function of our bootstrap contrast to that of a normal random variable.
Assume that satisfies (14), (15), (21). Let , as defined in (22), and let , as defined in (23), (24). Let , and let be such that gives a normalized regression coefficient from projecting onto a subset of the variables in . Then for any , there exists such that sufficiently large ,
We are now ready to present uniform asymptotic results for the plug-in and bootstrap TG statistics. We remind the reader the number of steps is treated as fixed below (as it is throughout).
Assume the conditions of Lemma 10. Suppose FS, LAR, or the lasso is run for steps on . Then under , the conditional plug-in TG statistic and conditional bootstrap TG statistic are each asymptotically larger than in distribution, uniformly over , and over . That is,
where denotes the positive part of . Further, given any catalog of vectors such that each yields a normalized coefficient in the projection of onto a subset of the variables in , for , the same results hold marginally under .
For simplicity, we analyzed the plug-in and bootstrap statistics simultaneously. Consequently, the conditions assumed to prove asymptotic properties of the plug-in approach are stronger than what we would need if we were to study this method on its own, but there are not major differences in these conditions.
Theorem 11 establishes that the plug-in and bootstrap versions of the TG statistic are asymptotically conservative when viewed as p-values under . If we look more broadly at the distribution of these test statistics under , for an arbitrary value of , then a technical barrier arises. For each statistic, our proof of its asymptotic conservativeness leverages the fact that the truncated Gaussian survival function decreases (in a pointwise sense), as its underlying variance parameter decreases. To extend these results to the case of an arbitrary pivot value , we would need the analogous fact to hold when we replace the survival function of the Gaussian variate truncated to , with that of tuncated to , on the event . Yet, without the guarantee that (which clearly cannot always be true, for an arbitrary value of ), it is no longer the case that decreasing the variance from to always decreases the survival functions of these two truncated Gaussians; see Appendix A.11. This means that confidence intervals given by directly inverting either the plug-in or bootstrap TG statistic do not have provably correct asymptotic coverage properties, under the current analysis.
From the arguments in the proof of Theorem 11, we can construct one-sided confidence intervals with conversative asymptotic coverage, by forcing them to include . We do not pursue the details here, as we have found that these one-sided intervals are practically too wide to be of interest.
Importantly, the plug-in and bootstrap TG statistics often display excellent empirical properties, as we will show in the next section. A more refined analysis is needed to establish asymptotic uniformity for the distribution of these statistics under . Such asymptotic uniformity, for arbitrary , would lead to asymptotic coverage guarantees for confidence intervals produced by inverting these statistics, and we leave this extension to future work.
Examples
We present empirical examples that support the theory developed in the previous sections, and also suggest that there is much room to refine and expand our current set of results. The first two subsections examine a low-dimensional problem setting that is covered by our theory. The last two look at substantial departures from this theoretical framework, the heteroskedastic and high-dimensional settings, respectively. In all examples, the LAR algorithm was used for variable selection and associated inferences; results with the FS and lasso paths were roughly similar. Also, in all examples, where not explicitly stated otherwise, the computed p-values are a test of whether the target population value is 0.
It may be worth discussing two potentially common reactions to our experimental setups, especially for the low-dimensional problems described in the next subsections. First, our plug-in statistic uses as an estimate for ; why not use an estimate from the full least squares model of on , since this would be less conservative? While experiments (not shown) confirm that this works in low-dimensional regression problems, such an estimate becomes anti-conservative as the number of variables grows (particularly, irrelevant ones), and is obviously not applicable in high-dimensional problems. Therefore, we stick with the simple estimate , as this is always applicable and always conservative.
Second, to determine variable significance in a low-dimensional problem, one could of course fit a full regression model and inspect the resulting p-values and confidence intervals. These p-values and intervals could even be Bonferonni-adjusted to account for selection. Of course, this strategy would not be possible for a high-dimensional problem, but if the number of predictors is small enough, then it may work perfectly fine. So when should one use more complex tools for post-selection inference? This is an important question, deserving of study, but it is not the topic of this paper. The examples that follow are intended to portray the robustness of the selective pivotal inference method against nonnormal error distributions; they are not meant to represent the ideal statistical practice in any given scenario.
Figure 3(a) displays QQ plots of p-values for testing the significance of the variable entered into the active model, across 3 steps of LAR. (The QQ plots compare the p-values to a standard uniform distribution.) The p-values were computed using the TG statistic with , the plug-in TG statistic with as its estimate for , and the bootstrap TG statistic with 50,000 bootstrap samples used to approximate the probabilities in the numerator and denominator of (20), and padding factor . (The scaling factor was ignored, i.e., set to , for the plug-in and bootstrap statistics.) In steps 1 and 2, the p-values are restricted to repetitions in which a correct variable selection was made—i.e., variable 1 or 2 was entered into the active LAR model. In step 3, the p-values are from repetitions in which an incorrect variable selection was made—i.e., one of variables 3 through 10 was entered into the active model. Since the underlying signal was fairly strong and the predictors uncorrelated, such selections happened the majority of the time; specifically, the p-values displayed for steps 1, 2, and 3 comprise approximately 95%, 85%, and 87% of the 500 repetitions, respectively. The p-values in steps 1 and 2 show reasonable power, for all 3 statistics (TG, plug-in, and bootstrap types), and all 4 error distributions. Also, the p-values in step 3 are uniform, as desired, again for all statistics and all error distributions. Though the guarantees (for uniform null p-values) are only asymptotic for the Laplace, uniform, and skew normal error distributions, such asymptotic behavior appears to kick in quite early for these distributions, as the sample size here is only . Further, the QQ plots reveal that the p-values for the nonnormal error distributions are not really any farther from uniform than they are in the normal case. This is somewhat remarkable, recalling that the p-values are, by construction, exactly uniform under normal errors.
Figure 3(b) inspects the TG statistic and plug-in and boostrap variants, when the pivot value is set to the true population value. That is, we set in computing the statistics in (18), (19), and (20), in each data instance and each step of LAR. The figure collects the p-values across all 3 steps of LAR, for each of the 4 error distribution types. According to our theory, the distribution of the TG pivotal statistics here should be asymptotically uniform. This is clearly supported by the QQ plots. Interestingly, both plug-in and bootstrap pivotal statistics also appear uniform in the QQ plots, and yet, this is not a case handled by our asymptotic theory: recall, Theorem 11 fixes the pivot value to be 0 (as, otherwise, technical difficulties are encountered in its proof). This gives empirical evidence to the idea that a more refined analysis could extend Theorem 11 to the broader setting (of arbitrary pivot values) handled by Theorem 7. Moreover, it suggests that inverting the plug-in and bootstrap TG statistics should yield intervals with proper coverage, which is verified in the next subsection.
2 Confidence interval examples
We stay in same setting as the last subsection, so that , , and for a coefficient vector with its first 2 components equal to and , and the rest equal to 0. We invert the TG, plug-in TG, and bootstrap TG statistics to obtain 90% confidence intervals at each LAR step. See Table 1 for a numerical summary. “Coverage” refers to the average fraction of intervals that contained their respective targets over the 500 repetitions, “power” is the average fraction of intervals that excluded zero, and “width” is the median interval width. These are all recorded in an unconditional sense, i.e., no screening of repetitions was performed based on the variables that were selected across the 3 steps of LAR (the conditional coverages however, were quite similar). From the table, we can see that all 3 methods lead to accurate coverage (around 90%) in all cases. We can further see that the intervals from the bootstrap TG statistic are shorter than those from the plug-in TG statistic in all cases, and considerably shorter than both the plug-in and original TG statistics in steps 2 and 3. The power from the bootstrap TG intervals is generally better than that from the plug-in TG intervals; also, it is on par with the power from the original TG statistic in step 1, but somewhat worse in step 2. Recall that the original TG statistic uses knowledge of the error variance () but the bootstrap and plug-in variants do not.
It is a bit surprising that the bootstrap intervals can be shorter but still have worse power than the original TG intervals. This is easier to understand once the intervals are visualized, as done in Figure 4. The figure shows 100 sample intervals from the first LAR step, under normally distributed errors. Sample intervals from the other error models are shown in Appendix A.13. We see that the bootstrap TG intervals are indeed shorter, but compared to the original TG intervals, they are more symmetric around the target population values. The original TG intervals, being more asymmetric, are often shorter on the side (of the target value) facing 0, and this results in better power.
3 Heteroskedastic errors
4 High-dimensional examples
A negative result in high dimensions
We prove that the TG statistic fails to converge to a uniform distribution, under the null hypothesis, in a data model that has nonnormal errors and is high-dimensional, but otherwise represents a fairly standard setting: the “many means” setting. We write the observation model as
where we interpret as replications, and as dimensions. In total there are hence observations. Denote
We assume that the errors , , in (25) are i.i.d. from the following mixture:
The mixing proportion and mean shift will both scale with . Moreover, they will be chosen so that (for each ) the error variance is
As mentioned, we will consider model selection events of the form
We note that this is exactly the same selection event as that from the first step of FS, LAR, or lasso paths, when run on the regression version of this problem with orthogonal design . It is not hard to check that the TG statistic for conditionally testing , given that , is
As per the spirit of our paper, we can also view this statistic unconditionally; for this it is helpful to define , and denote by the order statistics. Then from (27), we can see that the unconditional TG statistic for testing the selected mean being 0 is
The framework underlying the TG statistic tells us that if the errors in (25) are i.i.d. , then for any fixed model , the pivot is uniformly distributed conditional on . Further, if and are the largest and second largest absolute values of centered normal random variables (each with variance ), then the unconditional pivot is again uniform. But when are large, and are defined by the order statistics of nonnormal random variates, the statistic —which in this case is defined by the extreme tail behavior of the normal distribution—could be nonuniform. The next theorem asserts that such nonuniformity does indeed happen asymptotically if we choose the mixture distribution in (26) appropriately.
Assume the observation model (25), where the errors are all drawn i.i.d. from (26). Let and scale in such a manner that . Further, let
so that the error variance is fixed at . Then under the global null hypothesis, , the unconditional TG statistic in (28) does not converge in distribution to . In particular, on an event whose limiting probability is at least , the statistic converges to 0.
Further, the same results hold conditionally on any selected model. That is, for any fixed , the conditional TG statistic does not converge in distribution to , and on an event with limiting probability (conditional on ) at least , it converges to 0.
The assumed condition requires the dimension to diverge to , but not necessarily the number of replications , though it clearly allows to diverge at a sufficiently slow rate. On the other hand, if were fixed and diverged to , then the result of the theorem would no longer be true, and the limiting distribution of the TG p-value would revert to . (To be careful, here we would have cap the mixing probability at in order for the mixture to make sense, since the current definition of diverges with fixed and tending to .) In fact, this is ensured by our low-dimensional result in Theorem 7: after reformulating the many means problem in appropriate regression notation, all of the conditions of Theorem 7 are met by our current setup when is fixed. This is supported by the simulation in Figure 7.
The precise scaling is chosen since this implies , i.e., the extreme mixture components and each have probability tending to 0, an intuitively reasonable property for the error distribution. But we note that this scaling is not important for any other reason, and the proof would still remain correct if .
In Theorem 3 of Tian & Taylor 2017, the authors show that the TG statistic converges in distribution to a standard uniform random variable, in a high-dimensional problem setting, with some restrictions on the sequences of selection events that are allowed. One might ask what part of our high-dimensional setup here violates their conditions, because both results obviously cannot be true simultaneously. As far as we can tell, the issue lies in the role of in Assumption 1 of Tian & Taylor 2017. Namely, as we have defined the error distribution in (26), the value of needed to certify the third condition Assumption 1 of their work is too small for the main assumption in their Theorem 3 to hold. Hence Theorem 3 of Tian & Taylor 2017 does not apply to our current setup.
Discussion
We have studied the selective pivotal inference framework, with a focus on forward stepwise regression (FS), least angle regression (LAR), and the lasso, in regression problems with nonnormal errors. We have shown that the truncated Gaussian (TG) pivot is asymptotically robust in low-dimensional settings to departures from normality, in that it converges to a distribution (its pivotal distribution under normality), and does so uniformly over a broad class of nonnormal error distributions. When the error variance is unknown, we have proposed plug-in and bootstrap versions of the TG statistic, both of which yield provably conservative asymptotic p-values.
Our numerical experiments revealed that the statistics under theoretical investigation generally display excellent finite-sample performance, for highly nonnormal error distributions. These experiments also revealed findings not predicted by our theory: (i) the bootstrap TG statistic often produces shorter confidence intervals than those based on the plug-in TG statistic, and even the TG statistic that relies on the error variance ; and (ii) all three TG statistics show strong empirical properties well-outside of the classic homoskedastic, fixed regression setting that we presumed theoretically.
However, as we have clearly demonstrated, one should not hope for a convergence result in high dimensions that is as general as the result obtained in low dimensions. In a relatively simple many means problem, we showed the nonconvergence of the TG statistic to as , whereas in the same problem but with fixed, the TG statistic converges to its usual limit.
There is still much left to do in terms of understanding the behavior of selective pivotal inference tools that are constructed to have exact finite-sample guarantees under normality, like the TG statistic of Tibshirani et al. 2016, when applied in high-dimensional regression settings with nonnormal data. When the pivot, the central cog of this framework, is constructed under the assumption of normality, this creates robustness issues that are especially worrisome in high dimensions. Appendix A.16 provides a high-level discussion of some of these issues; a more detailed study will be the subject of future research.
We thank Jelena Markovic and Jonathan Taylor for many helpful discussions, and for their overall generosity. An initial version of our work contained only unconditional (i.e., marginal) results in the main theorems (Theorems 7, 11, and 12); Jelena Markovic pointed out that Theorem 7 should also hold conditionally, and the current version of this work has been revised accordingly.
Appendix A Appendix
We describe a modification of the conic conditioning set in Tibshirani et al. 2016 for FS. Our version is different in that we additionally condition on the sign of every active coefficient at every step, rather than just the coefficient of the variable to enter the model at each step. The modifications needed for the LAR and lasso conditioning sets, made on top of the sets for LAR and lasso given in Tibshirani et al. 2016, will follow similarly to that described for FS, and hence we omit the details.
where is the residual from regressing onto , and is the residual from regression onto . By expressing and , where projects onto the orthocomplement of the column space of , we can rewrite the above constraints as
a set of linear inequalities in . Meanwhile, the subevent can be characterized by inequalities expressed in block form,
A.2 Proof of Lemma 3
We prove the result for FS; the results for the LAR and lasso paths follows similarly, by inpsecting the form of the linear inequalities that determine their selection events.
Consider the first FS step as described in Appendix A.1. Multiplying through by , we see that an equivalent set of inequalities that characterize the selection event , is
This is clearly of the desired form , for a matrix dependent only on . At the th step of FS, there are two sets of inequalities to be examined: one that describes the variable to enter , and the second that describes the active signs . The first set, multiplying through by , is
while the second set, again multiplying through by , is
These inequalities are clearly all summarized by , where is a matrix that depends only on . This completes the proof.
A.3 Proof of Lemma 4
Under the conditions of the lemma, the TG pivot for fixed in (8) depends only on through the master statistic, because, as explained above the lemma, the only dependence in the pivot on is through the quantities , , , and each of these is in turn a function of the master statistic . Moreover, we may reexpress the TG statistic in (10) as
for some functions , or more succinctly, as , where
Note that the quantities , , depend smoothly on the master statistic at any point such that is nonsingular. This implies are smooth functions of at any point such that is nonsingular. Lastly, for all such that , we have , and thus the denominator of is positive. This proves the desired continuity result on .
A.4 Proof of Lemma 5
where we use to denote the submatrix of with rows in and columns in , and to denote the subvector of with entries in .
A.5 Proof of Lemma 6
Define , where , is the th row of , and , for . Note that , with independent, mean zero components. We compute
which converges to as , by assumption. Further, for any , consider
Now consider . Writing and for the standard normal CDF and density,
where the second line is due to the triangle inequality, and the third line is due to the simple bound , for any . Note that by the argument at the start of this proof, and by assumption in (17). This shows that converges in distribution to , uniformly over , and over .
Lastly, we establish the conditional result. By repeating the same arguments as above, the uniform Lindeberg-Feller central limit theorem and condition (17) imply that converges to , uniformly over , and over . Thus, along sequence , with , observe
at a rate that does not depend on the sequence in question. This is true because the numerator and denominator each converge to their normal probability counterparts, and the denominator remains bounded away from zero since has nonempty interior, and the set of limits of was assumed compact, in (17). Since was arbitrary, and the distribution of is continuous, we have (e.g., Lemma 2.11 in van der Vaart 1998)
And as the sequence , with was arbitrary, we have shown the desired uniform convergence.
A.6 Proof of Theorem 7
We begin with the proof of part (a). Let and . Also, let and . Recall that , by Lemma 3. Also, converges weakly to , uniformly over and over , by Lemma 6. As deterministically, we also have that converges uniformly in distribution to .
The choice of as specified in the theorem is now important for two reasons. First, by Lemma 4, we can express
The proof of part (b) follows from the expansion
As the number possible models is finite, we can simply apply the asymptotic pivotal result from part (a) to each to establish the asymptotic pivotal property of . The confidence interval result is again just a rearrangement of this pivotal property.
A.7 Proof of Lemma 8
By assumption, the vector can be written as
The denominator converges to by (14). The numerator satisfies
where is bounded by (21) and converges to by (14). This completes the proof.
A.8 Proof of Lemma 9
We start by proving the result about the event . First let us study its asymptotic probability marginally. Consider
where . Hence
for a constant only depending on . Hence, observe that
where in the last line we used (29). We consider individually. We have
where the second line again used (29), and the third used our assumptions on the error distribution in (22), and on in (24). We also have
For the second part, on the boundedness of , consider that for any we have
where the second and third lines used (31), and the last line used Rosenthal’s inequality (30), along with the abbreviations
A.9 Proof of Lemma 10
A.10 Proof of Theorem 11
First, we prove the result for the plug-in statistic. Denoting , we have
Consider the event , which has probability approaching 1 conditional on , uniformly over , and over , by Lemma 9. On this event, by the monotonicity of the truncated Gaussian survival function in its variance parameter, shown in Appendix A.11, we can replace by , and this cannot increase the value of the statistic. (To verify that the result in Appendix A.11 can indeed be applied, notice that , i.e., the left endpoint of the interval is at least the mean of the truncated Gaussian, which follows from the fact that by design.) Thus we can write
where the remainder term above is uniform over , over , and over . Applying part (a) of Theorem 7 proves the conditional result for the plug-in statistic.
Next, we turn to the bootstrap result, whose proof is a little more involved. Define a function
Rewriting the result in the last display, we have
where denotes the negative part of . In particular, at , this implies
Finally, this means that we can write, at an arbitrary level ,
where the term above is uniform over , over , and over . Applying part (a) of Theorem 7 proves the conditional result for bootstrap statistic.
The unconditional results for two modified TG statistics hold simply by marginalization.
A.11 Monotonicity of the truncated Gaussian distribution in σ2\sigma^{2}
the survival function for a normal random variable , truncated to lie in an interval , where . We will show, following the proof of a similar monotonicity result in Lemma A.1 of Lee et al. 2016, that for any ,
To emphasize, the above property is only true when the interval lies to the right of 0. Without this restriction, the survival function will not be monotone increasing in (if contains 0, then it will generally be nonmonotone, and if lies to the left of 0, then it will actually be monotone decreasing).
Over , the family of distributions forms an exponential family with natural parameter , as it is just a family of Gaussian distributions with the carrier measure changed. Therefore, it has a monotone likelihood ratio in its sufficient statistic , i.e., if we denote by the truncated Gaussian density function, and we fix , and , then
Integrating with respect to over , for some , we obtain
Now integrating with respect to , over , we obtain
A.12 P-value examples for correlated predictors
Figure 8 shows the results, in the same format as Figure 3: p-values for LAR steps 1, 2, and 3, and pivotal statistics aggregated over LAR steps, from 500 repetitions. The p-values at steps 1 and 2 were restricted to repetitions in which either variable 1 or 2 were selected (now comprising about 70% and 60% of the repetitions, respectively); the p-values at step 3 were restricted to repetitions in which one of variables 3 through 10 was selected (comprising about 80% of the repetitions). Similar to the display in Figure 3, we see power in the p-values from steps 1 and 2, albeit less power than in the uncorrelated case, and uniform p-values in step 3, as well as uniform pivotal statistics.
A.13 Confidence intervals for uniform, Laplace, and skew normal noise
Figures 9 through 11 show sample confidence intervals for the problem setting of Section 6.2, when the error distribution is uniform, Laplace, and skew normal, respectively.
A.14 Confidence interval summary statistics for correlated predictors
Table 2 gives summary statistics of confidence intervals obtained by inverting the original TG, plug-in TG, and bootstrap TG statistics, as in Table 1 of Section 6.2, but for the correlated predictors setup described in Section A.12.
A.15 Proof of Theorem 12
Let us denote by the number of observations in the th column of the data array , , that are drawn from the mixture component. Similarly, let denote the number of observations in the th column drawn from the mixture component. Then we will define to be the event
In words, is the event that exactly one column has all of its observations drawn from , and each of the rest of the columns have at least observations from . We calculate
where in the second line we used that by construction, and introduced the notation for the number of observations in column that are drawn from the mixture component; in the third line we used Markov’s inequality.
On the event , intersected with an event whose probability tends to one, we have , and furthermore
where denote standard normals. We note that the ultimate bounds on the right-hand sides in the two lines above are extremely loose, but will suffice for our purposes. Hence using Mills’ ratio, we can bound the TG statistic on the event in consideration by
for sufficiently large . But on this same event we have that
and it is straightforward to check that the right-hand side of the bound above diverges to , given our assumptions on . Therefore, we have shown that on an event whose probability tends to at least , the TG statistic converges to 0.
A.16 Some thoughts on instability in high dimensions
The TG statistic is defined by the ratio of normal tail probabilities. If the dimension is large (in which case we are searching through a large space of models), or there are some large effects, then we often find ourselves evaluating the pivot far into the tails. The point of evaluation is given by a linear function of the data, which should itself converge to a Gaussian distribution (at least when is finite). But even a small amount of non-Gaussianity is magnified when we are in the tails. To see this, consider the function
The left plot in Figure 12 shows two densities and which are nearly indistinguishable. The right plot shows their corresponding tail functions and . Even though and are close, we see that and are quite different. The message is that any inferential method that depends heavily on extreme tail behavior could be unreliable.