Robust Regression Revisited: Acceleration and Improved Estimation Rates
Arun Jambulapati, Jerry Li, Tselil Schramm, Kevin Tian
Abstract
We study fast algorithms for statistical regression problems under the strong contamination model, where the goal is to approximately optimize a generalized linear model (GLM) given adversarially corrupted samples. Prior works in this line of research were based on the robust gradient descent framework of Prasad et. al., a first-order method using biased gradient queries, or the Sever framework of Diakonikolas et. al., an iterative outlier-removal method calling a stationary point finder. We present nearly-linear time algorithms for robust regression problems with improved runtime or estimation guarantees compared to the state-of-the-art. For the general case of smooth GLMs (e.g. logistic regression), we show that the robust gradient descent framework of Prasad et. al. can be accelerated, and show our algorithm extends to optimizing the Moreau envelopes of Lipschitz GLMs (e.g. support vector machines), answering several open questions in the literature. For the well-studied case of robust linear regression, we present an alternative approach obtaining improved estimation rates over prior nearly-linear time algorithms. Interestingly, our method starts with an identifiability proof introduced in the context of the sum-of-squares algorithm of Bakshi and Prasad, which achieved optimal error rates while requiring large polynomial runtime and sample complexity. We reinterpret their proof within the Sever framework and obtain a dramatically faster and more sample-efficient algorithm under fewer distributional assumptions.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 236f1d9b-575a-449a-96e1-24aa2ef4eb3dCited by top-tier papers10
- Robust Nonparametric Regression under Poisoning AttackPuning Zhao, Zhiguo WanAAAI 2024 · 13 citations
- A New Approach to Learning Linear Dynamical SystemsAinesh Bakshi, Allen Liu, Ankur Moitra, Morris YauSTOC 2023 · 10 citations
- Hardness and Algorithms for Robust and Sparse OptimizationEric Price, Sandeep Silwal, Samson ZhouICML 2022 · 10 citations
- Trimmed Maximum Likelihood Estimation for Robust Generalized Linear ModelPranjal Awasthi, Abhimanyu Das, Weihao Kong, Rajat SenNeurIPS 2022 · 9 citations
- Robust Bayesian Regression via Hard ThresholdingZheyi Fan, Zhaohui Li, Qingpei HuNeurIPS 2022 · 6 citations
Builds on5
- Acceleration with a Ball Optimization OracleYair Carmon, Arun Jambulapati, Qijia Jiang, Yujia Jin et al.NeurIPS 2020 · 58 citations
- Robust Sub-Gaussian Principal Component Analysis and Width-Independent Schatten PackingArun Jambulapati, Jerry Li, Kevin TianNeurIPS 2020 · 45 citations
- Projection Efficient Subgradient Method and Optimal Nonsmooth Frank-Wolfe MethodKiran Koshy Thekumparampil, Prateek Jain, Praneeth Netrapalli, Sewoong OhNeurIPS 2020 · 31 citations
- Robust linear regression: optimal rates in polynomial timeAinesh Bakshi, Adarsh PrasadSTOC 2021 · 13 citations
- Clustering mixture models in almost-linear time via list-decodable mean estimationIlias Diakonikolas, Daniel M. Kane, Daniel Kongsgaard, Jerry Li et al.STOC 2022 · 6 citations
Related papers
- Near-Optimal Algorithms for Gaussians with Huber Contamination: Mean Estimation and Linear RegressionIlias Diakonikolas, Daniel Kane, Ankit Pensia, Thanasis PittasNeurIPS 2023 · 9 citations
- Corruption-Tolerant Algorithms for Generalized Linear ModelsBhaskar Mukhoty, Debojyoti Dey, Purushottam KarAAAI 2023 · 1 citation
- Online Robust Regression via SGD on the l1 lossScott Pesme, Nicolas FlammarionNeurIPS 2020 · 41 citations
- Robust Generalized Method of Moments: A Finite Sample ViewpointDhruv Rohatgi, Vasilis SyrgkanisNeurIPS 2022 · 3 citations
- Outlier Robust Mean Estimation with Subgaussian Rates via StabilityIlias Diakonikolas, Daniel M. Kane, Ankit PensiaNeurIPS 2020 · 76 citations
