Sub-Gaussian estimators of the mean of a random matrix with heavy-tailed entries
Stanislav Minsker
Introduction
Because the only assumption on is the existence of a second moment, it is natural to call such an estimator “robust” For the classical treatment of robust estimators based on the notion of a breakdown point, we refer the reader to .: it admits strong deviation bounds even for the heavy-tailed distributions that can be used to model outliers in the data. Ideas behind these results have also been extended to empirical risk minimization methods which cover a wide range of statistical applications. Let us emphasize that the aforementioned estimators do not require any assumptions on the “shape” of the distribution, such as unimodality or elliptical symmetry.
Generalizations of univariate results to the case of random vectors and random matrices are not straightforward since element-wise deviation inequalities do not always translate into desired bounds. In some cases, element-wise bounds yield inequalities for the “wrong” norm: for example, estimating each entry of the covariance matrix results in a deviation inequality for the Frobenius norm, while we are frequently interested in the bounds for the operator norm that can be much smaller. An approach which often yields “dimension - free” bounds was proposed in and (using generalizations of the median in higher dimensions); however, to the best of our knowledge, results of these papers are still not sufficient to obtain deviation guarantees in the operator norm that we are mainly interested in. Under more restrictive assumptions on the sequence of random matrices (such as almost surely for some fixed , , where stands for the operator norm), behavior of the sample mean has been analyzed with the help of matrix concentration inequalities .
A closely related covariance matrix estimation problem has been extensively studied in the past decades. A comprehensive review is beyond the scope of this introduction, so we will just mention few classical results and more recent work related to the current line of research. Statistical properties of the sample covariance matrix for Gaussian and sub-Gaussian observations have been investigated in detail, see and references therein; under weaker moment assumptions, sample covariance estimator has been studied in . Some popular robust estimators of scatter are discussed in , including the Minimum Covariance Determinant (MCD) estimator and the Minimum Volume Ellipsoid estimator (MVE). However, rigorous results for these estimators are available only for elliptically symmetric distributions; see for results on MCD and for results on MVE. Popular Maronna’s and Tyler’s M-estimators of scatter also admit theoretical guarantees for the family of elliptically symmetric distributions, but we are unaware of any results extending beyond this case.
Finally, let us mention that the problem of robust matrix recovery (that is discussed as an example below) has also received attention recently: for instance, the work investigates robust matrix completion under the “low rank + sparse” model. In , authors study low-rank matrix recovery under the assumption that the additive noise has only moments, and obtain strong results via truncation argument. We propose a different approach based on general techniques developed in this paper and achieve similar results for the matrix completion problem while requiring only the finite variance of the noise.
2 Organization of the paper
Section 2 contains definitions, notation and background material. Our main results are introduced in section 3. After presenting core results, we discuss applications to covariance estimation and low-rank matrix completion in section 4, and illustrate the role of various quantities involved in the general bounds through these examples. Sections 5 and 6 discuss adaptation to unknown parameters that appear in our construction, and contain longer proofs.
Appendix contains proofs of several technical lemmas and results that were omitted in the main text.
Preliminaries
In this section, we introduce main notation and recall several useful facts from linear algebra, matrix analysis and probability theory that we rely on in the subsequent exposition.
Given two self-adjoint matrices and , we will write iff is nonnegative (or positive) definite.
2 Tools from linear algebra
In this section, we collect several facts from linear algebra, matrix analysis and probability theory that are frequently used in our arguments.
Additionally, we will often use the following facts:
Matrix logarithm is operator monotone: if and , then .
Given a fixed self-adjoint matrix , the function
is concave on the cone of positive definite matrices.
See and . Let us mention that Lieb’s theorem is one of the key tools for proving matrix concentration inequalities, and its power in this context was first demonstrated by J. Tropp . ∎
This is a consequence of Peierls inequality, see Theorem 2.9 in and the comments following it. ∎
Since it is easy to see that . Another tool useful in dealing with rectangular matrices is the following lemma:
Main results
See remark 1 below for examples of such functions. Given , let be such that
(clearly, always exists due to monotonicity). Set and . Assuming that , it is shown in that with probability .
where is an appropriate constant. It follows from Fact 2.6 that exists, moreover, it is unique if is strictly increasing. It is also not hard to see that (3.3) is equivalent to
Indeed, if , then (3.4) simply states that the gradient of evaluated at is equal to zero; see Lemma A.1 in the appendix for more details.
To understand the properties of the estimator defined via (3.3) and (3.4), we will first consider another estimator that shares many important properties with but is easier to analyze.
The “preliminary estimator” is constructed as follows: given and a function satisfying (3.1), set and
Assume that is large enough and is chosen properly. Then the estimator defined via (3.4) satisfies the inequality
Most of our results do not depend on the concrete choice of the function . One possibility is
Since the latter function is bounded, it can provide additional advantages (such as robustness) in applications. However, note that does not satisfy (3.1); instead, it satisfies a slightly weaker inequality
hence all subsequent results hold for as well, albeit with slightly worse constant factors. We also note that both and are operator Lipschitz functions; see Lemma A.3 for details.
In this section, we will establish deviation inequalities for the estimator . The lemma below is the cornerstone of our results. As before, given , let .
It remains to note that by Fact 2.1 and the inequality (that holds ), for all
To establish the second inequality of the lemma, we use the relation (which follows from (3.1) and Fact 2.1) together with the Fact 2.2 to deduce that
and apply inequality (3.8) to the sequence with
We are ready to state and prove the main result of this section.
In particular, setting , we get the “sub-Gaussian” tail bound , for a given . Alternatively, setting (independent of ), we obtain sub-exponential concentration with tail for all .
where was defined in (3.5).
As before, set . Then
where we used the second inequality of Lemma 3.1 instead. The result follows by taking since for a self-adjoint matrix , . ∎
Sub-Gaussian guarantees provided by Theorem 3.1 hold for a given confidence parameter that has to be fixed a priori: in particular, the optimal value of depends it. However, as it was noted in , this is sufficient to construct (via Lepski’s method ) estimators that admit sub-Gaussian tails uniformly over in a certain range.
2 Bounds depending on the effective dimension
The bound obtained in Theorem 3.1 explicitly depends on the dimension of random matrices. Example is subsection 3.2.1 below shows that the dimensional factor in the right-hand side of the inequality is unavoidable in general. However, it is possible to prove a similar inequality which only includes the “effective dimension” defined as
As before, we can set to get
For the values of (when the bound becomes useful), it further simplifies to
For the “sub-exponential regime” with , we get that for all simultaneously,
The argument is similar in spirit to the proof of Theorem 3.1. Details are included in appendix C. ∎
with for some absolute constant . Since , it follows from Lemma A.5 that
for any and some constant . This shows that the dimensional factor can not grow slower than for any .
3 Bounds for arbitrary rectangular matrices
and the first inequality follows. To obtain the second inequality, it is enough to use Theorem 3.2 instead of Theorem 3.1 and note that
4 Bounds under weaker moment assumptions
The argument repeats the steps of Lemma 3.1 and Theorem 3.1, the only difference being that application of Fact 2.4 is replaced by Lemma A.2. ∎
Note that for , we recover (3.11).
Before we proceed with discussion or further improvements and adaptation issues, let us demonstrate applications of developed techniques to popular problems in statistics and highlight the advantages over existing results.
Examples
We present two examples which highlight the potential improvements obtained via our technique in popular scenarios: estimation of the covariance matrix in Frobenius and operator norms, and low-rank matrix completion problem.
Note that for any matrix of rank (where ),
Of course, the initial assumption that is known is often unrealistic, hence we modify the estimator as follows. Given , set
Before presenting the proof, let us make several additional remarks.
It is not hard to show that (see Corollary A.1) that
Construction of essentially halves the effective sample size. While the loss of a constant factor can be deemed insignificant in non-asymptotic theoretical bounds, it is undesirable in applications. A more natural version of the estimator based on a sample of size is the U-statistic
Another possibility to avoid “halving” the sample size is to center the data using a robust estimator of location, such as the spatial median or the median-of-means estimator . Analysis of the estimators of these types is not covered in the present paper, and requires a slightly different set of technical tools to deal with dependent summands; see for results in this direction.
2 Estimation of the covariance matrix in Frobenius norm
Next, we present an estimator which achieves strong deviation guarantees in the Frobenius norm. Estimation of the covariance matrix with respect to this norm has been previously investigated in the literature, for instance, see , and references therein; Frobenius norm is a natural choice when one wants to understand the effect of the rank of an unknown covariance matrix on the estimation error . Let be the sample covariance estimator based on :
The following “soft thresholding” estimator has been studied in ; here, is a fixed threshold parameter:
We propose to replace the sample covariance by , and consider
It is not hard to see (e.g., see the proof of Theorem 1 in ) that can be written explicitly as
where and are the eigenvalues and corresponding eigenvectors of . The following result holds:
Result stated above mimics the (almost) optimal rates obtained in (in the situation when no data is missing) under significantly weaker assumptions on the underlying distribution.
The proof is based on the following lemma:
Inequality (4.3) holds on the event .
To verify this statement, it is enough to repeat the steps of the proof of Theorem 1 in , replacing each occurrence of the sample covariance by its robust counterpart . Result then follows from corollary 4.1 that whenever . ∎
3 Matrix completion
To incorporate the structural (low-rank) assumption on , the following estimator has been considered in the literature: let , and define
However, strong theoretical guarantees for this estimator exist only when the “noise term” is either bounded with probability 1, or has sub-exponential tails. We propose to replace with a robust estimator
The reasoning behind this choice of is explained below. Consider
Assume that is independent of , and that . For any
By the definition of , we see that
If we replace by , the result follows from Theorem 1 in immediately. To obtain the current statement, it is enough to repeat the argument of Theorem 1 in , replacing each occurrence of the matrix by . ∎
To complete the proof, we will estimate each side of the inequality of Lemma 4.2. First, it is obvious from the definition of the Frobenius norm that
It remains to estimate the probability of the event . Let
Assume that is independent of , . Then
with probability . Final result now follows from the combination of this inequality with (4.4), (4.3) and Lemma 4.2.
Optimal choice of θ𝜃\theta and adaptation to the unknown second moment
Parameters and are “crude” preliminary bounds that can differ from by several orders of magnitude. Let and
be a set of cardinality , and for each set . Define
where satisfies (3.1). Finally, set
Next result shows that adaptation is possible at the cost of an additional multiplicative constant factor in the deviation bound.
The following inequality holds for any :
Let (hence ). First, we will show that with high probability. Indeed,
where we used Theorem 3.1 to bound each of the probabilities in the sum. The display above implies that the event
of probability is contained in . Hence, on we have that
Let and be known constants such that
We will first discuss the simplest (but not the most efficient) approach based on splitting the sample into two disjoint subsets and of cardinality each, and performing one step of the steepest descent. The main advantage of this approach is the fact that it requires very mild assumptions. The idea is to apply Lepski’s method (as discussed in section 5) twice: on the first step, we obtain an estimator based on subsample , and on the second step we apply Lepski’s method again to the subsample .
Here is the more detailed description: set ,
and ,
and let be the “Lepski-type” adaptive estimator based on the subsample defined as
, satisfies (3.1) and
is then defined as follows:
The main feature of this result is the variance term that can be much smaller compared to as long as .
We will next show how to design an estimator with deviations controlled by “correct” variance term without sample splitting (however, subject to the condition that the sample size is sufficiently large). In what follows, we will make an additional assumption about the function :
For example, we may take or (see Lemma A.3 for details). As before, let be fixed, set ,
For all , define and
for . Next, for each , we define
for . Finally, we apply Lepski’s method to the collection of estimators . To this end, define , where
Note that the estimator is completely data-dependent. We are ready to state the main result of this section:
where is an absolute constant, and assume that . Moreover, assume that
with probability .
The next corollary easily follows from the preceding result. Let be the event of probability
defined in Theorem 6.2. Since by the properties of the steepest descent scheme converges to the solution (denoted ) of the problem (3.3), we can easily deduce the following inequality.
Let satisfy the equations
One can further apply Lepski’s method (see section 5) to the collection to obtain a completely data-dependent estimator that satisfies
with high probability (in particular, on event ).
Numerical simulation results
The goal of numerical experiment was to evaluate the quality of estimation of the covariance matrix as well as its first eigenvector corresponding to . We tested two scenarios with sample sizes equal to and . In both cases, we generated , i.i.d. copies of , and centered the data via the spatial (or geometric) median defined as
We compared two estimators, and constructed as follows: set for brevity, and
which is the analogue of sample covariance with “robust centering”.
Next, was constructed using a version of Lepski’s method described in section 5. We provide details for completeness: set
and let be the function defined in (3.6). Let , and for , set and Finally, define
(note that we modified some constants compared to the “theoretical” version), and finally set .
Quality of covariance estimation was evaluated via comparing with over 500 runs of simulations. We also compared errors of estimation of projectors onto the first principal component,
where denotes the eigenvector corresponding to the largest eigenvalue of a matrix. Histograms illustrating performance of both estimators are presented in figures 1a and 1b (for the sample size ), and in figures 2a and 2b (for the sample size ). It is clear from the graphs that in all scenarios, performs significantly better than .
Acknowledgements
I want to thank L. Goldstein, A. Juditsky, A. Nemirovski, as well as the anonymous Referees and the Associate Editor for their insightful suggestions that helped to improve the quality of presentation.
References
Appendix A Supplementary technical results
hence the claim holds for monomials. By linearity, it also holds for arbitrary polynomials. It remains to extend the claim to arbitrary continuously differentiable function via a standard approximation argument (for instance, see [4, chapter 5, section 3]).
Let and . Then and
To check the first claim, it is enough to note that is convex and its minimum is attained for . It is easy to check that , which implies that which always holds since and .
Choosing , , we get which is further bounded above by for . ∎
Functions and defined in Remark 1 are operator Lipschitz, with Lipschitz constants independent of the dimension.
Lipshitz property of follows from Theorem 1.6.1 in . Result for follows from Theorem 1.1.1 in the same paper. ∎
For a self-adjoint matrices , iff . Clearly,
It implies that and . Since , we obtain
The following lemma is a generalization of Chebyshev’s association inequality.
where the last inequality follows from Lemma A.4 inequality by setting and . ∎
for every .
Appendix B Tools from probability theory and linear algebra
We recall several useful results that we will need in the proofs below.
Then with probability .
We conclude this section by recalling the notion of Talagrand’s generic chaining complexity (see ) and several related results. Given a metric space , let be a nested sequence of partitions of such that and . For , let be the unique subset of containing . The generic chaining complexity is defined as
See Theorem 3.2 in for a more general statement. ∎
Appendix C Proof of Theorem 3.2
Define and . Proceeding as in the proof of Theorem 3.1, we get that for ,
Here we have used the fact that implies for , and the equality . We have shown that
where we used an elementary inequality on the last step.
Combining the same steps with Fact 2.4 and the equality , we get
Finally, replace by to get the bound in the required form.
Appendix D Proof of Theorem 6.1
Let be the event defined by
In particular, on event ,
Here, we use the definition of on the first step and (D.3) on the second step. The last inequality follows from independence of from (under ) and Theorem 5.1 applied conditionally on : indeed, this can be done since (D.3) holds on . It remains to combine the last bound with (D.4) and (D.1). ∎
Appendix E Proof of Theorem 6.2
We will first state several technical results that are required in the proof. Let be such that , and define
which is a consequence of scalar inequality and fact M.2.1, hence we can deduce from fact M.2.2 that
by the definition of , we conclude that
Result follows from Theorem M.3.1 and the inequality
which is a consequence of lemma E.2. Indeed,
We are ready to proceed with the proof of the theorem. Let
where was defined in M.6.2 as , and note that by Theorem M.3.1. Let
and note that . Define
Expression under the supremum in (E.2) can be decomposed as follows:
We will treat 3 terms separately: first, it follows from Lemma E.2 that on
Putting the bounds (E.3),(E.4),(E.5) together, we can estimate the supremum in (E.2) as
Note that we have used bounds and (indeed, inequality (6.3) implies that for all and ) to get the second inequality above. Since was chosen such that and by assumption, we have shown that
where the last equality follows from the fact that the sequence defined in (6.1) satisfies the recursive relation
To complete the proof, it is enough to follow the steps of the proof Theorem 5.1 applied to the collection of estimators : first, let , and note that the event
has probability . Moreover, on this event , hence
where we used the fact that in the last inequality. ∎
To this end, we will use a chaining argument. Recall that the function is operator Lipschitz with Lipschitz constant by assumption. Recall that . It follows from Assumption 2 (see also Lemma A.3) that for any Hermitian and ,
Matrix Hoeffding’s inequality (Lemma B.1) applies with
and , and yields that
where denotes the Lebesgue measure of a set , and stands for the Minkowski sum of the sets and . For , we get . The volume of the unit ball is given by
with probability . Recall the Dudley’s entropy integral bound (B.1):
Noting that and combining Dudley’s bound with the estimate of Lemma E.4, we get
where . Bound (E.7) implies that with probability ,