Regularization, sparse recovery, and median-of-means tournaments
Gábor Lugosi, Shahar Mendelson
Introduction
However, in statistical problems, the joint distribution of is unknown and the regression function is impossible to compute. Instead, a sample of independent copies of the pair is available (such that and the pair are independent).
has an acceptable performance, and the standard assumption is that the minimum is attained and is unique. We assume that is a closed and convex subset of —where denotes the distribution of —, guaranteeing the existence and uniqueness of .
The quality of a learning procedure is typically measured by the mean squared error, which is the conditional expectation
where, for , we use the notation
A closely related, though not equivalent, measure of performance is the excess risk, defined by the conditional expectation
The goal of a statistical learning problem is to find a learning procedure that achieves a good accuracy with a high confidence. In particular, for and , we say that a procedure performs with accuracy parameter with confidence in the class (e.g., for the mean squared error) if
High accuracy and high confidence (i.e., small and small ) in the given class are obviously conflicting requirements. The achievable tradeoff has been thoroughly studied and it is fairly well understood. We refer the reader to Lecué and Mendelson , Lugosi and Mendelson for recent accounts.
The most standard approach for a learning procedure is empirical risk minimization (erm), also known as least squares regression in which
(where we assume that the minimum is achieved). One may show (see, e.g., Lecué and Mendelson ) that unless the function class and target are sub-GaussianHere, Sub-Gaussian means that the and the norms are equivalent in ; that is, there is a constant such that for any and any , , and that the same holds for any for any ., then empirical risk minimization is far from achieving the optimal accuracy/confidence tradeoff. The reason for the suboptimal behaviour of empirical risk minimization is that outliers distort the empirical means unless the problem is very close to being Gaussian. Thankfully, learning procedures that can tackle heavy-tailed problems exist, as it was recently pointed out by Lugosi and Mendelson with the introduction of the median-of-means tournament.
A common problem that all learning procedures encounter is that of overfitting, which occurs when the underlying class is too big relative to the (random) information at the learner’s disposal. A standard way of dealing with learning problems involving classes that are too large is giving priority to functions in the class according to some prior belief of “simplicity”. For example, in regularized risk minimization, one selects a norm defined on a vector space containing . A small value of is interpreted as simplicity and simple functions are given priority by way of adding a penalty term to the empirical risk that is proportional to . In particular, for some regularization parameter , a regularized risk minimizer selects
and the term is sometimes called the penalty.
Just as the tournament procedure from outperforms empirical risk minimization (in fact, the tournament procedure attains the optimal tradeoff between accuracy and confidence under minimal assumptions), the regularized tournament which we present here, outperforms regularized risk minimization. Since regularized procedures require the minimization of a functional that has the empirical mean as a component, they suffer from the same disadvantages as empirical risk minimization. Therefore, the accuracy/confidence tradeoff exhibited by regularized risk minimization is suboptimal once one leaves the sub-Gaussian realm, and deteriorates further if the problem is more heavy-tailed. In contrast, we show that the regularized tournament attains the optimal accuracy/confidence tradeoff under rather minimal conditions and, in particular, in heavy-tailed problems.
The paper is organized as follows. In Section 2 we introduce a new regularized “tournament” procedure in a quite general framework and illustrate how it works on an important specific case, the tournament lasso (see Section 2.1). In Section 3 the main general performance bound is presented for the regularized tournament procedure under certain specific choice of the parameters of the procedure. The proof of the main result is detailed in Section 4. Finally, in Section 5 two examples are worked out. The first is a “tournament” version of lasso (introduced in Section 2.1) and the second is the tournament slope, a generalized version of tournament lasso.
The procedure
Let us now describe the regularized tournament procedure. Recall that the learner is given a closed and convex class of functions and a regularization function which is assumed to be a norm on .
The first three stages of the procedure use independent data. In order to accommodate this, one needs to split the available data into three independent parts. For simplicity of the presentation, we assume that these parts have equal size, each containing samples. (Thus, the total sample size is rather than but this change of convention only affects the constants in the bounds that we do not make explicit in any case.)
split to disjoint blocks of equal size, denoted by ;
More accurately, the second phase is defined as follows:
(observe that it is possible that and at the same time);
The fourth phase: naming a winner
The first three phases are the key components of the procedure. Out of the three, the first one is an adaptation of the distance oracle used in and which had been introduced in ; the third component is essentially the same as the champions league stage in the tournament procedure from .
The truly new component is the second phase. Its analysis combines ideas from (which focused on regularized risk minimization in ‘sparse’ problems) and from . As it is the main novelty in this article we present it in detail and only sketch the arguments needed in the analysis of the other components.
1 The tournament lasso
The problem with the lasso is that when either the class members or the target are heavy-tailed, the tradeoff between the accuracy with which the lasso performs and the confidence with which that accuracy is attained is far from optimal. That suboptimal tradeoff is what the tournament lasso aims to remedy.
Let us go through the four phases of the tournament lasso.
In what follows, denote appropriately chosen constants whose value depends only on . (The precise form may be extracted from the analysis but it is of secondary importance for our purpose.)
The fourth phase
The following theorem, proved in Section 5.1, summarizes the performance of the tournament lasso.
Theorem 2.6 shows that the tournament lasso attains the optimal accuracy/confidence tradeoff even though can be heavy-tailed. In fact, the estimate is what one would expect in the most friendly of scenarios: if were a sub-Gaussian random vector (i.e., linear forms exhibiting a norm equivalence with constant rather than an moment equivalence going only up to ), and were a Gaussian random variable, independent of . The lasso does not come close to such an accuracy/confidence tradeoff under the weak moment assumption of Theorem 2.6.
Note that is the best accuracy parameter one can hope for even if the learner knows that is -sparse and and are Gaussian. The difference between the performance of the tournament lasso and the standard lasso can be seen in the confidence with which this accuracy is attained. In the situation described in Theorem 2.6, the standard lasso performs with that accuracy parameter only with constant confidence, because all that we assume on is that it is square-integrable. In contrast, the tournament lasso attains the accuracy (2.6) with the optimal exponential probability estimate (2.4).
The tournament lasso does not require prior information on the degree of sparsity of to be carried out, but its success does depend on having a large-enough sample and on that is in the right range. The former is a constraint that any recovery procedure faces while the latter is easily achieved by running the procedure for for a large initial value of , followed by a standard validation argument at each step.
The one item that does require extra attention is that an upper estimate on is used in the choice of parameters of the tournament lasso. At times one is simply given that information; this is often the case in signal processing problems, where the nature of the ‘noise’ is known to the learner. If not, one may use the data-dependent procedure from which holds for more general noise models: it leads to upper and lower estimates on that are sharp up to absolute multiplicative constants and under minimal assumptions.
Of course, proving Theorem 2.6 requires some work, and the choice of parameters used in the first three phases has to be clarified. We explain the choice in the general case in the next two sections and return to the example of the tournament lasso in Section 5.
The main result
In the general setup we study, we merely assume a rather weak fourth-moment assumption. More precisely, we work under the following conditions.
Let be a locally compact, convex class of functions. Let and assume that, for some constant ,
for every , ;
for a known value .
The condition that may easily be replaced by a combination of two assumptions: that for every , ; and that for some known constant . Also, in the case of independent additive noise, that is, when where and that is mean-zero, square-integrable and independent of , the assumption that may be replaced by the weaker one, that for a known constant .
The necessary modifications to the proofs are straightforward and we do not explore this observation further. Also, as noted previously, we refer the reader to for a data-dependent procedure of estimating which may be easily modified to an estimate on . Since that is not the main focus of this paper we do not pursue it further and instead assume that the learner has access to or to .
The complexity is measured in terms of four parameters, depending both on the class and the distribution of . The four play an essential role in describing the optimal performance of learning procedures and for detailed discussion on the meaning we refer to Mendelson and Lugosi and Mendelson .
Before we define the four parameters we need some notation. Denote the unit ball in by and let be the unit sphere. For and , we write . In a similar fashion for the norm used as a regularization function, let , set and .
In what follows we make two important modifications to the definitions of the complexity parameters used in . First, just like in the above-mentioned articles, we are interested in “localized” classes. However, because regularized procedures are affected by two norms, and , the localization has to be with respect to both of them. Therefore, the “localization” of , centred in and of radii is defined by
The second minor modification is that each complexity parameter is associated with the ‘worse case’ centre for some fixed , and not necessarily with the whole of .
Two of the four parameters are defined using the notion of packing numbers.
Given a set and , denote the -packing number of by . In other words, is the maximal cardinality of a subset , for which for every .
Fix and . For , set
For let
Fix and . Let , , and define
Also, for let
For the remaining two complexity parameters, let be independent, symmetric -valued random variables that are independent of .
Fix and . For let
and for set .
Finally, suppose that the distribution of is such that for a known constant . The “complexity” of relative to centres in and radius is
Here are appropriate positive numerical constants. (“Appropriate” means that satisfies Propositions 4.1, 4.4 and 4.7 below). The existence of such constants is proved in when , i.e., when any function in is a ‘legal choice’ of a centre, and under Assumption 3.1. In that case, the constants depend only on the value of .
When and are clear from the context, we simply write for .
2 Properties of the hierarchy
Recall that is a (convex) subset of a normed space ; is also a subspace of , though the norms and may have nothing to do with each other. Let and denote the unit ball and unit sphere in the dual space to , respectively. Therefore, consists of all the linear functionals for which . A linear functional is a norming functional for if .
Let be the collection of functionals that are norming for some . Set
where the inner infimum is taken in the set
A lower bound on the term plays an essential role in the study of the elimination phase of the regularized tournament, when one has to compare
(1) is a finite hierarchy;
We are now ready to specify the parameters used in the definition of a regularized tournament.
Let , , , and be well chosen constants that depend only on the norm equivalence constant from Assumption 3.1, and assume that one has access to the value from that assumption.
We set the following choice of parameters:
With these choices set in place, let us formulate the main result of this article.
3 Discussion
We emphasize that Theorem 3.9 is quite general though finding the adequate parameters of the procedure requires additional work. For the “tournament” version of lasso and slope we work out the details in Section 5 under certain assumptions (such as isotropic design vector and approximately sparse linear regression function) for illustration. Some of these assumption may be weakened but we prefer to keep the presentation as simple as possible.
Related work. The sensitivity of empirical risk minimization (or least squares regression) to heavy-tailed distributions has been pointed out and several proposals of robust regression function estimates have been made that avoid this sensitivity. We refer to Audibert and Catoni , Hsu and Sabato , Lerasle and Oliveira , Minsker , Brownlees, Joly, and Lugosi , Lugosi and Mendelson for a sample of the literature. This paper mostly builds upon the methodology of median-of-means tournaments, developed in , (see also Lugosi and Mendelson ). Here we extend this methodology to the analysis of regularized robust risk minimization similarly to how the paper of Lecué and Mendelson analyzes standard regularized risk minimization. The analysis of the lasso and slope procedures of was extended and generalized by Bellec, Lecué, and Tsybakov . In an independent parallel work to ours, and building on the arguments developed in , Lecué and Lerasle point out a connection of median-of-means tournaments to Le Cam’s estimators, develop a version of lasso–the so-called mom-lasso–and prove a performance bound quite similar to Theorem 2.6.
Analyzing the four phases
where is a median of the values .
The behaviour of described below has been established in Mendelson (see also Lugosi and Mendelson ):
Let satisfy Assumption 3.1. There exist constants , , and and , all of them depending only on , for which the following holds.
Then, with probability at least , for any that satisfies ,
if then
if then .
Although Proposition 4.1 is formulated for a designated single centre , it is straightforward to extend it to any centre in and obtain a uniform distance oracle that holds for any pair .
for a well-chosen constant that depends only on the equivalence constant from Assumption 3.1. We also set
with the choices of both constants and specified below.
The function defeats (denoted by if
1 Proof of Proposition 4.4—highlights
To explain why this elimination phase preforms well even when is very large, define, for each block (),
Note that the regularized empirical excess risk of on block is .
which is the natural decomposition of the empirical excess risk functional into its quadratic and multiplier components. Setting
The first observation we require is a version of a deterministic result from [8, Theorem 3.2] (see the appendix for the proof).
and if also then
Moreover, if such that for then also
The fact that we have the required control over coordinate blocks is formulated in the following lemma. Its proof may be found in the appendix.
Recall that the function wins its home match against (denoted by ) if
Combining all these observations, Corollary 4.8 describes the outcome of the first three phases in the regularized tournament procedure, for each member of the hierarchy.
Selection of a final winner
which completes the proof of Theorem 3.9.
Examples
In what follows we present two examples: a tournament version of lasso, and also, a tournament version of another popular sparse recovery procedure—slope.
the value of the constant is given in Theorem 5.2.
Assume further that for a known constant .
In other words, Assumption 5.1 means that linear forms satisfy a sub-Gaussian moment growth, but only up to a rather low exponent—logarithmic in the dimension of the underlying space. This moment assumption is a sufficient and almost necessary condition for the celebrated basis pursuit procedure to have a unique minimizer (see ), and as such, it is a natural assumption when studying such sparsity-driven bounds. Note that even with Assumption 5.1 replacing Assumption 3.1, the fact that the ‘noise’ may only be in means that there is not hope that
exhibits a fast tail decay, even in the extreme case when . This indicates why regularized risk minimization can only perform with a rather weak accuracy/confidence tradeoff in such situations.
On the other hand, (5.1) suffices to obtain bounds on the expectation of empirical and multiplier processes, as long as the indexing set has enough symmetries, and a suitable bound on the expectation suffices for the analysis of regularized tournaments.
Given a vector , let be the non-increasing rearrangement of .
The following fact has been established in :
There exists an absolute constant and for , and there exists a constant that depends only on , and for which the following holds. Consider
for some ,
If are independent copies of then
Therefore, as long as is sufficiently symmetric and linear forms exhibit a sub-Gaussian moment growth up to , the expectations of empirical and multiplier processes indexed by behave as if were the standard Gaussian vector and were independent of . In the cases we are interested in the indexing sets have enough symmetries, and since , the conditions of Theorem 5.2 hold for .
In this section we prove Theorem 2.6, the performance bound of the “tournament lasso” procedure.
Note that for any and every ,
respectively. The indexing set is invariant under coordinate permutations and sign reflections, and therefore satisfies the conditions of Theorem 5.2. Hence, an upper bound on follows if
Equations (5.4) and (5.5) cannot be improved; they are tight bounds on (5.2) and (5.3) when, for example, and is a Gaussian variable that is independent of .
which is precisely the type of condition in (5.4).
On the other hand, there is a functional that is norming for both and ; hence,
This shows that, as long as the ratio is larger than the square-root of the degree of sparsity of vectors we are interested in, as our procedure requires. A similar observation is true if is not sparse, but rather well approximated by an -sparse vector (see for a detailed argument).
Set and assume without loss of generality that is an integer. We also restrict ourselves to values , intuitively because the above implies that should capture the degree of sparsity. Recall that
(see, e.g. for the standard proof). Hence, (5.4) becomes
We consider only the case , which is the more interesting range in sparse recovery—when the number of given linear measurements is significantly smaller than the dimension of the underlying space. An argument following the same path may be used when and we omit it.
It follows from a rather tedious computation that (5.6) holds provided that
as long as .
Using the constraint that , it is evident from (5.8) that
, and that
Therefore, to have a ‘legal’ choice of and , we must have
This naturally leads to the choices made in Section 2.1: set
2 The tournament slope
As a second example, we present and analyze a “tournament” version of the regularized risk minimization procedure slope. slope is defined using a set of non-increasing weights . The corresponding norm is
where as always, denotes the non-increasing rearrangement of . Clearly, slope is a generalized version of lasso, as the latter is given by the choice for .
Just like the lasso, most of the known results on the performance of slope hold only when both the random vector and the target have well behaved tails.
The tournament slope we present below is defined for the penalty , where . We obtain the following performance bound:
the tournament slope produces that satisfies
The estimate corresponds to the optimal accuracy/confidence tradeoff any procedure can attain even if the learner knows that is -sparse. Moreover, in the heavy-tailed situations we study here, the performance of slope is significantly weaker than in Theorem 5.3.
The argument we use here is similar to the one used for the tournament lasso, and so we skip most of the details.
In tournament slope one selects and therefore the corresponding indexing set is contained in
Because has enough symmetries, one may apply Theorem 5.2, leading to an upper bound on when
Next, one may verify (see Lemma 4.3 in ) that if we set and if , then for centres that are ‘well approximated’ by -sparse vectors. Also, for our choice of , . Hence, for a fixed degree of sparsity , one has the constraint that
for a constant that depends only on .
Following the same path used for the tournament lasso let
Appendix A Additional proofs
The proofs of Lemma 4.5 and Lemma 4.6 are, in fact, the same as in and , respectively. The minor modifications to the original proofs are presented in this appendix solely for the sake of completeness and not in full detail.
The proof of Lemma 4.5 follows the same path as that of Theorem 3.2 in . Let us begin by examining
Fix and write ; thus . Set to be a linear functional that is norming for and observe that for any ,
In other words, if is chosen to satisfy both (A.2) and (A.5), and , it follows that
Next, if , there are and that satisfy
If , then by the triangle inequality for followed by (A),
If, on the other hand, , then
Now, all that remains is to control and show that if , then
Proof of Lemma 4.6
The first part of Lemma 4.6 is identical to Lemma 5.1 from , with the trivial modification that the constant replaces used in . The second part of Lemma 4.6 was not needed in , but its proof follows the same path as Lemma 5.1 from .
and by a straightforward symmetrization argument,
Applying Assumption 3.1, it is evident that
where we use the fact that and select and . Therefore,
and with probability at least ,
The rest of the argument is identical to the proof of Lemma 5.1 from : let be a maximal separated subset of with respect to the norm, of cardinality , and with the following property: for any there is for which
here denotes the mesh of the net. The existence of such a separated set is established in (see Lemma 5.3), and one may show that the mesh is a small proportion of .
By (A.6), we have that with probability at least , for every
For every let be as in (A.7), and at the heart of the proof of Lemma 5.4 in is that with probability at least ,
Combining (A.8) and (A.9), there is an event of probability at least on which for any there is a set of coordinate blocks , of cardinality and for ,
Acknowledgements
We thank the referees for valuable suggestions that significantly helped us improve the presentation.