High Dimensional Robust $M$-Estimation: Arbitrary Corruption and Heavy Tails
Liu Liu, Tianyang Li, Constantine Caramanis

TL;DR
This paper introduces a flexible framework for high-dimensional $M$-estimation under heavy tails and arbitrary corruptions, providing new robust algorithms with optimal statistical guarantees.
Contribution
The paper defines the Robust Descent Condition (RDC) and shows that it enables robust, minimax-optimal $M$-estimation algorithms in heavy-tailed and corrupted data scenarios.
Findings
Median-of-means gradient estimator satisfies RDC for heavy tails.
Trimmed gradient estimator satisfies RDC for arbitrary corruptions.
Robust Hard Thresholding achieves minimax optimal rates in tested scenarios.
Abstract
We consider the problem of sparsity-constrained -estimation when both explanatory and response variables have heavy tails (bounded 4-th moments), or a fraction of arbitrary corruptions. We focus on the -sparse, high-dimensional regime where the number of variables and the sample size are related through . We define a natural condition we call the Robust Descent Condition (RDC), and show that if a gradient estimator satisfies the RDC, then Robust Hard Thresholding (IHT using this gradient estimator), is guaranteed to obtain good statistical rates. The contribution of this paper is in showing that this RDC is a flexible enough concept to recover known results, and obtain new robustness results. Specifically, new results include: (a) For -sparse high-dimensional linear- and logistic-regression with heavy tail (bounded 4-th moment) explanatory and response…
Peer Reviews
No public reviews on file for this paper yet. If you reviewed it on a platform where reviews are public (OpenReview, ICLR, NeurIPS, ICML), you can paste yours below so the community can read it here.
Videos
No videos yet. Explain this paper in a talk, walkthrough, or lecture? Add one.
Taxonomy
TopicsSparse and Compressive Sensing Techniques · Statistical Methods and Inference · Stochastic Gradient Optimization Techniques
High Dimensional Robust -Estimation: Arbitrary Corruption and Heavy Tails
Liu Liu
Tianyang Li
Constantine Caramanis
The University of Texas at Austin
Abstract
We consider the problem of sparsity-constrained -estimation when both explanatory and response variables have heavy tails (bounded 4-th moments), or a fraction of arbitrary corruptions. We focus on the -sparse, high-dimensional regime where the number of variables and the sample size are related through . We define a natural condition we call the Robust Descent Condition (RDC), and show that if a gradient estimator satisfies the RDC, then Robust Hard Thresholding (IHT using this gradient estimator), is guaranteed to obtain good statistical rates. The contribution of this paper is in showing that this RDC is a flexible enough concept to recover known results, and obtain new robustness results. Specifically, new results include: (a) For -sparse high-dimensional linear- and logistic-regression with heavy tail (bounded 4-th moment) explanatory and response variables, a linear-time-computable median-of-means gradient estimator satisfies the RDC, and hence Robust Hard Thresholding is minimax optimal; (b) When instead of heavy tails we have -fraction of arbitrary corruptions in explanatory and response variables, a near linear-time computable trimmed gradient estimator satisfies the RDC, and hence Robust Hard Thresholding is minimax optimal. We demonstrate the effectiveness of our approach in sparse linear, logistic regression, and sparse precision matrix estimation on synthetic and real-world US equities data.
1 Introduction
-estimation is a standard technique for statistical estimation [vdV00]. The past decade has seen successful extensions of -estimation to the high dimensional setting with sparsity (or other low-dimensional structure), e.g., using Lasso [Tib96, BvdG11, HTW15, Wai19]. Yet sparse modeling in high dimensions is NP-hard in the worst case [BDMS13, ZWJ14]. Thus theoretical sparse recovery guarantees for most computationally tractable approaches (e.g., minimization [Don06, CRT04, Wai09], Iterative Hard Thresholding [BD09]) rely on strong assumptions on the probabilistic models of the data, such as sub-Gaussianity. Under such assumptions, these approaches achieve the minimax rate for sparse regression [RWY11].
Meanwhile, statistical estimation with heavy tailed outliers or even arbitrary corruptions has long been a focus in robust statistics [Box53, Tuk75, Hub11, HRRS11].111Following [Min18, FWZ16], by heavy-tail we mean satisfying only weak moment bounds, specifically, bounded 4-th order moments (compared to sub-exponential or sub-Gaussian). But heavy-tails and arbitrary corruptions in the data violate the assumptions required for convergence of the usual algorithms. A central question then, is what assumptions are sufficient to enable efficient and robust algorithms for high dimensional -estimation under heavy tails or arbitrary corruption.
Huber’s seminal work [Hub64] and more modern followup work [Loh17] has considered replacing the classical least squared risk minimization objective with a robust counterpart (e.g., Huber loss). Other approaches (e.g., [Li13]) considered regularization-based robustness approaches. However, when there are outliers in the explanatory variables (covariates), these approaches do not seem to succeed [CCM13]. Meanwhile, approaches combining recent advances in robust mean estimation and gradient descent have proved remarkably powerful in the low-dimensional setting [PSBR18, KKM18, DKK*+*18], but for high dimensions, have so far only managed to address the setting where the covariance of the explanatory variables is the identity, or sparse [BDLS17, LSLC18]. Meanwhile, flexible and statistically optimal approaches ([Gao17]) have relied on intractable estimators such as Tukey-depth.
For the heavy-tail setting, another line of research considers estimators such as Median-of-Means (MOM) [NY83, JVV86, AMS99, Min15] and Catoni’s mean estimator [Cat12, Min18] only use weak moment assumptions. [Min15, BJL15, HS16] generalized these ideas to -estimation, yet it is not clear if these approaches apply to the high-dimensional setting with heavy tailed covariates.
Main Contributions. In this paper, we develop a sufficient condition that when satisfied, guarantees that an efficient algorithm (a variant of IHT) achieves the minimax optimal statistical rate. We show that our condition is flexible enough to apply to a number of important high-dimensional estimation problems under either heavy tails, or arbitrary corruption of the data. Specifically:
We consider two models. For our arbitrary corruption model, we assume that an adversary replaces an arbitrary -fraction of the authentic samples with arbitrary values (2.1). For the heavy-tailed model, we assume our data (response and covariates) satisfy only weak moment assumptions (2.2) without sub-Gaussian or sub-exponential concentration bounds. 2. 2.
We propose a notion that we call the Robust Descent Condition (RDC). Given any gradient estimator that satisfies the RDC, we define RHT – Robust Hard Thresholding (Algorithm 1) for sparsity constrained -estimation, and prove that Algorithm 1 converges linearly to a minimax statistically optimal solution. Thus the RDC and Robust Hard Thresholding form the basis for a Deterministic Meta-Theorem (3.1) that guarantees estimation error rates as soon as the RDC property of any gradient estimator can be certified. 3. 3.
We then obtain non-asymptotic bounds via certifying the RDC for different robust gradient estimators under various statistical models. (A) For corruptions in both response and explanatory variables, we show the trimmed gradient estimator satisfies the RDC. Thus our algorithm RHT has minimax-optimal statistical error, and tolerates -fraction of outliers. This fraction is nearly independent of the , which is important in the high dimension regime. (B) In the heavy tailed regime, we use the Median-of-Means (MOM) gradient estimator. Our RHT algorithm obtains the sharpest available error bound, in fact nearly matching the results in the sub-Gaussian case. With either of these gradient estimators, our algorithm is computationally efficient, nearly matching vanilla gradient descent. This is in particular much faster than algorithms relying on sparse PCA relaxations as subroutines ([BDLS17, LSLC18]). 4. 4.
We use Robust Hard Thresholding for neighborhood selection [MB06] for estimating Gaussian graphical models, and provide model selection guarantees under adversarial corruption of the data; our results share similar robustness guarantees with sparse regression. 5. 5.
We demonstrate the effectiveness of Robust Hard Thresholding on both arbitrarily corrupted/heavy tailed synthetic data and (unmodified) real data.
A concrete illustration of 3(B) above: Consider a sparse linear regression problem without noise (sparse linear equations), with scaling . When the covariates are sub-Gaussian, Lasso succeeds in exact recovery with high probability (as expected). When the covariates have only 4-th moments, we do not expect Lasso to succeed, and indeed experiments indicate this. Moreover, to the best of our knowledge, no previous efficient algorithm with samples can guarantee exact recovery in this observation model ([FWZ16] has a statistical rate depending on the norm of the parameter , and thus exact recovery for is not guaranteed). Our contributions show that Robust Hard Thresholding using MOM achieves this (see also simulations in Figure 2(b)).
Related work
Sparse regression with arbitrary corruptions or heavy tails. Several works in robustness of high dimensional problems consider heavy tailed distributions or arbitrary corruptions only in the response variables [Li13, BJK15, BJK17, Loh17, KP18, HS16, Min15, CLZL]. Yet these algorithms cannot be trivially extended to the setting with heavy tails or corruptions in explanatory variables. Another line [ACG13, VMX17, YLA18, SS18] focuses on alternating minimization approaches which extend Least Trimmed Squares [Rou84]. However, these methods only have local convergence guarantees, and cannot handle arbitrary corruptions.
[CCM13] was one of the first papers to provide guarantees for sparse regression with arbitrary outliers in both response and explanatory variables by trimming the design matrix. Similar trimming techniques are also used in [FWZ16] for heavy tails in response and explanatory variables. Those results are specific to sparse regression, however, and cannot be easily extended to general -estimation problems. Moreover, even for linear regression, the statistical rates are not minimax optimal. [LM16] uses Median-of-Means tournaments to deal with heavy tails in the explanatory variables and obtains near optimal rates. However, Median-of-Means tournaments is not known to be computationally tractable. [LL17] deals with heavy tails and outliers in the explanatory variables, but they require higher moment bound (whose order is ) in the isotropic design case. [Gao17] optimizes Tukey depth [Tuk75, CGR18] for robust sparse regression under the Huber -contamination model, and their algorithm is minimax optimal and can handle a constant fraction of outliers. However, computing Tukey depth is intractable [JP78]. Recent results [BDLS17, LSLC18] leverage robust sparse mean estimation in robust sparse regression. Their algorithms are computationally tractable, and can tolerate , but they require very restrictive assumptions on the covariance matrix ( or sparse), which precludes their use in applications such as graphical model estimation.
Robust -estimation via robust gradient descent. Works in [CSX17, HI17] and later [YCRB18a] first leveraged the idea of using robust mean estimation in each step of gradient descent, using a subroutine such as geometric median. A similar approach using more sophisticated robust mean estimation methods was later proposed in [PSBR18, DKK*+*18, YCRB18b, SX18, Hol18] for robust gradient descent. These methods all focused on low dimensional robust -estimation. Work in [LSLC18] extended the approach to the high-dimensional setting (though is limited to or sparse covariances). Even though the corrupted fraction can be independent of the ambient dimension by using sophisticated robust mean estimation algorithms [DKK*+*16, LRV16, SCV17], or the sum-of-squares framework [KKM18], these algorithms (except [LSLC18]) are not applicable to the high dimensional setting (), as they require at least samples.
Robust estimation of graphical models. A line of research using a robustified covariance matrix in Gaussian graphical models [LHY*+*12b, WG17, LT18] leverages GLasso [FHT08] or CLIME [CLL11] to estimate the sparse precision matrix. These robust methods are restricted to Gaussian graphical model estimation, and their techniques cannot be generalized to other -estimation problems.
Notation. We denote the Hard Thresholding operator of sparsity by , and denote the Euclidean projection onto the ball by . We use to denote the expectation operator obtained by the uniform distribution over all samples .
2 Problem formulation
We now define the corruption and heavy tails model and sparsity constrained -estimation.
Definition 2.1** (-corrupted samples).**
Let be i.i.d. observations with distribution . We say that a collection of samples is -corrupted if an adversary chooses an arbitrary -fraction of the samples in and modifies them with arbitrary values.
This corruption model allows corruptions in both explanatory and response variables in regression problems where we observe . 2.1 also allows the adversary to select an -fraction of samples to delete and corrupt.
Definition 2.2** (heavy-tailed samples).**
For a distribution of with mean and covariance , we say that has bounded -th moment, if there is a universal constant such that, for a unit vector , we have .
2.2 allows heavy tails in both explanatory and response variables for . For example, in 4.3, we study linear regression with bounded 4-th moments for and bounded variance for and noise.
Let be a convex and differentiable loss function. Our target is the unknown sparse population minimizer , and we write as the population risk, . Note that ’s definition allows model misspecification. The following 2.3 provides general assumptions for the population risk.
Definition 2.3** (Strong convexity/smoothness).**
For the population risk , we assume , where is the strong-convexity parameter and is the smoothness parameter. The condition number is .
A well known result [NRWY12] considers ERM with convex relaxation from to , by certifying the RSC condition for sub-Gaussian ensembles – this obtains uniform convergence of the empirical risk. From an optimization viewpoint, existing results reveal that gradient descent algorithms equipped with soft-thresholding [ANW12] or hard-thresholding [BD09, JTK14, SL17, YLZ18, LB18] have linear convergence rate, and achieve known minimax lower bounds in statistical estimation [RWY11, ZWJ14].
Given samples , running ERM on the entire input dataset: , cannot guarantee uniform convergence of the empirical risk, and can be arbitrarily bad for -corrupted samples. The next two sections outline the main results of this paper, addressing this problem.
3 Robust sparse estimation
via Robust Hard Thresholding
We introduce our meta-algorithm, Robust Hard Thresholding, that essentially uses a robust gradient estimator to run IHT. We require several definitions to specify the algorithm, and describe its results. We use as a placeholder for the estimate at , obtained from whichever robust gradient estimator we are using. Let denote the population gradient. We use and when the context is clear.
Many previous works ([CSX17, HI17, PSBR18, DKK*+*18, YCRB18a, YCRB18b, SX18]) have provided algorithms for obtaining robust gradient estimators, then used as subroutines in robust gradient algorithms. However, those results require controlling , and do not readily extend to high dimensions, as sufficiently controlling seems to require . A recent work [LSLC18] on robust sparse linear regression uses a robust sparse mean estimator [BDLS17] to guarantee with sample complexity . However, their algorithm requires the restrictive assumption or sparse, and thus cannot be extended to more general -estimation problems.
To address this issue, we propose Robust Hard Thresholding (Algorithm 1), which uses hard thresholding after each robust gradient update222Our theory requires splitting samples across different iterations to maintain independence between iterations. We believe this is an artifact of the analysis, and do not use this in our experiments. [BWY17, PSBR18] use a similar approach for theoretical analysis. . In line 7, we use a gradient estimator to obtain the robust gradient estimate . In line 8, we update the parameter by hard thresholding , where the hyper-parameter proportional to is specified in 2.3. A key observation in line 8 is that, in each step of IHT, the iterate is sparse, and thus the perturbation from outliers or heavy tails only depends on IHT’s sparsity instead of the ambient dimension . Based on a careful analysis of the hard thresholding operator in each iteration, we show that rather than controlling , it is enough to control a weaker quantity: this is what we call the Robust Descent Condition 3.1 and we define it next; it plays a key role in obtaining sharp rates of convergence for various types of statistical models.
Robust Descent Condition
The Robust Descent Condition eq. 1 provides an upper bound on the inner product of the robust gradient estimate and the distance to the population optimum. This is a natural notion to control the potential progress obtained by using a robust gradient update instead of the population gradient.
Definition 3.1** (-Robust Descent Condition (RDC)).**
For the population gradient at , a robust gradient estimator satisfies the robust descent condition if for any sparse ,
[TABLE]
We begin with a Meta-Theorem for Algorithm 1 that holds under the Robust Descent Condition 3.1 and assumptions on population risk 2.3. In 3.1, we prove Algorithm 1’s global convergence and its statistical guarantees. The proofs are collected in Appendix B.
Theorem 3.1** (Meta-Theorem).**
Suppose we observe samples from a statistical model with population risk satisfying 2.3. If a robust gradient estimator satisfies -Robust Descent Condition (3.1) where , then Algorithm 1 with outputs such that , by setting .
We note that 3.1 is deterministic in nature. In the sequel, we omit the log term in the sample complexity due to sample splitting. We obtain high probability results via certifying that the RDC holds for certain robust gradient estimators under various statistical models. To obtain the minimax estimation error rate in 3.1, the key step is providing a robust gradient estimator with sufficiently small , in the definition of RDC.
Section 4 uses the RDC and 3.1 to obtain new results for sparse regression under heavy tails or arbitrary corruption. Before we move to this, we observe that we can use the RDC and 3.1 to recover existing results in the literature. Some immediate examples are as follows:
**Uncorrupted gradient satisfies the RDC. ** Suppose the samples follow from sparse linear regression with sub-Gaussian covariates and noise . The empirical average of gradient samples satisfies eq. 1 with , by assuming constraint on and [LW11]. Plugging in this to 3.1 recovers the well-known minimax rates for sparse linear regression [RWY11].
**RSGE implies RDC. ** When or is sparse, [BDLS17] and [LSLC18], respectively, provide robust sparse gradient estimators (RSGE) which upper bound , for a constant fraction of corrupted samples. Noting that , we observe that RSGE implies RDC. Hence any RSGE can be used in Algorithm 1. The RSGE for in [BDLS17] guarantees an RDC with when , and the RSGE for unknown sparse from [LSLC18] guarantees when . Again plugging these values for into our theorem, recovers the results in those papers. 333It remains an open question to obtain a RSGE for a constant fraction of outliers for robust sparse regression with arbitrary covariance .
4 Main Results: Using the RDC and Algorithm 1
In the remainder of our paper, we use 3.1 and the RDC to analyze two well-known and computationally efficient robust mean estimation subroutines that have been used in the low-dimensional setting: the trimmed mean estimator and the MOM estimator. We show that these two can obtain a sufficiently small in the definition of the RDC. This leads to the minimax estimation error in the case of arbitrary corruptions or heavy tails.
4.1 Gradient estimation
The trimmed mean and MOM estimators have been successfully applied to robustify gradient descent [YCRB18a, PSBR18] in the low dimensional setting. They have not been used in the high dimensional regime, however, because until now we have not had the machinery to analyze their algorithmic convergence, statistical rates and minimax optimality in the high dimensional setting.
To show they satisfy the RDC with a sufficiently small , we observe that by using Hölder’s inequality on the LHS of eq. 1, we have Using Algorithm 1, the Hard Thresholding step enforces sparsity of . Therefore, controlling amounts to bounding the infinity norm of the robust gradient estimate.
In Section 4.2, we show that by using coordinate-wise robust mean estimation, we can certify the RDC with sufficiently small to guarantee minimax rates. Specifically, we show this for the trimmed gradient estimator for arbitrary corruption, and and the MOM gradient estimator for heavy tailed distributions.
Definition 4.1**.**
Given gradients samples , for each dimension ,
: Trimmed gradient estimator removes the largest and smallest fraction of elements in , and calculates the mean of the remaining terms. We choose for constant , and require for a small constant .
: MOM gradient estimator partitions into blocks and computes the sample mean of within each block, and then take the median of these means.444Without loss of generality, we assume the number of blocks divides , and is chosen in [HS16].
4.2 Statistical guarantees
In this section, we consider some typical models for general -estimation.
Model 4.1** (Sparse linear regression).**
Samples are drawn from a linear model : , with being -sparse. We assume that ’s are i.i.d. with normalized covariance matrix , with , and the stochastic noise has mean [math] and variance .
Model 4.2** (Sparse logistic regression).**
Samples are drawn from a binary classification model , where the binary label follows the conditional probability distribution , with being -sparse. We assume that ’s are i.i.d. with normalized covariance matrix , where for all .
To obtain the following corollaries, we first certify the RDC for a certain robust gradient estimator over random ensembles with corruption or heavy tails, and then use them in 3.1. We collect the results for gradient estimation in Appendix A, and the proofs for corollaries in Appendix B.
Arbitrary corruption case.
Based on 3.1, we first provide concrete results for arbitrary corruption case 2.1, where the covariates and response variables in the authentic distribution are assumed to be sub-Gaussian.
Corollary 4.1**.**
Suppose we observe -corrupted (2.1) sub-Gaussian samples from sparse linear regression model (4.1). Under the condition , and \epsilon=O\Bigl{(}\frac{1}{\rho^{2}\sqrt{k}\log(nd)}\Bigr{)}, with probability at least , Algorithm 1 with trimmed gradient estimator satisfies the RDC with , and thus 3.1 provides
Time complexity. 4.1 has a global linear convergence rate. In each iteration, we only use operations complexity to calculate trimmed mean. We incur logarithmic overhead compared to normal gradient descent [Bub15].
Statistical accuracy and robustness. Compared with [CCM13, BDLS17], our statistical error rate is minimax optimal [RWY11, ZWJ14], and has no dependencies on . Furthermore, the upper bound on is nearly independent of , which guarantees Algorithm 1’s robustness in high dimensions.
Corollary 4.2**.**
Suppose we observe -corrupted (2.1) sub-Gaussian samples from sparse logistic regression model (4.2). With probability at least , Algorithm 1 with trimmed gradient estimator satisfies the RDC with , and thus 3.1 provides .
Statistical accuracy and robustness. Under the sparse Gaussian linear discriminant analysis model (a typical example of 4.2), Algorithm 1 achieves the statistical minimax rate [LPR15, LYCR17].
Heavy-tailed distribution case.
We next turn to the heavy tailed distribution case 2.2.
Corollary 4.3**.**
Suppose we observe samples from sparse linear regression model (4.2) with bounded 4-th moment covariates. Under the condition , with probability at least , Algorithm 1 with MOM gradient estimator satisfies the RDC with , and thus 3.1 provides .
Time complexity. Similar to 4.1, 4.3 has a global linear convergence. In each iteration, we only use operations complexity – the same as normal gradient descent [Bub15].
Statistical accuracy. [LM16] uses Median-of-Means tournaments to deal with sparse linear regression with bounded moment assumptions for the covariates, and they obtain near optimal rates. We obtain similar rates, however our algorithm is efficient, where as Median-of-Means tournaments is not known to be computationally tractable. [FWZ16, Zhu17] deal with the same problem by truncating and shrinking the data to certify the RSC condition. Their results require boundedness of higher moments of the noise , and the final error depends on . Our estimation error bounds exactly recover optimal sub-Gaussian bounds for sparse regression [NRWY12, Wai19], and moreover, we obtain exact recovery when ’s variance .
Corollary 4.4**.**
Suppose we observe samples from sparse logistic regression model (4.2). With probability at least , Algorithm 1 with MOM gradient estimator satisfies the RDC with , and thus 3.1 provides .
4.3 Sparsity recovery and Gaussian graphical model estimation
We next demonstrate the sparsity recovery performance of Algorithm 1 for graphical model learning [MB06, Wai09, RWL10, RWRY11, BvdG11, HTW15]. Our sparsity recovery guarantees hold for both heavy tails and arbitrary corruption, though we only present results in the case of arbitrary corruption in this section.
We use to denote top indexes of with the largest magnitude. Let denote the smallest absolute value of nonzero element of . To control the false negative rate, 4.5 shows that under the -condition, is exactly . The proofs are given in Appendix C. Sparsity recovery guarantee for sparse logistic regression is similar, and is omitted due to space constraints. Existing results on sparsity recovery for regularized estimators [Wai09, LSRC15] do not require the RSC condition, but instead require an irrepresentability condition, which is stronger. If , 4.5 has the same -condition as IHT for sparsity recovery [YLZ18].
Corollary 4.5**.**
Under the same condition as in 4.1, and a -condition on , , Algorithm 1 with trimmed gradient estimator guarantees that , with probability at least .
We consider sparse precision matrix estimation for Gaussian graphical models. The sparsity pattern of its precision matrix matches the conditional independence relationships [KFB09, WJ08].
Model 4.3** (Sparse precision matrix estimation).**
Under the contamination model 2.1, authentic samples are drawn from a multivariate Gaussian distribution . We assume that each row of the precision matrix is -sparse – each node has at most edges.
For the uncorrupted samples drawn from the Gaussian graphical model, the neighborhood selection (NS) algorithm [MB06] solves a convex relaxation of the following sparsity constrained optimization to regress each variable against its neighbors
[TABLE]
where denotes the -th coordinate of , and denotes the index set . Let denote ’s -th column with the diagonal entry removed. and denote the -th diagonal element of . Then, the sparsity pattern of can be estimated through . Details on the connection between and are given in Appendix C.
However, given -corrupted samples from the Gaussian graphical model, this procedure will fail [LHY*+*12b, WG17]. To address this issue, we propose Robust NS (Algorithm 2 in Appendix C), which robustifies Neighborhood Selection [MB06] by using Robust Hard Thresholding (with least square loss) to robustify eq. 2. Similar to 4.5, a -condition guarantees consistent edge selection.
Corollary 4.6**.**
Under the same condition as in 4.1, and a -condition for , , Robust NS (Algorithm 2) achieves consistent edge selection, with probability at least .
Similar to 4.1, the fraction is nearly independent of dimension , which provides guarantees of Robust NS in high dimensions. Other Gaussian graphical model selection algorithms include GLasso [FHT08], CLIME[CLL11]. The experimental details comparing robustified versions of these algorithms are presented in Section D.4.
5 Experiments
We provide the complete details for our experiment setup in Appendix D.
**Sparse regression with arbitrary corruption. ** We generate samples from a sparse regression model (4.1) with a Toeplitz covariance . Here, the stochastic noise , and we vary the noise level in different simulations. We add outliers with , and track the parameter error in each iteration. Left plot of Figure 2 shows Algorithm 1’s linear convergence, and the error curves flatten out at the final error level. Furthermore, Algorithm 1 can achieve machine precision when , which means exactly recovering of .
**Sparse regression with heavy tails. ** We consider a log-normal distribution (a typical example of heavy tails) in 4.1. More specifically, , and . Here, is the same Toeplitz covariance, each entry of and follows from , where . We fix , and vary sample size . For log-normal samples, we run Algorithm 1 with MOM and vanilla Lasso. We then re-generate standard Gaussian samples using the same dimensions with and run Vanilla Lasso. Each curve in the right plot of Figure 2 is the average of 50 trials. Algorithm 1 with MOM significantly improves vanilla Lasso on log-normal data, and has the same performance as Lasso on sub-Gaussian data
**Real data experiments. ** We next apply Algorithm 2, to a US equities dataset [LHY*+*12a, ZLR*+*12], which is heavy-tailed and has many outliers [dP18]. The dataset contains 1,257 daily closing prices of 452 stocks (variables). It is well known that stocks from the same sector tend to be clustered together [Kin66]. Therefore, we use Robust NS (Algorithm 2) to construct an undirected graph among stocks. Graphs estimated by different algorithms are shown in Figure 2. We can see that stocks from the same sector are clustered together, and these clustering centers can be easily identified. We also compare Algorithm 2 to the baseline NS approach (as in the ideal setting). We can observe that stocks from Information Technology (colored by purple) are much better clustered by Algorithm 2.
Notations in Appendix.
In our proofs, the exponent in tail bounds is arbitrary, and can be changed to other larger constant without affecting the results. denote universal constants, and they may change line by line.
Appendix A Proofs for the gradient estimators
In Robust Hard Thresholding (Algorithm 1), we use trimmed gradient estimator or MOM gradient estimator. And in 3.1, the key quantity to control the statistical rates of convergence is the Robust Descent Condition (3.1).
By Holder inequality, we have
[TABLE]
In this section, we provide one direct route for obtaining upper bound of Robust Descent Condition via bounding the infinity norm of the robust gradient estimate (A.1 and A.2).
Later, in Appendix B, we will leverage A.1 and A.2 in verifying the Robust Descent Condition for trimmed/MOM gradient estimator under sparse linear/logistic regression. Together with 3.1, this will complete 4.1 – 4.4.
Proposition A.1**.**
Suppose we observe -corrupted sub-Gaussian samples (2.1). With probability at least , the coordinate-wise trimmed gradient estimator can guarantee
- •
* for sparse linear regression (4.1).*
- •
* for sparse logistic regression (4.2).*
Proposition A.2**.**
Suppose we observe samples from the heavy tailed model with bounded 4-th moment covariates. With probability at least , the coordinate-wise Median of Means gradient estimator can guarantee
- •
* for sparse linear regression;*
- •
* for sparse logistic regression.*
A.1 Proofs for the MOM gradient estimator
We first prove A.2. A.1 of trimmed gradient estimator for -corrupted sub-Gaussian samples has the same dependency on . The proof of A.1 leverages standard concentration bound for sub-Gaussian samples, and then uses trimming to control the effect of outliers.
Proof of A.2.
For loss function, we have , where we omit the subscript in the proof. We denote , and bound the operator norm of the covariance of gradient samples
[TABLE]
where (i) follows from the Holder inequality, and (ii) follows from the 4-th moment bound assumption.
Hence, by using coordinate-wise Median of Means gradient estimator, we have
[TABLE]
with probability at least , where (i) follows from Proposition 5 in [HS16]. Applying union bounds on all d indexes, we have with probability at least .
For logistic loss, the gradient can be computed as: where we omit the subscript in the proof.
Since , and , we directly have Similar to the case of loss, we have , with probability at least .
∎
A.2 Proofs for the trimmed gradient estimator
We then turn to the trimmed gradient estimator for the case of arbitrary corruption. Before we proceed to the trimmed estimator, let us first visit the definition and tail bounds of sub-exponential random variable, as it will be used in sparse linear regression, where the gradient’s distribution is indeed sub-exponential under the sub-Gaussian assumptions in 4.1.
We first present standard concentration inequalities ([Wai19]).
Definition A.1** (Sub-exponential random variables).**
A random variable with mean is sub-exponential if there are non-negative parameters such that
[TABLE]
Lemma A.1** (Bernstein’s inequality).**
Suppose that , are i.i.d. sub-exponential random variables with parameters . Then
[TABLE]
We also have a two-sided tail bound
[TABLE]
We define -trimmed mean estimator for one dimensional samples, and denote it as .
Definition A.2** (-trimmed mean estimator).**
Given a set of -corrupted samples , the coordinate-wise trimmed mean estimator removes the largest and smallest fraction of elements in , and calculate the mean of the remaining terms. We choose , for a constant . We also require that , for some small constant .
A.2 shows the guarantees for this robust gradient estimator in each coordinate. We note that A.2 is stronger than guarantees for trimmed mean estimator (Lemma 3) in [YCRB18a]. In our contamination model 2.1, the adversary may delete -fraction of authentic samples, and then add arbitrary outliers. And A.2 provides guarantees for trimmed mean estimator on sub-exponential random variables. The trimmed mean estimator is robust enough, that it allows the adversary to arbitrarily remove -fraction of data points. We use to denote the samples at the -th coordinate of . We can also define in the same way.
Lemma A.2**.**
Suppose we observe -corrupted samples from 2.1. For each dimension , we assume the samples in are i.i.d. -sub-exponential with mean . After the contamination, we have the -th samples as . Then, we can guarantee the trimmed mean estimator on -th dimension that
[TABLE]
with probability at least .
We leave the proof of A.2 at the end of this section. Then, we present analysis of trimmed gradient estimator for sparse linear regression and sparse logistic regression by using A.2. For sparse linear regression model with sub-Gaussian covariates, the distribution of authentic gradients are sub-exponential instead of sub-Gaussian. More specifically, we first prove that when the current parameter iterate is , the sub-exponential parameter of all authentic gradient is , where .
To gain some intuition for this, we can consider the sparse linear equation problem, where . When , we exactly recover , and all stochastic gradients of authentic samples are actually zero vectors, as all observations are noiseless. It is clear that we will have sub-exponential parameter as [math].
Proof of A.1.
For any , the gradient for one sample can be written as
[TABLE]
where we omit the subscript in the proof. For any fixed standard basis vector , and define , we have
[TABLE]
To characterize the tail bounds of , we study the moment generating function:
[TABLE]
We denote as a Rademacher random variable, which is independent of and . Then we can use a standard symmetrization technique [Wai19],
[TABLE]
where follows from the exponential function’s power series expansion, and follows from the independence of , together with the fact that all odd moments of the terms have zeros means.
By the Cauchy-Schwarz inequality, we have
[TABLE]
It is clear that is a sub-Gaussian random variable with parameter . Since , we have . For any fixed standard basis vector , we can conclude that is sub-Gaussian with parameter at most based on 4.1. By basic properties of sub-Gaussian random variables [Wai19], we have
[TABLE]
where follows from the fact that is the weighted summation of two independent sub-Gaussian random variables. Hence, we have
[TABLE]
where follows from (proof by mathematical induction). When we have , eq. 5 converges to. Hence,
[TABLE]
That being said, is a sub-exponential random variable. By choosing as each coordinate in , each coordinate of gradient has sub-exponential parameter as .
Then, applying A.2 on this collection of corrupted sub-exponential random variables, we have
[TABLE]
with probability at least .
Applying union bounds on eq. 6 for all indexes, we have
[TABLE]
with probability at least .
In this subsection, we use A.2 to bound for sparse logistic regression. The technique for sparse logistic regression is similar to linear regression. Since we can directly show the sub-Gaussian distribution of gradient in this case, applying A.2 leads to the bound for .
Under the statistical model of sparse logistic regression, the gradient can be computed as:
[TABLE]
where we omit the subscript in the proof.
Since , and , then for any fixed standard basis vector , is sub-Gaussian with parameter at most based on 4.2. Notice that -sub-Gaussian random variables are still -sub-exponential. Applying A.2 again, we have
[TABLE]
with probability at least .
Applying union bounds on eq. 7 for all indexes, we have
[TABLE]
with probability at least . ∎
A.3 Trimmed mean estimator for strong contamination model
Now, it only remains to prove A.2. The proof technique is as follow: even though an adversary may delete samples from , we can still show the concentration inequalities for remaining authentic samples (denoting as in the proof). Then, we show that by using trimmed mean estimator, either the abnormal outliers will be removed, or their effect is controlled.
Proof of A.2.
Without loss of generality, we assume throughout the proof.
For each dimension , we can split the -th one-dimensional samples as . To study the performance of , we first show a concentration inequality of the sub-exponential variables in , without worrying about removing points from . This part of our proof is similar to Lemma 4.5 in [DKK*+*16].
Concentration inequality for
We consider the set in . Since is a subset of , by triangle inequality we have,
[TABLE]
The first term is simply the average of i.i.d. sub-exponential random variables. By A.1, we have
[TABLE]
For the second term , We now wish to show that with probability , there does not exist a subset so that the is more than . This event is equivalent to
[TABLE]
Let . For one subset , by A.1, we have
[TABLE]
Then, we take union bounds over all possible , which have events. Hence, the tail probability of can be bounded as
[TABLE]
where (i) follows from the fact that for large enough, and is the binary entropy function. Choosing , and hence , we have .
Combining the analysis on and (eq. 9 and eq. 10), we have
[TABLE]
This completes the concentration bounds on for all possible samples in without worrying about sample removing.
Trimmed mean estimator for
Then, we can consider the contribution of each part in . We denote the remaining set after trimming as , and the trimmed set as . Recall that we assume , we only need to bound , which is the empirical average of all samples in the remaining set .
As can be easily separated by the union of two distinct set and , we have the following inequalities,
[TABLE]
For any , by A.1, we have
[TABLE]
Applying a union bound for all samples, we can control the maximum magnitude for any ,
[TABLE]
We can bound by applying eq. 11. For the trimmed good samples , we have . Since we choose , we have .
Putting together the pieces, and choosing for some universal constant , we have
[TABLE]
with probability at least . This completes the proof for A.2. ∎
Appendix B Statistical estimation
via Robust Hard Thresholding
Here, we provide the Meta-Theorem 3.1 for statistical estimation performance of Algorithm 1 under statistical models.
We first introduce a supporting Lemma on the property of hard thresholding operator.
Lemma B.1** (Lemma 1 in [LB18]).**
We set in hard thresholding operator as , where , then we have
[TABLE]
Note that will be specified as later in the proof, as we choose as in 2.3.
Proof of 3.1.
We first study the objective function gap . Since the population risk satisfies -strong convexity and -smoothness (2.3), we have
[TABLE]
where (i) follows from the fact that , and -strong convexity holds.
Combining these two inequalities, we obtain
[TABLE]
Expanding the last term, we also have
[TABLE]
For the term , recall that is obtained from hard thresholding, and , we apply B.1 with :
[TABLE]
The term can be bounded by using eq. 1 in 3.1. We have
[TABLE]
with probability at least .
We denote and . Since, , putting together the pieces, we have
[TABLE]
with probability at least . Applying convexity, , as is the population minimizer. Hence, we have
[TABLE]
with probability at least .
Notice that eq. 15 is a quadratic inequality for , and we can use the root of eq. 15 to upper bound :
[TABLE]
where (i) follows from the basic inequality for non-negative .
We choose , and this leads to . Under the condition , we have . Then,
[TABLE]
Since is projection onto a convex set, by the property of Euclidean projection [Bub15], we have
[TABLE]
Together with eq. 17, eq. 16 establishes global linear convergence of .
We apply a union bound on iterates. Since for sufficiently large , we have
[TABLE]
with probability at least . Hence, we can achieve the final error
[TABLE]
by setting .
∎
B.1 Sparse linear regression
Proof of 4.1 and 4.3.
With A.2, A.1 in hand, we prove the arbitrary corruption case 4.1, and the proof of heavy tailed distribution 4.3 is similar. We evaluate the RDC 3.1 in Algorithm 1 for trimmed gradient estimator. With probability at least , we have
[TABLE]
where (i) follows from Holder inequality, (ii) follows from the sparsity of in Algorithm 1, (iii) follows from plugging in A.1, which yields .
We apply a union bound on iterates, and for sufficiently large . The condition in 3.1 can be achieved if
[TABLE]
Since , and , these conditions can be expressed as
[TABLE]
The final error can be expressed as
[TABLE]
∎
B.2 Sparse logistic regression
Proof of 4.2 and 4.4.
We prove 4.2, and the proof of 4.4 is similar. With probability at least , we have
[TABLE]
where (i) follows from the proof of 4.1 by using in A.1, and .
Similar to the proof in sparse linear regression, this final error can be expressed as
[TABLE]
∎
Appendix C Sparsity recovery
and sparse precision matrix estimation
C.1 Sparsity recovery guarantee
The same as the main text, we use to denote top indexes of with the largest magnitude. Let denote the smallest absolute value of nonzero elements of .
Proof of 4.5.
The sparsity recovery guarantee is similar to [YLZ18]. Since is sparse () by the definition of hard thresholding operator, we use to denote . We use the technique proof by contradiction. If , we at least have error as . Hence, , where (i) follows from the triangle inequality and definition of hard thresholding , and (ii) follows from the statistical guarantee in 4.1.
This contradicts with the -condition in 4.5, and hence we have the result in 4.5. ∎
C.2 Model selection for Gaussian graphical models
We then start to consider the sparsity recovery results for sparse precision matrix estimation – this is the part of 4.6. We first use following notations for a Gaussian graphical model.
We use to denote the -th samples of Gaussian graphical model, and to denote the -th random variable. Let be the index set We use to denote the sub-matrix of covariance matrix with both -th row and -th column removed, and use to denote ’s -th column with the diagonal entry removed. Also, we use to denote ’s -th column with the diagonal entry removed. and to denote the -th diagonal element of .
By basic probability computation, for each , the variable conditioning follows from a Gaussian distribution . Then we have the linear regression formulation , where and . Notice the definition of precision matrix , we have , and . Thus for the -th variable, and have the same sparsity pattern. Hence, the sparsity pattern of can be estimated through via solving the optimization eq. 2 (Neighborhood Selection in [MB06]).
In Algorithm 2, we robustify Neighborhood Selection by using Robust Hard Thresholding (with loss and trimmed gradient estimator) to robustify eq. 2. In line 6, we use Robust Hard Thresholding to regress each variable against its neighbors. In line 9, the sparsity pattern of can be estimated by aggregating the neighborhood support set of via intersection or union. Similar to 4.5, a -condition guarantees consistent edge selection.
Proof of 4.6.
Algorithm 2 iteratively uses Algorithm 1 as a Neighborhood Selection approach for each variable. Hence, we can apply 4.5 for each variable, and the sparsity patterns are the same according to . The stochastic noise term in sparse linear regression can be expressed as . Hence, under the same condition as 4.1, for each , we require a -condition for , .
Using a union bound, we conclude that Algorithm 2 is consistent in edge selection, with probability at least . ∎
Appendix D Full experiments details
We study empirical performance of Robust Hard Thresholding (Algorithm 1 and Algorithm 2). And we present the complete details of experimental setup in Section 5.
D.1 Synthetic data – sparse linear models
We first consider the performance of Algorithm 1 under (generalized) linear models with -corrupted samples.
**Sparse linear regression. ** In the first experiemtn, we consider an exact sparse linear regression model (4.1). In this model, the stochastic noise , and we vary the noise level in different simulations. We first generate authentic explanatory variables with parameters , from a Gaussian distribution , where the covariance matrix is a Toeplitz matrix with an exponential decay . This design matrix is known to enjoy the RSC-condition [RWY10], which meets the requirement of 4.1. The entries of the -sparse true parameter are set to either or . Fixing the contamination level at , we set the covariates of the outliers as , where is a random matrix of dimension , and the responses of outliers to .
To show the performance of Algorithm 1 under different noise levels determined by , we track the parameter error in each iteration. In the left plot of Figure 3, Algorithm 1 shows linear convergence, and the error curves flatten out at the level of the final error, which is consistent with our theory. Furthermore, Algorithm 1 can achieve machine precision when , which means exact recovery of .
**Misspecified model. ** For the second experiment, we use a sparse linear regression with model misspecification – the underlying authentic samples do not follow a linear model. We use the same Toeplitz covariates and true parameter , but and corresponding ’s are calculated as . Although this is a non-linear function, sparse linear regression on these authentic samples can still recover the support, as the cubic function is monotone and is sparse. We generate outliers using the same distribution as the first experiment, but with a different fraction of corruptions .
For simplicity, we track the function evaluated on all authentic samples . In the right plot of Figure 3, we show the performance of Algorithm 1 under different , and the oracle curve means using IHT only on authentic samples. The right plot has similar convergence under different values of corrupted fraction , and shows the robustness of Algorithm 1 without assuming an underlying linear model.
D.2 Robust -estimators via Robust Hard Thresholding
Classical robust -estimators [Loh17] (such as empirical risk minimization using Huber loss) are widely used in robust statistics in the case where the error distribution is heavy tailed or when there are arbitrary outliers only in the response variables. In the high dimensional setting, given -corrupted samples 2.1, we can use
[TABLE]
where can be chosen as Huber loss with parameter :
[TABLE]
[Loh17] studied robust -estimators in high dimensions, and proposed a composite optimization using instead of . They established local convergence guarantee for this composite optimization procedure, using a local RSC condition in a neighborhood around . Yet their results do not trivially extend to settings with arbitrarily corrupted covariates.
In our experiments, we use Huber loss in Robust Hard Thresholding to deal with heavy-tailed error distribution. In addition to heavy-tailed noise, -fraction of are still arbitrarily corrupted.
For the experiments, we use the same Toeplitz covariates and true parameter as in previous experiments on sparse linear models with fixed dimension parameters . The error distribution is a Cauchy distribution, which is a special case model misspecification, as it doesn’t meet the sub-Gaussian requirement in 4.1. For different contamination levels, we set the covariates of the outliers as , where is a random matrix of dimension , and the responses of outliers to .
Empirically, we observe linear convergence, and this is shown in Figure 4. This linear convergence results validates the local RSC condition proposed in [Loh17], and we can still achieve this even with -fraction of corrupted covariates.
D.3 Sparse logistic regression
For binary classification problem, we generate samples from a sparse LDA problem, where the distributions of the explanatory variables conditioned on the response variables follow multivariate Gaussian distributions with the same covariance matrix but different means.
We generate authentic samples from a Gaussian distribution if , and another distribution if . The parameters are fixed . We set , where is -sparse and its entries are set to be either or . And we set . The Bayes classifier is . This is a special case of 4.2, and it is known that sparse logistic regression attains fast classification error rates [LPR15]. We then set the covariates of the outliers as , where is a matrix of dimension , where the entries are random . The responses of outliers follow the distribution , which is exactly the opposite of 4.2.
We run Algorithm 1 with logistic loss under different levels of outlier fraction . In the left plot of Figure 5, we observe similar linear convergence as sparse linear regression This is consistent with 4.2 for sparse logistic regression, and it is clear that we cannot exactly recover unless the number of samples is infinite.
We then compare Algorithm 1 with the Trimmed Lasso estimator for sparse logistic regression [YLA18]. Although they also use a trimming technique, their algorithm is totally different from Algorithm 1, as we use coordinate-wise trimmed mean estimator for gradients in hard thresholding, but they trim samples in each iteration according to the each sample’s loss. Under the same sparse LDA model, we set . In simulation, we increase , and plot classification error (averaged over 50 trials on authentic test set) for different . The right plot of Figure 5 shows that Robust Hard Thresholding is better than Trimmed Lasso.
D.4 Synthetic data – Gaussian graphical model
We generate Gaussian graphical model samples by huge [ZLR*+*12]. We choose the “cluster” sparsity pattern, where the clustering parameters are default values in the package where the number of clusters in the graph is , the probability that a pair of nodes within a cluster are connected is 0.3, and there are no edges between nodes within different clusters. The off-diagonal elements of the precision matrix is denoted as , which is an experiment parameter for SNR.
We then add an additional fraction of samples sampled from another distribution. Following the experimental design in [YL15, WG17], each outlier is generated by a mixture of -dimensional Gaussian distributions , where We compare Algorithm 2 with other existing methods: Trimmed GLasso [YLA18], RCLIME [WG17], Skeptic [LHY*+*12b], and Spearman [LT18]. The latter two are based on robustifying the covariance matrix, and then using standard graphical model selection algorithms such as GLasso or CLIME. To directly compare these methods, we use CLIME for both of them.
To evaluate model selection performance, we use receiver operating characteristic (ROC) curves to compare our method to others over the full regularization paths. We generate regularization paths for other robust algorithms by tuning the in CLIME and GLasso. For Algorithm 2, we explicitly tune different sparsity level to generate the regualization path.
We set , and vary , and the SNR parameter for off-diagonal elements. We use different . For different off-diagonal values, we set (Low SNR), and (High SNR). We show ROC curves to demonstrate model selection performance in Figure 6. For the entire regularization path, our algorithm (denoted as Robust NS) has a better ROC compared to other algorithms.
In particular, Robust NS outperforms other methods with higher true positive rate when the false positive rate is small. This is the case where we use smaller hard thresholding sparsity in Algorithm 2, and larger regularization parameter for other methods based on GLasso and CLIME. This validates our theory in 4.6, which guarantees sparsity recovery when hard thresholding hyper-parameter is suitably chosen to match ’s sparsity .
D.5 Real data experiments
Here, we present details of the experiment using US equities data [ZLR*+*12]. We preprocess it by taking log-transformation and calculate the corresponding daily returns. Obvious outliers are removed by winsorizing each variable so that all samples are within five times the winsorized standard deviation from the winsorized mean. After preprocessing, we present example histograms and QQ plots from the Information Technology sector. In Figure 7 , we list the histograms of two typical companies in this sector. As we can see from Figure 8, even after preprocessing on these stock prices, they are still highly non-normal and heavy tailed. We do not add any manual outliers as financial data is already heavy tailed and have many outliers [dP18]. We also compare Algorithm 2 with the baseline NS approach (without consideration for corruptions or outliers).
We limit the number of edges to 2,000 for both methods. The cluster colored by purple denotes the Information Technology sector. In Figure 9, we can easily separate different clusters by using Robust NS. However, the Vanilla NS approach cannot distinguish the sector Information Technology (purple). Furthermore, we can observe that stocks from Information Technology (colored by purple) are much better clustered by Algorithm 2.
The reference list from the paper itself. Each links out to its DOI / PubMed record.
- 1[ACG 13] Andreas Alfons, Christophe Croux, and Sarah Gelper. Sparse least trimmed squares regression for analyzing high-dimensional large data sets. The Annals of Applied Statistics , pages 226–248, 2013.
- 2[AMS 99] Noga Alon, Yossi Matias, and Mario Szegedy. The space complexity of approximating the frequency moments. Journal of Computer and system sciences , 58(1):137–147, 1999.
- 3[ANW 12] Alekh Agarwal, Sahand Negahban, and Martin J. Wainwright. Fast global convergence of gradient methods for high-dimensional statistical recovery. Ann. Statist. , 40(5):2452–2482, 10 2012.
- 4[BD 09] Thomas Blumensath and Mike E Davies. Iterative hard thresholding for compressed sensing. Applied and computational harmonic analysis , 27(3):265–274, 2009.
- 5[BDLS 17] Sivaraman Balakrishnan, Simon S. Du, Jerry Li, and Aarti Singh. Computationally efficient robust sparse estimation in high dimensions. In Proceedings of the 2017 Conference on Learning Theory , 2017.
- 6[BDMS 13] Afonso S. Bandeira, Edgar Dobriban, Dustin G. Mixon, and William F. Sawin. Certifying the restricted isometry property is hard. IEEE Transactions on Information Theory , 59(6):3448–3450, 2013.
- 7[BJK 15] Kush Bhatia, Prateek Jain, and Purushottam Kar. Robust regression via hard thresholding. In Advances in Neural Information Processing Systems , pages 721–729, 2015.
- 8[BJK 17] Kush Bhatia, Prateek Jain, and Purushottam Kar. Consistent robust regression. In Advances in Neural Information Processing Systems , pages 2107–2116, 2017.
