Learning from MOM's principles: Le Cam's approach
Lecué Guillaume, Lerasle Matthieu
Introduction
Consider the problem of estimating minimizers of the integrated square-loss over a convex class of functions : based on a data set . The labels and ’s are real-valued while the inputs and ’s take values in an abstract measurable space .
These estimators are optimal in i.i.d. subgaussian setups but suffer several drawbacks when data are heavy-tailed or corrupted by “outliers”, see Catoni (2012); Huber and Ronchetti (2009). These issues are critical in many modern applications such as high-frequency trading, where heavy-tailed data are quite common or in various areas of biology such as micro-array analysis or neuroscience where data are sometimes still nasty after being preprocessed. To overcome the problem, various methods have been proposed. The most common strategy is to replace the square-loss function to make it less sensitive to outliers. For example, Huber (1964) proposed a loss that interpolates between square and absolute loss to produce an estimator between the unbiased (but non robust) empirical mean and the (more robust but biased) empirical median. Huber’s estimators have been intensively studied asymptotically by Huber (1964); Huber and Ronchetti (2009), non-asymptotic results have also been obtained more recently by Chichignoud and Lederer (2014); Mendelson (2015b); Fan et al. (2017) for example. An alternative approach has been proposed by Catoni (2012) and used in learning frameworks such as least-squares regression by Audibert and Catoni (2011) and for more general loss functions by Brownlees et al. (2015).
Another line of research to build robust estimators and robust selection procedures was initiated by Le Cam (1973, 1986) and further developed by Birgé (2006), Baraud (2011) and Baraud et al. (2017). It is based on comparisons or tests between elements of . More precisely, the approach builds on tests statistics comparing and . These tests define the sets of all ’s that have been preferred to and the final estimator is a minimizer of the diameter of . The measure of diameter is directly related to statistical performances one seeks for the estimator. These methods mostly focus on Hellinger loss and are generally considered difficult to compute, see however Baraud et al. (2014); Sart (2014).
In a related but different approach, Lugosi and Mendelson (2017) have recently introduced “median-of-means tournaments”. Median-of-means estimators of Alon et al. (1999); Jerrum et al. (1986); Nemirovsky and Yudin (1983) compare elements of . A “champion” is an element such that is smaller than a computable upper bound on the radius of . They prove that the risk of any champion is controlled by this upper bound. An important message of this paper is that Le Cam’s estimators are quite common in statistics, in particular in robust statistics. For example, Section 3 shows that any penalized empirical loss function can be obtained by Le Cam’s approach and that Le Cam’s estimators based on median-of-means tests are champions of median-of-means tournaments.
This paper studies estimators derived from Le Cam’s procedure based on regularized median-of-means (MOM) tests (see Section 4.1). Our estimators are therefore particular instances of champions of MOM’s tournaments and another motivation is to push further the analysis of this particular champion. The main advantage of MOM’s tests over Le Cam’s original ones is that they allow for more classical loss functions than Hellinger loss. This idea is illustrated on the square-loss. Compared to Huber or Catoni’s losses, this approach allows to control easily the risk of our estimators by using classical tools from empirical process theory, it also allows to tackle the problem of “aggressive” outliers.
The closest work is certainly that of Lugosi and Mendelson (2017), but we believe that our paper contains substantial improvements. We stress the intimate relationship between their estimator and Le Cam general construction and use this parallel to propose a much simpler estimator. Our risk bounds are always better and we extend their results to possibly corrupted data-sets.
To investigate robustness properties of median-of-means estimators, we partition the dataset into two parts. One is made of outliers data. They are indexed by of cardinality . On those data, absolutely nothing is assumed : they may not be independent, have distributions totally different from , with no moment at all, etc.. These are typically data polluting datasets like in the case of declarative data on internet or when something went wrong during the storage, compression or transfer which resulted in complete non sense data. They may also be observations met in biology as in the classical eQTL (Expression Quantitative Trait Loci and The Phenogen Database) from Saba et al. (2008). Many other examples of datasets containing outliers could be provided, this includes frauds detection and terrorist activity as examples. Of course, outliers are not flagged in advance and the statistician is given no a priori information on which data is an outlier or not. The other part of the dataset is made of data on which the MOM estimator rely on to estimate the oracle . There should be enough information in those data so that the estimation of is possible, even in the presence of outliers provided they remain in a “decent proportion”. We therefore call the non-outliers, the informative data, those that bring information on . We denote by the set indexing these data. We therefore end up with a partition of as which, again, is not known from the statistician.
The radii of the sets are computed for regularization and norms. The regularization norm is chosen in advance by the statistician to promote sparsity or smoothness. It can be used freely in our procedure, but it doesn’t ensure a small risk for the estimator. The -norm is unknown in general since it depends on the distribution of . Furthermore, the classical -empirical metric fails to estimate the metric without subgaussian properties of the design vector . Fortunately, it can be replaced by a median-of-means metric. To handle simultaneously both regularization and norms, we will also slightly extend Le Cam’s principle. Our first important result shows that the resulting estimator is well localized w.r.t. both regularization and norms.
Median-of-means estimators rely on a data splitting into blocks and this parameter drives the resulting statistical performances (cf. Devroye et al. (2016)). To achieve optimal rates, should be ultimately chosen using parameters that depend on the oracle like its sparsity which is not in general available to the statistician. To bypass this problem, the strategy of Lepski (1991) is used as in Devroye et al. (2016) to select adaptively and get a fully data-driven procedure.
[Theorem 1.4 in Lecué and Mendelson (2016a)] Assume is -sparse, , is isotropic and
and (no outliers in the dataset),
\zeta=Y-\bigl{<}X,t^{*}\bigr{>}\in L_{q_{0}} for some
This paper shows that Theorem 1 holds for a MOM version of the LASSO estimator under much weaker assumptions, with a better probability estimate than (1). More precisely, the following theorem is proved.
Assume that is -sparse, , is isotropic and
and (the number of outliers may be proportional to the sparsity times ),
\zeta=Y-\bigl{<}X,t^{*}\bigr{>}\in L_{q_{0}} for some
There exists an estimator , called MOM-LASSO, satisfying for every ,
Theoretical properties of MOM LASSO outperform those of LASSO in several ways.
Estimation rates achieved by MOM-LASSO are the actual minimax rates , see Bellec et al. (2016), while classical LASSO estimators achieve the rate . This improvement is possible thanks to the adaptation step in MOM-LASSO.
the probability deviation in (1) is polynomial – in (1) – it is exponentially small for MOM LASSO. Exponential rates for LASSO hold only if is subgaussian ( for all ).
MOM LASSO is insensitive to data corruption by up to times outliers while only one outlier can be responsible of a dramatic breakdown of the performances of LASSO.
From a mathematical point of view, our results are based on a slight extension of the Small Ball Method (SBM) of Koltchinskii and Mendelson (2015); Mendelson (2014a) to handle non-i.d. data. SBM is also extended to bound both quadratic and multiplier parts of the quadratic loss. Otherwise, all arguments are standard, which makes the approach very attractive and easily reproducible in other frameworks of statistical learning.
The paper is organized as follows. Section 2 briefly presents the general setting and our main illustrative example. Section 3 presents Le Cam’s construction of estimators based on tests. We also show why many learning procedures may be obtained by this approach. The construction of estimators and the main assumptions are gathered in Section 4. Our main theorems are stated in Section 5 and proved in Section 6.
Let also . We also denote by the indicator function of the set which equals to when and otherwise.
Setting
Let denote a norm defined onto a linear subspace of containing .
Let f^{*}=\bigl{<}\cdot,t^{*}\bigr{>}\in F, where
Learning from tests
This section details the ideas underlying the construction of a MOM estimator using an extension of Le Cam’s approach.
By definition of the oracle , one has
As depends on , we estimate it by test statistics that is, real random variables such that
These statistics are used to compare to , simply by saying that -beats iff . In this paper, the statistics are median-of-means estimators of (cf. (12) in Section 4.1).
Le Cam’s construction
Let denote a collection of test statistics and let denote a pseudo-distance on measuring (or related to) the risk we want to control. Let for all ,
be the set of all functions that beat . If is far from , then is expected to have a large radius w.r.t. . We therefore introduce this radius as a criteria to minimize : for all , let .
By (3), or (both happen if ), hence . In particular, for all ,
Risk bounds for follow from (6) and upper bounds on the radii of .
More generally, one can compare only the elements of a subset , typically a maximal -net by introducing for all , the set
and then by minimizing the diameter of over . This usually improves the rates of convergence for constant deviation results when there is a gap in Sudakov’s inequality of the localized sets of (cf. Section 5 in Lecué and Mendelson (2013) for more details). These results are not presented because we are interested in exponentially large deviation results for which our results are optimal.
Dealing with regularization : the link function
Statistical performances of estimators and the radius of can be measured by two norms: the regularization norm and . As (5) allows only for one distance , we propose the following extension of Le Cam approach to handle two metrics.
To introduce this extension, assume first that can be computed for all (this is the case if the distribution of the design is known). The next paragraph explains how to deal with the more common framework where this distance is unknown. Remark that
The main point to extend Le Cam’s approach to simultaneously control two norms is to design a link function . In a nutshell, the values is the -minimax rate of convergence in a ball of radius for the regularization norm (cf. (13) in Section 4.3 for a formal definition). Then one can define
Theorem 3 shows that while a minimizer of has only a nice risk for , a minimizer of has both and properly controlled.
Dealing with unknown norms : the isometry property
In general, -distances cannot be directly computed and have to be estimated. To deal with this issue, one considers usually the empirical distance and prove that empirical and actual distances are equivalent outside a -ball centered in (cf. for instance, remark after Lemma 2.6 in Lecué and Mendelson (2013)). Unfortunately this approach only works under strong concentration property that we want to relax in this paper.
The unknown -metric is instead estimated by a median-of-means approach, that is, we use MOM estimators of all (cf. Section 4.4). The final estimator is therefore defined as a minimizer of
2 Examples
Le Cam’s approach has been used by Birgé to define -estimators (cf. Baraud and Birgé (2009); Birgé (2006, 2013)) and by Baraud, Birgé and Sart to define -estimators (cf. Baraud and Birgé (2016); Baraud et al. (2017)). Baraud (2011); Baraud et al. (2014) also built efficient estimator selection procedures with this approach. It also extends many common procedures in statistical learning theory, as shown by the following examples.
is obtained by Le Cam’s construction with the tests
These examples encompass classical empirical risk minimizers of Vapnik (1998) but also their robust versions from Huber (1964); Audibert and Catoni (2011).
Example 2 : median-of-means estimators
Another, perhaps less obvious example is the median-of-means estimator Alon et al. (1999); Jerrum et al. (1986); Nemirovsky and Yudin (1983) of the expectation of a real valued random variable . Let denote a sample and let denote a partition of into bins of equal size . The estimator is the (empirical) median of the vector of empirical means . Recall that
Basic properties of the median (recalled in Eq (8) and (9) of Section 4.1) yield
Example 3 : “Champions” of a Tournament
Construction of the regularized MOM estimators
This section presents median-of-means (MOM) tests used in this work. Designing a family of tests is one of the most important building blocks in Le Cam’s approach together with the right choice of the metric measuring the diameters for .
With some abuse of notations, we shall write these properties respectively
To compare/test functions and in , median-of-means tests between and are now defined by
From (9), satisfies (3) and is a tests statistic in the sense of Section 3.
2 Main assumptions
Recall that and that is a set of outliers on which we make no assumption so these may be aggressive in any sense one can imagine. The remaining informative data need to bring enough information onto . We therefore need some assumption on the sub-dataset and, in particular, some connexion between the distributions for and . These assumptions are pretty weaksince we only assume essentially that the and geometries are comparable in the following sense.
There exists such that, for all and ,
There exists such that, for all and ,
Let us give some examples where Assumption 2 holds. If the noise random variable (resp. for ) has a variance conditionally to (resp. for ) that is uniformly bounded then Assumption 2 holds. This is, for example, the case, when (resp. for ) is independent of (resp. for ) and has finite -moment with . It also holds without independence under higher moment conditions. For example, assume and, for every , then by Cauchy-Schwarz inequality, and so Assumption 2 holds for .
There exists such that for all and all
By Cauchy-Schwarz inequality, for all and . Therefore, Assumptions 1 and 3 together imply that all norms are equivalent over . Note also that Assumption 3 is related to the small ball property (cf. Koltchinskii and Mendelson (2015); Mendelson (2014a)) as shown by Proposition 1 bellow. The small ball property has been recently used in Learning theory and signal processing. We refer to Koltchinskii and Mendelson (2015); Lecué and Mendelson (2014); Mendelson (2015b, 2014b, a); Rudelson and Vershynin (2014) for examples of distributions satisfying this assumption.
Let be a real-valued random variable.
As one can assume that , .
3 Complexity parameters and the link function
This section defines the link function making the connections between norms that will be required in the extension of Le Cam’s approach to a simultaneous control of two norms (one of the two being unknown). For any and any , let
Let be independent Rademacher random variables, independent from and let . For any and let ,
Note that if the function is itself continuous and non-decreasing then it can be taken equal to . In the next paragraph, we provide an explicit computation of the functions , and in the “LASSO case”.
Under Assumption 4, if , (Mendelson, 2016, Theorem 1.6) shows that, for every ,
Therefore, a link function is explicitly given by
4 The estimators
Let denote the family of tests defined in (12). For every function , let denote the set of all functions that beats . As explained in Section 3, these sets will be measured by two metrics. First, let
Lemma 4 below proves that, with large probability, and are isomorphic distances. The second criterion is then given by
where is a link function as defined in Definition 1. That is a continuous and non-decreasing function such that for all , where the choice of and is given in Theorem 3 below. The associated estimator is then given by
5 The sparsity equation
By (6), estimation rates for will be derived from upper bounds on . To get these, our strategy is to show that for all such that or is large.
Recall that the quadratic / multiplier decomposition of the excess quadratic risk:
Let and . When is large and is small, thanks to the regularization term in (16) because the quadratic term is likely to be small. We will therefore derive a lower bound on the regularization term when the subdifferential of is “large” in the following sense.
First, we recall that the subdifferential of in is the set
where is the dual normed space of (and is the linear space containing onto which is defined). For all , let denote the set
where is the link function from Definition 1. Let denote the union of all subdifferentials of at functions “close” to
Intuitively, every norm is associated with a notion of “sparsity” if one agrees to say that a non-zero function is sparse w.r.t. the norm when the subdifferential of this norm at is a “large subset” of the dual sphere (i.e. the sphere of ). Sparse functions are useful in our context because a large lower bound on (and so for when is small enough) can be derived when the vector is in the right direction. This intuition are formalized in the sparsity equation. More precisely, let
is a uniform lower bound on if . Thus, , if or if the following sparsity equation of Lecué and Mendelson (2016a) holds.
A radius satisfies the sparsity equation if .
If satisfies the sparsity equation, so do all . Therefore, one can define
The equation has been solved in this example in (Lecué and Mendelson, 2016a, Lemma 4.2), recall this result.
If and if there exists a -sparse vector in , Lemma 1 and the choice of in (15) imply that for ,
then satisfies the sparsity equation and is the rate of convergence of the LASSO (cf. Lecué and Mendelson (2016a)).
Main results
Theorem 3 gathers estimation error bounds satisfied by the estimators for defined in Section 4.4.
Grant Assumptions 1, 2 and 3 and let , anr denote the functions introduced in Definition 1 for
Let be defined in (17) and let denote the smallest integer such that
For all , let be a solution of . Assume that for every , and ,
when the regularization parameter satisfy
To the best of our knowledge, Theorem 3 provides the first statistical performance of an estimator operating in such a “nasty” environment: the dataset may be corrupted by complete outliers, the informative data may be heavy-tailed and their distribution for is only asked to have a and geometry over equivalent to that of . The most surprising thing is that the rate we obtain for in Theorem 3, i.e. when the number of outliers is less than is the minimax rate we would have gotten in a very good i.i.d. subgaussian framework with independent noise. This means that the quality of a dataset does not have to be as good as it is classically assumed in the literature to make estimation possible: all we need is that a large fraction of the data should be independent (even though we believe that some “weak dependence” could also be introduced) and distributed according to distributions inducing and geometries equivalent to the one.
In Theorem 3, can be as small as the infimum between the number of outliers and times the minimax rate of convergence. Henceforth, if the optimal rate is known, as in Lugosi and Mendelson (2017), Theorem 3 shows that Le Cam’s champion of the median of means tournament with reaches the same performances as any champion in this paper. Theorem 3 is thus an extension of Lugosi and Mendelson (2017) to a non-i.d. corrupted setting for Le Cam’s champion. Moreover, our control improves theirs if the upper bound on the radius of used in Lugosi and Mendelson (2017) is pessimistic (cf. Example 3.2 in Section 3.2).
Assumption 1 is automatically satisfied in the i.i.d. case and so is Assumption (18). Theorem 3 goes beyond this i.i.d. setup, relaxing the i.d. assumptions into proximity assumptions between and geometries, for informative data.
for the function defined in (15). Therefore,
The regularization parameter depends on the “level of noise” , the -norm of . This parameter is unknown in practice. Nevertheless, it can be estimated and replaced by this estimator in the regularization parameter as in (Giraud, 2015, Sections 5.4 and 5.6.2).
2 Adaptive choice of K𝐾K by Lepski’s method
The main drawback of Theorem 3 is that optimal rates are only achieved when . Since is unknown, it cannot be used in general. This issue is tackled in this section by Lepski’s method.
Let and be defined as in Theorem 3. For any integer , let and be defined as in Theorem 3 and for denote by for this choice of . These estimators are the building blocks of the following confidence sets. For all , let
Finally, define adaptive (to ) estimators via Lepski’s method: for , .
Grant assumptions and notations of Theorem 3. There exist absolute constants such that the estimators for satisfy for every , with probability at least ,
In particular, for , if the following regularity assumption holds: there exists an absolute constant such that for all , then with probability at least
Theorem 4 shows that achieves the same rate of convergence with the same exponentially high confidence as a minimax estimator does in the Gaussian regression model (with independent noise). These rates are achieved here under very weak stochastic assumptions allowing the presence of outliers, without assuming that the regression function lies in or that the data are i.i.d.. Compared to Lugosi and Mendelson (2017), using a Lepski method, we don’t have to choose the integer in advance, we let the data decide the best choice and automatically get an estimator with the correct minimax rate of convergence. Moreover, the regularization parameter is chosen adaptively, which yields to exact minimax rates and, since this minimax rate is not required to build the estimators, these are naturally adaptive.
The following result follows from Theorem 4 together with the computation of , , and from the previous sections. This is a slight extension of Theorem 2 to the case where the oracle is not exactly sparse but close to a sparse vector.
and ,
\zeta=Y-\bigl{<}X,t^{*}\bigr{>}\in L_{q_{0}} for some
The MOM-LASSO estimator such that \hat{f}_{LE}=\bigl{<}\hat{t}_{LE},\cdot\bigr{>} satisfies, with probability at least , for every ,
In particular, Theorem 5 shows that, for our estimator contrary to the one in Lugosi and Mendelson (2017), the sparsity parameter does not have to be known in advance in the LASSO case.
Proofs
We consider the set of indices of blocks containing only informative data:
Grant Assumptions 1 and 3. Fix , and let be such that . Let .
For all , let . For , by Assumption 3 and by Assumption 1. By Markov’s inequality, for all ,
Since then and so we have
By the Giné-Zynn symmetrization argument (Boucheron et al., 2013, Lemma 11.4),
where are independent Rademacher variables independent of the data. Moreover, is 1-Lipschitz and . By the contraction principle (cf. (Ledoux and Talagrand, 1991, Theorem 4.12) or (Boucheron et al., 2013, Theorem 11.6)),
Applying again the symmetrization and contraction principles, we get,
It follows from the convexity of that for all , and it also belongs to the sphere of radius . Therefore, by definition of and for ,
In conclusion, on , all is such that
In other words, on , for all , there exist at least blocks such that . For any of these blocks , , hence, on , and .
2 Upper Bound on the multiplier process
For all and , define and
Let and . It follows from Markov’s inequality and Assumption 2 that
Denote and remark that as defined in Definition 1. Let for simplicity. We have
where in the last but one inequality we used that is convex and the same argument as in the proof of Lemma 2. Moreover, since the random variables are centered and independent, the symmetrization argument applies and, by definition of ,
Now, let for all and note that is -Lipschitz, and satisfies for all . Therefore, all satisfies
Furthermore, it follows from the symmetrization argument that
and, from the contraction principle and (6.2), that
In conclusion, on , for all ,
Besides the controls of the quadratic and multiplier MOM processes presented in Lemmas 2 and 3 respectively, the estimation error bounds for the MOM estimators rely on the following isometry property of the MOM processus .
and if then .
In particular, for , on the event , for all , if , then
It follows from Lemma 2 that on the event for all , if then . This yields the “lower bound” result in (25).
For the upper bound of the isomorphic result, we essentially repeat the proof of Lemma 3. Let us just highlight the main differences. We will use the same notation as in the proof of Lemma 3 except that for all , we define
It follows from Chebyshev’s inequality and Assumption 1 that
Moreover, by convexity of , we have, for ,
and then using a symmetrization argument, we obtain that
In particular, on the event , for all there are more than blocks for which, . Now, the result follows from Assumption 1 since .
4 Conclusion to the proof of Theorem 3
The proof relies on the following proposition.
Grant conditions of Theorem 3. Let , for some and the regularization parameter be such that
Using (8), (9) and (11) together with the quadratic / multiplier decomposition of the excess quadratic loss yields that for all ,
Note that when one chooses
For this choice of constants, Lemma 2 applies and for we get that there exists an event with probability larger than and on that event, for all , if then
Moreover, for the choice of parameters as in (27), we also have , hence Lemma 3 applies and for we get that there exists an event with probability larger than and on that event, for all there are more than blocks with such that
Combining the last result with Assumpion (18), it follows that on the event , for all ,
Let us now prove that on the event , one has for all ,
Assume that holds and let . First assume that . Then, it follows from (6.4), (28) and (29), the choice of in (27) and the definition of that
Now, if then it follows from (6.4), (29) and the definition of that
Let be such that and . It follows from the triangular inequality that . Combining this together with (31), it follows that
when .
For all such that
For every and every ,
Assume that, for all ,
Then (32) holds for all such that .
Let be such that . Define and remark that and that, by convexity of , . It follows from (32) that for , one has
Let be such that . By Lemma 5,
Therefore, it will follow from Lemma 6 that
if we can prove that for all such that one has
Let us now prove that (33) holds. Let be such that . First assume that so that . By definition and, since , satisfies the sparsity equation and thus, . Therefore, thanks to (30), when , one has
Finally assume that . Since , it follows from (31) that
when .
End of the proof of Theorem 3
On the event of Proposition 2, is included in the ball , therefore, by definition of (cf. (6)),
Again, by Proposition 2, on the same event , , hence, on , where is an event defined in Lemma 4, for all ,
where according to (27). Therefore, , which implies that (cf. (6)) and that and therefore, by Lemma 4, on , either and so or and so
5 Conclusion to the proof of Theorem 4
First, it follows from Theorem 3 that for all , with probability at least , for both , , so , which implies that both and belong to , therefore, .
Let be the event defined as the following intersection:
So, in particular, . By definition of , this implies that on . Therefore, on ,
Now on , one has for all , if then
Therefore on , one has either or and in the latter case,