Learning subgaussian classes : Upper and minimax bounds
Guillaume Lecué, Shahar Mendelson
Preface
Most the results contained in this note have been presented at the SMF meeting, which took place in May 2011; the rest have been obtained shortly after the time of the meeting.
The question we study has to do with the optimality of Empirical Risk Minimization as a learning procedure in a convex class – when the problem is subgaussian. Subgaussian learning problems are a natural object because they are the simplest unbounded learning scenarios. However, an additional reason for studying such problems was that at the time of the SMF meeting, the technical machinery required for the analysis of more heavy-tailed problems was simply not known. Since 2011, significant progress has been made in the understanding of learning problems in heavy-tailed situations , though this progress does not make the results presented here obsolete. We show that ERM performed in a convex class is an optimal learning procedure (in a sense that will be clarified) when the learning problem is subgaussian. This happens to be a rather special feature of subgaussian learning problems, and under weaker tail assumptions ERM fails to deliver the optimal accuracy/confidence trade-off at the high level of accuracy we are interested in here.
The results presented here are complemented in , which also focuses on subgaussian learning problems and addresses some of the cases that have not been resolved in this note.
Introduction and main results
In the classical statistics setup, one usually assumes that the regression function of given belongs to some particular function space (called a statistical model). In contrast, in the learning setup on which we focus here, one is given a function class (sometimes, called a model as well), and the goal is to construct a procedure that satisfies a sharp or exact oracle inequality (following ; such bounds are called excess risk bounds in and ). An exact oracle inequality ensures that with high probability,
and one would like to make the residue in (1.1) as small as possible.
For the sake of simplicity, we assume that there is some minimizing the risk in (though the claims presented here remain true even without that assumption), and we set
Note that in (1.1) the performance of the procedure is compared to the best performance possible in , i.e., to the risk of the best element . This exhibits the point of view of Learning Theory, where one wishes to identify a function that is almost as good as the best possible in , regardless of whether the best function in has a small risk. It is different from typical questions in classical Statistics, where a statistical model is given and the risk of an estimator is compared to the one of the regression function (or Bayes rule). The latter are usually called excess risk bounds (cf. ) and are actually very different from exact oracle inequalities like (1.1) (see, for example, or Chapter 1.3 in for more details on those differences).
The performance of a procedure is measured relative to a set of admissible targets in some class of random variables . Naturally, one would like to make as large as possible, for example, all random variables bounded by , all the random variables in for some , or a similar weak condition of that flavor.
Clearly, while the true risk of is not known, simply because and are not known, one still has access to its empirical counterpart:
Thus, a natural procedure that comes to mind is finding a function in that best fits the data: a minimizer of the empirical risk in . This procedure is called empirical risk minimization (ERM) and is defined by
ERM has been studied extensively over the last years (see, e.g. , , and references therein), and the main goal has always been to identify connections between the structure of and the accuracy and confidence that ERM yields, while trying to minimize the restrictions on . Among the natural questions regarding the performance of ERM are:
1. Given any confidence parameter , what is the error rate that one may obtain using ERM, and what features of govern that rate?
2. Given any , is ERM an optimal procedure for the confidence level ? In other words, is there a procedure that can perform with a better accuracy than ERM, given the same confidence level?
and set .
There exist absolute constants and for which the following holds. If consists of functions that are bounded by and is a convex class of functions that are bounded by , then for any and every , with probability at least ,
A result of a similar flavor was obtained in : let be the number of translates of needed to cover . Set to be the unit ball in and let
for absolute constants .
The result in is that under various assumptions on the class (assumptions that allow one to upper bound the function using the entropy integral in (1.4)), may serve as a residual term.
These two facts rely heavily on the assumption that and are bounded in and their proofs do not extend beyond the bounded case.
Let be a probability measure and let be distributed according to . The -norm of a function is
The space of functions with a finite -norm is denoted by .
A function class is -subgaussian with respect to the probability measure if for every , .
Note that for any , . A class is a subgaussian class when the reverse inequality holds, and in particular when the and norms are equivalent on .
Note that norm equivalence is very different from being bounded. Having such a norm equivalence implies that on a relatively large event. In contrast, even though a bounded function has a finite norm (by selecting in the definition of the norm), the fact that is bounded does not mean that is equivalent to , nor that on a relatively large event. Because of the substantial difference between the two notions, one should not expect that learning procedures exhibit the same performance when one assumes that is bounded in or when the and norms are equivalent on .
where here, and throughout this note we write if for an absolute constant . Thus, the measure associated with the random vector is -subgaussian. Also, the measure is clearly isotropic.
By Khintchine’s inequality (see, for example, ),
If is a mean-zero, variance one, -subgaussian random variable, and is a matrix whose coordinates are independent copies of , then defines a subgaussian, isotropic measure on the space of matrices of the right dimensions, relative to the natural trace inner product. The same holds if has independent rows, distributed according to an isotropic, -subgaussian random vector. The proof of both facts is straightforward and are omitted.
The strategy we use here for the study of ERM is the isomorphic method, introduced in and analyzed there in the bounded setup. Before presenting it, recall that the excess loss of is
A rather obvious but very useful observation is that for every , , while the empirical minimizer satisfies that .
The isomorphic method is based on the following idea. Consider an event , on which for every function in the set ,
It follows that on , ERM produces that satisfies
because ; therefore, .
Consequently, an exact oracle inequality with a confidence parameter may be derived by identifying for which has probability at least ; that is, the level for which
with probability at least (see Theorem 4.4 in for results of a similar flavor).
Note that only the lower estimate in (1.6) is needed for the argument outlined above to work. This observation is the key in the application of the recent works on the small-ball method in learning theory (cf. ), which allows one to deal with heavy-tailed scenarios that are far more general than subgaussian problems.
Just like in (1.3) and in (1.4) – and many other well known estimates on the performance of ERM (e.g. ) – the residual term we use is defined in terms of fixed points. Unlike and , the geometric complexity measure we use here is based on gaussian averages associated with localizations of the class. We refer the reader to Chapter 12 in for more details on gaussian processes (in particular to Theorem 12.1.3 for the existence of such a process and to Theorem 12.1.4 for its linearity).
This supremum is called the lattice supremum (see Chapter 2.2 in for more details).
We are now in a position to introduce the two complexity parameters that will serve as residual terms in the exact oracle inequalities satisfied by ERM.
For any , set and . For every , let
In what follows we will always assume without mentioning it explicitly that the sets in (1.7) and (1.8) are nonempty (for example, this forces that ).
With these definitions in place, one may formulate a restricted version of the upper bound on the performance of ERM – for a convex, -subgaussian class of functions.
Theorem A. For every there exist constants and that depend only on for which the following holds. Let be a convex, -subgaussian class of functions, assume that and set and .
1. If then with probability at least ,
2. If then with probability at least ,
Hence, with probability at least ,
Once noise is introduced to the problem and passes a certain threshold, it is no longer realistic to expect that an intrinsic parameter, which does not depend on the noise level, can serve as an upper bound. And, indeed, measures the interaction between the ‘noise’We keep the terminology from Statistics: the difference between the output variable and the target function is called the noise. This coincides with the classical definition of noise in Statistics when is the regression function. and the class through the choice of . Thus, beyond a certain noise-level , which depends on the ‘complexity’ of the class , becomes the dominant term in the upper bound.
Note that in the free-noise case, , one has . Therefore, the error rate of ERM depends only on . Also, when the number of observations is large enough, one also has , leading to exact reconstruction.
Of course, Theorem A would be better justified if one could obtain matching lower bounds, showing that ERM is an optimal procedure for subgaussian problems. To that end, it seems natural to employ minimax theory (see, e.g., for more details on minimax bounds).
What is a reasonable way of identifying a lower bound on the performance of a learning procedure is to see what accuracy and confidence it can guarantee for a minimal set of admissible targets , and a natural choice of a minimal set of targets is
for every and that is a centered gaussian random variable that has variance and is independent of . Thus, this minimal set of targets consists of ‘independent perturbations’ of realizable learning problems, and thus is arguably the smallest set of ‘noisy’ targets. The minimax rate is (at least) the best accuracy/confidence trade-off that a learning procedure may attain in for the set targets (1.9). Our main focus will be on the accuracy/confidence tradeoff for the accuracy level described in Theorem A.
Standard minimax bounds are based on information-theoretical results such as Fano’s Lemma, Assouad’s Lemma or Pinsker’s inequalities. Unfortunately, these results do not yield lower bounds in the high probability realm of Theorem A; rather, these results are restricted to constant confidence or hold in expectation. To treat the high probability regime, we present a new minimax bound that is based on the gaussian shift theorem (and therefore on the gaussian isoperimetric inequality).
Note that no assumption on the underlying measure is required in Theorem A′. Moreover, Theorem A′ makes a natural connection between accuracy and confidence: the higher the confidence the larger must be.
An important outcome of Theorem A and Theorem A′ is that for the set of admissible targets as in (1.9), and as long as the class is convex and -subgaussian, ERM is optimal in the following sense:
Theorem A′′. There exist absolute constants for which the following holds. Let be a convex, -subgaussian class of functions and consider the set of admissible targets as in (1.9). Set and . If then for any target , the ERM satisfies
Thus, up to the constant in the exponent, the upper bound and the lower bound match and the ERM achieves this bound.
The second question we wish to address is what happens when the desired confidence is an absolute constant – for example, when is, say, , but the noise level is nontrivial in the sense that dominates . We will show that in such a situation, Theorem A is optimal in a minimax sense under some regularity assumptions on . This complements Theorem A′′ which proves the optimality of ERM (under no extra structural assumption) in the high probability case – when .
To explore the constant confidence regime, let us consider the ‘Sudakov analog’ of the gaussian-based parameter : recall that by Sudakov’s inequality (see, for example, ), for any ,
Put and set
Theorem B is known, and may be derived from Theorem 2.5 in or from . The proof presented here is new, and follows the same path as the proof of Theorem A′.
With Theorem A in mind, Theorem B implies that if the learning problem is subgaussian, and are equivalent for and , the minimax rate in the constant probability regime is attained by ERM.
Finally, let us consider the low-noise case, in which . Although it is not clear if is an optimal bound in that range (except when ), it turns out that it is not far from optimal.
with the probability taken with respect to the product measures endowed by .
and by Theorem C, is a lower bound on the minimax rate in the constant confidence regime. Therefore, when , it follows that for every , is the constant-probability minimax rate, and that rate is achieved by ERM.
It should be noted that although our presentation focuses on oracle inequalities in a given class, oracle inequalities for model selection and regularized procedures can be derived from the isomorphic method in general, specifically, from Theorem 2.8 below. This strategy is rather standard and has been used, for example, in , in Chapter 3.6 of or recently in . We will not present results on regularization methods or model selection methods in what follows since those may be easily obtained from results on ERM.
We end this introduction with a word about notation. Throughout, absolute constants or constants that depend on other parameters are denoted by , , , , etc., (and, of course, we will specify when a constant is absolute and when it depends on other parameters); their values may change from line to line. The notation (resp. ) means that there exist absolute constants for which (resp. ). If is a parameter then means that for some constant that depends only on .
The proofs of our main results are presented in the next two sections. We then present several examples of applications of those results, in which the rates established in Theorem A are shown to be sharp in both the high and constant confidence regimes. The final section contains some concluding remarks.
Proof of Theorem A
The proof of Theorem A shows that it is more general than stated. Rather than convexity, the two properties that are actually needed are the following:
A class is star-shaped around if for every , the interval is contained in .
We will assume that is star-shaped around , otherwise, one may consider the star-shaped hull of with , that is, the set
which is not much larger than . The second property required is a variant of the Bernstein condition (cf. ).
A class is -Bernstein relative to the target , if for every ,
In what follows, we shall assume that is star-shaped around and that satisfies the Bernstein condition (2.1).
The next lemma (which will be proved in the Appendix) shows that the assumption that is star-shaped around adds some regularity to the gaussian process .
For and any , , and for any , .
Let . For any , and for any , .
A straightforward outcome of Lemma 2.3 which will be used later is as follows:
Let , set and consider and as introduced in Definition 1.6.
If then , and if then .
If then .
The proof of Lemma 2.4 will also be presented in the Appendix.
When considering the parameters and , what may seem odd at first glance is the different normalization in their definition – the first condition is linear, while the second is quadratic. The two originate from the need to compare the way in which two processes, the quadratic component and the multiplier component of the excess loss functional scale with . Indeed, note that
and that is the source of the seemingly less-natural normalization in the definition of .
Let us begin with an estimate on the quadratic component, which is based on a functional Bernstein type inequality (see ).
There exist absolute constants and for which the following holds. Let be an -subgaussian class. For every , with probability at least ,
The following result is a straightforward application of Theorem 2.5 and illustrates the role of .
There exist absolute constants and for which the following holds. Let be an -subgaussian class, assume that is star-shaped around and let . If and , then with probability at least 1-2\exp\big{(}-c_{1}Q^{2}N\big{)},
Remark. Using the notation of Lemma 2.5, consider . If (2.3) holds then for every that satisfies , one clearly has
The second ingredient required for the proof of Theorem A is a bound on multiplier processes.
[Theorem 4.4 in ] There exist absolute constants and for which the following holds. If is an -subgaussian class and , then for every , and every integer , with probability at least
Combining the estimates on the quadratic and multiplier process leads to the following ratio estimate:
For every and there exist constants and that depend only on and for which the following holds. Let be an -subgaussian class that is -Bernstein relative to the target . Assume that is star-shaped around and that . Set and .
1. If , then with probability at least ,
2. If , then with probability at least ,
Fix and let . Since satisfies the -Bernstein condition relative to , it follows that for every , . Moreover, if then
and . Recall that is star-shaped around , and by (2.5) one has that
Fix and for suitable absolute constants and . Set and note that by Lemma 2.4, if then and , and if then ; also .
First, consider the case . Applying Lemma 2.6 for , it follows that with probability at least
provided that . Moreover, by (2.4), and because , one has that with probability at least
Thus, for any and , if then with probability at least , the following holds: for every that satisfies that ,
Next, let us consider that case which follows a very similar path to the first case. Recall that . Setting , it follows from Lemma 2.6 and (2.4) that with probability at least
as long as and . The claim now follows because and by the choice of , namely, that .
Theorem A is an immediate outcome of Theorem 2.8 for and the isomorphic method described in the introduction.
Minimax lower bounds (proofs of Theorem A′, B and C)
where .
The first estimate presented here is the high probability lower bound, formulated in Theorem A′.
The main component in the proof of Lemma 3.3 is a version of the gaussian shift theorem.
Let , and set . Using the notation of Theorem 3.4, the corresponding halfspace is
and the claim follows from Theorem 3.4 and the definition of .
Next, let us turn to the proof of Theorem B, which is a straightforward application of the next observation:
Proof. Observe that if then and Theorem 3.5 is trivially true. Hence, one may assume that .
Let , set and put to be a maximal -separated subset of with respect to the norm. Thus, is a family of disjoint subsets of .
where is a density function of a the standard gaussian and
and it remains to lower bound each expectation.
Another application of Chebyshev’s inequality shows that with -probability at least ,
because . Therefore, with -probability at least ,
and since ,
Thus, by (3.2), 1\gtrsim|\Lambda|\exp\big{(}-c_{3}N\theta^{2}a^{2}/\sigma^{2}\big{)}, as claimed.
Taking the expectation in () with respect to ,
and if , the error rate obtained in Theorem A is the minimax rate in the constant probability range.
Examples
In this section, we present two examples in which our results lead to sharp upper and lower minimax bounds, thus showing the optimality (in some minimax sense) of ERM.
and if then and
Setting and , it is straightforward to verify that
where and are constants that depend only on .
When , decays rapidly from to . Thus, when one only has an upper estimate on , and we will therefore only consider the cases and .
Let us present the exact oracle inequalities satisfied by the ERM in that follow from Theorem A. First, assume that . If then , and
and applying Theorem A, it follows that if , then with probability at least ,
and if , then with probability at least ,
for constants that depend on .
In a similar fashion, if then , and thus, if , . Therefore, the error rate of ERM is given by . When (the noise-free case) then and with probability larger than , , implying exact reconstruction.
Turning to the lower estimate, assume that the set of admissible targets contains every Y^{t}=\bigl{<}t,x\bigr{>}+W, for and that is a centered gaussian random variable with variance that is independent of . It follows from Theorem A′′ that if , ERM is an optimal procedure in the following sense: it achieves the accuracy
if , and the accuracy
if (note that when then in (4.1) is larger than and the probability estimate is negative).
For a minimax lower bound that holds with constant probability we shall apply Theorem B. To that end, let us bound the covering numbers from below. First note that
and it suffices to study the covering numbers for various choices of .
Fix , and without loss of generality assume that is an integer. For , let be the Euclidean sphere supported on the coordinates , and note that
Moreover, one can prove (via Maurey’s empirical method) that this estimate is sharp (see, e.g., ). Thus it follows that for any ,
If than and by a volumetric estimate, . If, on the other hand, then and since (which is evident from the argument used above), then . Finally, when , .
We conclude that when (and in particular, when ), and if , then . This estimate also exhibits that Sudakov’s inequality for the set is sharp at the scale in the following sense: for every ,
Therefore, by Theorem B, one has that if and if the set of admissible targets contains every Y^{t}=\bigl{<}X,t\bigr{>}+W as above, then the minimax rate in the constant confidence regime is and that ERM is optimal procedure when .
Note that when the estimates on and on do not coincide. And, it turns out that if one extends the set of admissible targets, ERM cannot perform with a better accuracy than in this range. Indeed, consider the one dimensional case and a target defined as follows: the marginal law of given is
for that will be specified later, and that is distributed uniformly in . The corresponding class of one-dimensional linear functionals is .
It is straightforward to verify that for every ,
and if then the minimizer of in is .
Next, let us identify the minimizer of the empirical risk . Given the sample , let . Observe that for every ,
Thus, when (i.e., when ), the best accuracy that ERM can achieve with constant probability is .
Finally, turning to the low noise regime (), one can show that the rate is actually sharp. Recall that by Theorem C it suffices to show that the Gelfand -width of satisfies . By a result due to Garanaev and Gluskin , when one has
and when . Therefore, when either or . In particular, when , the minimax rate is and it is achieved by the ERM.
2 Low-rank matrix inference via the max-norm
In this section, the goal is to estimate the real-valued output by a linear function of a low-rank (or approximately low rank) matrix. Since the rank is not a convex constraint, one may consider “a convex relaxation” given by the factorization-based norm
Let be the unit ball relative to that norm and set {\mathcal{F}}=\{f_{A}=\bigl{<}\cdot,A\bigr{>}:A\in{\mathcal{B}}_{max}\}. Thus,
Assume that is isotopic and -subgaussian relative to the normalized Frobenius norm, and in particular,
To apply Theorem A, one has to estimate the fixed points and for that depends only on and .
Let be the unit ball relative to the Frobenius norm. Since is isotropic, the relative unit ball is
and the corresponding gaussian process has a covariance structure given by
A simple application of Grothendieck’s inequality (see, e.g., ) shows that
where is the Grothendieck constant and ; in particular, .
Let be a matrix with independent, centered gaussian entries with variance . Thus, for every ,
By standard properties of gaussian processes,
In the reverse direction, by Lemma 3.1 in , if
Hence, it follows from Sudakov’s inequality that in that range of ,
as long as both are smaller than and larger than ; that is, when , and .
Applying Theorem A, if then with probability at least , ERM satisfies that
and if , then with probability at least ,
To see that the estimate is sharp in the minimax sense when (and as long as , i.e., ), observe that Theorem A′′ implies that ERM achieves the minimax rate for the confidence parameter . Moreover, by Theorem B and (4.4), any procedure with confidence parameter has accuracy , matching the upper bound.
Concluding remarks
Subgaussian classes are the first family of unbounded classes one is likely to consider, and it turns out that just like bounded classes, the study of subgaussian learning problems may be carried out using a two-sided concentration argument. Unfortunately, this is as far as concentration goes: the substantial technical machinery needed for the proof of Theorem A is not true beyond the subgaussian framework, and the analysis of more ‘heavy-tailed’ problems requires a totally different machinery (see ). Moreover, in more heavy-tailed situations, ERM does not attain the optimal accuracy/confidence tradeoff.
as long as is convex and centrally-symmetric. No procedure can outperform this rate, say with confidence at least provided that:
Let us mention once again that a complete characterization of the minimax rate in this case was recently established in , and the optimal procedure happens to be a minor modification of ERM: it is ERM performed in an appropriate net in .
The parameter may be compared with the fixed points used in . In all those cases, the fixed points are associated with Dudley’s entropy integral for the localized class, rather than with the localized gaussian process; as such, the resulting bounds are always weaker than ours. For example, the results in which deal with the same situation as Theorem A′′ show that if the noise level is large enough and there is no gap in both Sudakov’s AND Dudley’s inequalities at the correct level (given by the fixed point), ERM is a minimax procedure in expectation. Theorem A′′ clearly improves that result.
Finally, although the importance of convexity may have been obscured by the Bernstein condition, a uniform Bernstein condition implies that the class is convex, at least if a nontrivial error rate is to be expected.
Indeed, observe that if is closed but not locally compact in then the minimax rate of does not tend to as the sample size tends to infinity. This is an immediate outcome of Theorem B and the fact that there is some and for which contains an infinite set that is separated in . Thus, one may restrict oneself to classes that are locally compact, and, in which case, one has the following:
Let be a probability measure and let be distributed according to . If is a locally compact subset of , the following are equivalent:
Proof. If is a nonempty, closed and convex subset of a Hilbert space, the metric projection exists and is unique. By its characterization, \bigl{<}f(X)-f^{*}(X),Y-f^{*}(X)\bigr{>}\leq 0 for every , and
In the reverse direction, if is locally compact, the set-value metric projection onto exists, and since it is -Bernstein for any , the metric projection is unique. Indeed, if are minimizers then by the Bernstein condition,
Thus, any has a unique best approximation in , making a locally compact Chebyshev set in a Hilbert space. By a result due to Vlasov , (see also , Chapter 12), is convex.
Appendix A Additional proofs
First note that the canonical gaussian process we are interested in is a restriction of the isonormal process on to a subset (see Section 12 in ). In particular, it inherits the linearity of the isonormal process – a fact we shall use below.
Proof of Lemma 2.3. Fix and . Assume that and observe that since is star-shaped around and , it follows that
Since (A.1) clearly holds if , by taking the supremum over all possible choices of ,
which is equivalent to ; therefore, is non-increasing on .
The two other parts of the claim can be established using a similar argument and their proofs are omitted.
Proof of Lemma 2.4. First, assume that . Let and note that by Lemma 2.3,
For the reverse direction, let and set . Thus, by Lemma 2.3,
Hence, if then . But if , which clearly holds.