Geometric median and robust estimation in Banach spaces
Stanislav Minsker
Introduction
for some absolute constant without any extra assumptions on ? What happens if the sample contains a fixed number of outliers of arbitrary nature? Does the estimator still exist?
A (somewhat surprising) answer is yes, and several ways to construct are known. The earliest reference that we are aware of is the book by Nemirovski and Yudin , where related question was investigated in the context of stochastic optimization. We learned about problem (1) and its solution from the work of Oliveira and Lerasle who used the ideas in spirit of to develop the theory of “robust empirical mean estimators”. Method described in consists of the following steps: divide the given sample into blocks, compute the sample mean within each block and then take the median of these sample means. A relatively simple analysis shows that the resulting estimator indeed satisfies (1). Similar idea was employed earlier in the work of Alon, Matias and Szegedy to construct randomized algorithms for approximating the so-called “frequency moments” of a sequence. Recently, the aforementioned “median of the means” construction appeared in in the context of multi-armed bandit problem under weak assumptions on the reward distribution. A different approach to the question (1) (based on PAC-Bayesian truncation) was given in . A closely related independent recent work applies original ideas of Nemirovski and Yudin to general convex loss minimization.
The main goal of this work is to design a general technique that allows construction of estimators satisfying a suitable version of (1) for Banach space-valued . To achieve this goal, we show that a collection of independent estimators of a Banach space-valued parameter can be transformed into a new estimator which preserves the rate and admits much tighter concentration bounds. The method we propose is based on the properties of a geometric median, which is one of the possible extensions of a univariate median to higher dimensions.
Many popular estimators (e.g., Lasso in the context of high-dimensional linear regression) admit strong theoretical guarantees if the distribution of the noise satisfies restrictive assumptions (such as sub-Gaussian tails). An important question that we attempt to answer is: can one design algorithms which preserve nice properties of existing techniques and at the same time:
admit strong performance guarantees under weak assumptions on the noise;
are not affected by the presence of a fixed number of outliers of arbitrary nature and size;
can be implemented in parallel for faster computation with large data sets.
Our results imply that in many important applications the answer is positive. In Section 4, we illustrate this assertion with several classical examples, including principal component analysis, sparse linear regression and low-rank matrix recovery. In each case, we present non-asymptotic probabilistic bounds describing performance of proposed methods.
For an overview of classical and modern results in robust statistics, see , and references therein. Existing literature contains several approaches to estimation in the presence of heavy-tailed noise based on the aforementioned estimators satisfying (1). However, most of the previous work concentrated on one-dimensional versions of (1) and used it as a tool to solve intrinsically high-dimensional problems. For example, in authors develop robust estimator selection procedures based on the medians of empirical risks with respect to disjoint subsets of the sample. While this approach admits strong theoretical guarantees, it requires several technical assumptions that are not always easy to check it practice. Another related work discusses robust estimation in the context of ridge regression. Proposed method is based on a “min–max” estimator which has good theoretical properties but can only be evaluated approximately based on heuristic methods. It is also not immediately clear if this technique can be extended to robust estimation in other frameworks. An exception is the approach described in and further explored in , where authors use a version of the multidimensional median for estimator selection. However, this method has several weaknesses in statistical applications when compared to our technique; see Section 3 for more details and discussion. The main results of our work require minimal assumptions, apply to a wide range of models, and allow to use many existing algorithms as a subroutine to produce robust estimators which can be evaluated exactly via a simple iterative scheme.
Geometric median
The cornerstone of our subsequent presentation is the following lemma, which states that if a given point is “far” from the geometric median , then it is also “far” from a constant fraction of the points . We will denote .
and . Then there exists a subset of cardinality such that for all , .
For general Banach spaces, the claim holds with a constant .
(a) Assume that the implication is not true. Without loss of generality, it means that , .
For and , we clearly have (see Figure 1)
while for . This yields
whenever , which leads to a contradiction.
In general Banach spaces, it might be convenient to consider
where is the convex hull of . The claim of Lemma 2.1 remains valid for whenever .
“Boosting the confidence” by taking the geometric median of independent estimators
A useful property of the geometric median is that it transforms a collection of independent estimators that are “weakly” concentrated around the true parameter of interest into a single estimator which admits significantly tighter deviation bounds. For , define
where is a constant defined in Lemma 2.1 above.
Assume that event occurs. Lemma 2.1 implies that there exists a subset of cardinality such that for all , hence
If has Binomial distribution , then
(see Lemma 23 in for a rigorous proof of this fact). Chernoff bound (e.g., Proposition A.6.1 in ) implies that
If (5) is replaced by a weaker condition assuming that
is satisfied only for , , where for , then the previous argument implies
In particular, this version is useful in addressing the situation when the sample contains a subset of cardinality at most consisting of “outliers” of arbitrary nature.
It is also clear that results of Theorem 3.1 can be used to positively answer question (3) posed in the Introduction. Indeed, if several autonomous computational resources (e.g., processors) are available, one can evaluate estimators , in parallel and combine them via the geometric median as a final step. In many situations, the improvement in computational cost will be significant.
Note that it is often easy to obtain an estimator satisfying (5) with the correct rate under minimal assumptions on the underlying distribution. In particular, if is the mean and is the sample mean, then (5) can be deduced from Chebyshev’s inequality, see Section 4.1 below for more details.
Next, we describe the method proposed in which is based on a different notion of the median. Let be a collection of independent estimators of and assume that is chosen to satisfy
and if none of ’s satisfy the condition in braces. It is not hard to show that
It is important to mention the fact that (10) and the inequality (7) of Theorem 3.1 have different constants in front of : it is equal to in (7) and to in (10). Note that in the Hilbert space case, as , while for general Banach spaces as . In particular, for all sufficiently small (e.g., for in Hilbert space framework). This difference becomes substantial when is of the form
where the first term in the sum is a constant and the second term decreases with the growth of the sample size. This is a typical situation when the model is misspecified, see Section 4.4 below for a concrete example related to matrix regression. Our method allows to keep the constant in front of the approximation error term arbitrary close to (and often leads to noticeably better constants in general).
Examples
In this section, we discuss applications of Theorem 3.1 to several classical problems, namely, estimation of the mean, principal component analysis, sparse linear regression and low-rank matrix recovery.
Our priority was simplicity and clarity of exposition of the main ideas which could affect optimality of some constants and generality of obtained results.
Set and (these numerical values allow to optimize the constants in Corollary 4.1 below). Let be the confidence parameter, and set
(we will assume that is such that ). Divide the sample into disjoint groups of size each, and define
We will apply Theorem 3.1 to the independent estimators .
To this end, we need to find satisfying (5). Since for all
which is further bounded by whenever . The claim now follows from Theorem 3.1 and the bounds and . ∎
It is easy to see that the proof of Corollary 4.1 actually yields a better bound
with .
which yields a noticeably larger constant ( versus ).
It is easy to see that for the univariate median, inequality (7) holds with and , hence the union bound over implies that
(here, was set to be ). This bound should be compared to (13) – the latter becomes better only when is sufficiently large (e.g., for and for ).We want to thank the anonymous reviewer for pointing this out. Note that the constant in (14) can be further improved in a situation when tight upper bounds on the true variances or kurtoses of coordinates of are known by using a univariate estimator of to construct .
Our estimation technique naturally extends to the problem of constructing the confidence sets for the mean. Indeed, when faced with the task of obtaining the non-asymptotic confidence interval, one usually fixes the desired coverage probability in advance, which is exactly how we build our estimator. To obtain a parameter-free confidence ball from (12), one has to estimate . To this end, we will apply Theorem 3.1 to a collection of independent statistics , where
and are the sample means defined in (4.1). Let (if is even, the median is not unique, so we pick an arbitrary representative).
Chebyshev’s inequality gives (assuming that )
hence . Theorem 3.1 implies that
Since whenever (15) is satisfied, the result follows. ∎
Combining (16) with Corollary 4.1, we immediately get the following statement.
and let be the estimator defined by (4.1). If (15) holds, then
2 Robust Principal Component Analysis
It is well known that classical Principal Component Analysis (PCA) is very sensitive to the presence of the outliers in a sample. The literature on robust PCA suggests several computationally efficient and theoretically sound methods to recover the linear structure in the data. For instance, if part of the observations is contained in a low-dimensional subspace while the rest are corrupted by noise, the low-dimensional subspace can often be recovered exactly, see and references therein.
However, for the case when no additional geometric structure in the data can be assumed, we suggest a simple and easy-to-implement alternative which uses the geometric median to obtain a robust estimator of the covariance matrix. In this section, we study the simplest case when the geometric median is combined with the sample covariance estimator. However, it is possible to use various alternatives in place of the sample covariance, such as the shrinkage estimator , banding/tapering estimator , hard thresholding estimator or the nuclear norm-penalized estimator , to name a few.
Assume first that the data is centered (so that ). As before, set , , divide the sample into disjoint groups of size each, and let
Note that is positive semidefinite as a convex combination of positive semidefinite matrices.
Let be the orthogonal projector on a subspace corresponding to the largest positive eigenvalues of . Let be the orthogonal projector of the same rank as corresponding to the largest eigenvalues of . In this case, the following bound holds.
Let and assume that
Similar bounds can be obtained in a more general situation when is not necessarily centered. To this end, let
Note that are independent. Then, using the fact that for any
and assume that . Then
3 High-dimensional sparse linear regression
We are interested in the case when and is sparse, meaning that
In this situation, a (version of) the famous Lasso estimator of is obtained as a solution of the following optimization problem:
The goal of this section is to extend the applicability of some well-known results for this estimator to the case of a heavy-tailed noise distribution.
Existing literature on high-dimensional linear regression suggests several ways to handle corrupted measurements, for instance, by using a different loss function (e.g., the so-called Huber’s loss ), or by implementing a more flexible penalty term . In particular, in authors study the model
However, as in the case of the usual Lasso, confidence of estimation depends on the distribution of . In particular, Gaussian-type concentration holds only if the entries of have sub-Gaussian tails.
The main result of this subsection (stated in Theorem 4.2) provides strong performance guarantees for the robust version of the usual Lasso estimator (22) and requires only standard conditions on the degree of sparsity and restricted eigenvalues of the design. Similar method can be used to improve performance guarantees for the model (23) in the case of heavy-tailed noise .
Let and . We will say that the restricted eigenvalue condition holds if
Let . The following result shows that the amount of regularization sufficient for recovery of is closely related to the size of .
On the event , the following inequality holds:
In particular, when and ,
This result is similar to the statement of Theorem 7.2 in , and its proof can be obtained along the same lines. See for more details. ∎
Our next goal is to construct an estimator of which admits high confidence error bounds without restrictive assumptions on the noise variable, such as sub-Gaussian tails. Let be fixed, and set , (as before, we will assume that ). For , let and
be the design matrix corresponding to the th group of design vectors . Moreover, let be the corresponding restricted eigenvalues.
Assume that and . Then for any
We will first obtain a “weak concentration” bound from Theorem 4.1 and then apply Theorem 3.1 with to get the result.
To this end, we need to estimate , .
Assume that . Then for any , ,
whenever . In particular, for , the bound of Theorem 4.1 holds for with probability ; note that . It remains to apply Theorem 3.1 to complete the proof. ∎
4 Matrix regression with isotropic sub-Gaussian design
The problem of low-rank matrix estimation has attracted a lot of attention during the last several years, for example, see and references therein. Recovery guarantees were later extended to allow the presence of noise. Results in this direction can be found in , to name a few.
In particular, these assumptions hold in the following important cases:
is symmetric and such that , , , , where are i.i.d. Rademacher random variables (i.e., random signs).
In a special case when all involved matrices are diagonal, the problem becomes a version of sparse linear regression with random design. In this case, isotropic design includes a situation when is a random diagonal matrix , where are i.i.d. standard normal or Rademacher random variables.
In what follows, denote the constants that may depend on parameters of the underlying distribution (such as ).
It is easy to see that for any sub-Gaussian random variable . It follows from Proposition 9.1 in that there exists such that for any sub-Gaussian isotropic matrix ,
We will also need the following useful inequality: for any ,
The proofs of the facts mentioned above can be found in .
Let be i.i.d. observations with the same distribution as . We are mainly interested in the case when . In this situation, it is impossible to estimate consistently in general, however, if is low-rank (or approximately low-rank), then the solution of the following optimization problem provides a good approximation to :
The noise variable is such that .
If Assumption 4.1 is satisfied, then, whenever
where is an absolute constant.
Indeed, recall that , and apply Theorem 4.4 to , noting that by (26), (27)
where , depend only on , hence giving the result.
As we mentioned above, our goal is to construct the estimator of which admits bounds in flavor of Theorem 4.3 that hold with high probability under a much weaker assumption on the tail of the noise variable .
To achieve this goal, we follow the same pattern as before. Let be fixed, let , and assume that . Divide the data into disjoint groups of size each, and define
where the geometric median is evaluated with respect to the Frobenius norm.
We will start by deriving a “weak concentration” bound from Theorem 4.3. To this end, we need to estimate
The following result is a direct consequence of the so-called multiplier inequality (Lemma 2.9.1 in ).
Let be i.i.d. Rademacher random variables independent of . Then
Next, it follows from Chebyshev’s inequality that for any , with probability
the inequality of Theorem 4.3 (with the confidence parameter equal to ) applied to the estimator gives that with probability
The claim (33) now follows from Theorem 3.1. ∎
Numerical evaluation of the geometric median and simulation results
As was mentioned in the introduction, the function is convex, moreover, its minimum is unique unless are on the same line.
where . Kuhn proved that Weiszfeld’s algorithm converges to the geometric median for all but countably many initial points (additionally, his result states that converges to the geometric median if none of belong to ). It is straightforward to check that (36) is actually a gradient descent scheme: indeed, it is equivalent to
where and is the gradient of (we assume that ).
Ostresh proposed a method which avoids the possibility of hitting one of the vertices by considering the following descent scheme: starting with some in the affine hull of , let
For other approaches to fast numerical evaluation of the geometric median, see and references therein.
2 Simulation results
In this case, the usual sample covariance matrix does not provide any useful information about the principal components. However, in most cases our method gave reasonable approximation to the truth. We used the estimator described in Section 4.2 with the number of groups containing observations each. The error was measured by the spectral norm , where is a projector on the eigenvectors corresponding to largest eigenvalues of the estimator. Figures 2, 3 show the histograms of the errors evaluated over runs of the simulation. Figure 4 shows performance of a “thresholded geometric median” estimator which is defined in Section 6 below.
2.2 High-dimensional sparse linear regression
The following model was used for simulation:
where and takes values with probability each. All parameters , , , , were sampled independently. Error of the estimator was measured by the ratio . Size of the regularization parameter was chosen based on -fold cross validation. On each stage of the simulation, we evaluated the usual Lasso estimator (22) and the “median Lasso” estimator (24) based on partitioning the observations into groups of size each. Figures 5 and 6 show the histograms of the errors over runs of the simulation. Note that the maximal error of the “median Lasso” is while the error of the usual Lasso exceeded in out of cases.
Final remarks
Let , be the coefficients such that is the geometric median of a collection of estimators . Our numerical experiments reveal that performance of can be significantly improved by setting the coefficients below a certain threshold level to , that is,
An interesting problem that we plan to address in subsequent work is the possibility of adaptive choice of the threshold parameter.
Examples presented above cover only a small area on the map of possible applications. For instance, it would be interesting to obtain an estimator in the low-rank matrix completion framework that admits strong performance guarantees for the heavy-tailed noise model. Results obtained in Section 4.4 for the matrix regression problem do not seem to yield a straightforward solution in this case. Another promising direction is related to design of robust techniques for Bayesian inference and evaluation of the geometric median in the space of probability measures. We plan to address these questions in the future work.
Appendix: Proof of Lemma 2.1, part (b)
Once again, assume that the claim does not hold and , .
be the directional derivative of at the point in direction . We need the following useful fact from convex analysis:
There exists such that , where .
Using Lemma .1, it is easy to see that there exist , such that
Moreover, for any , by the definition of . Note that for any
By the definition of and triangle inequality,
and, since ,
whenever .
Acknowledgements
S. Minsker was supported by grants NSF DMS-0847388, NSF CCF-0808847, and R01-ES-017436 from the National Institute of Environmental Health Sciences (NIEHS) of the National Institutes of Health (NIH).
I want to thank Anirban Bhattacharya, David Dunson, the anonymous Referees and the Area Editor for their valuable comments and suggestions, and Philippe Rigollet for pointing out several missing references.