Robust linear regression: optimal rates in polynomial time
Ainesh Bakshi, Adarsh Prasad
Abstract
We obtain robust and computationally efficient estimators for learning several linear models that achieve statistically optimal convergence rate under minimal distributional assumptions. Concretely, we assume our data is drawn from a k-hypercontractive distribution and an ǫ-fraction is adversarially corrupted. We then describe an estimator that converges to the optimal least-squares minimizer for the true distribution at a rate proportional to ǫ 2-2/k , when the noise is independent of the covariates. We note that no such estimator was known prior to our work, even with access to unbounded computation. The rate we achieve is informationtheoretically optimal and thus we resolve the main open question in Klivans, Kothari and Meka [COLT'18].
Our key insight is to identify an analytic condition that serves as a polynomial relaxation of independence of random variables. In particular, we show that when the moments of the noise and covariates are negatively-correlated, we obtain the same rate as independent noise. Further, when the condition is not satisfied, we obtain a rate proportional to ǫ 2-4/k , and again match the information-theoretic lower bound. Our central technical contribution is to algorithmically exploit independence of random variables in the "sum-of-squares" framework by formulating it as the aforementioned polynomial inequality.
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.
Cited by top-tier papers28
- Robustly learning mixtures of k arbitrary GaussiansAinesh Bakshi, Ilias Diakonikolas, He Jia, Daniel M. Kane et al.STOC 2022 · 21 citations
- Robust Regression Revisited: Acceleration and Improved Estimation RatesArun Jambulapati, Jerry Li, Tselil Schramm, Kevin TianNeurIPS 2021 · 18 citations
- List-Decodable Subspace Recovery: Dimension Independent Error in Polynomial TimeAinesh Bakshi, Pravesh K. KothariSODA 2021 · 17 citations
- Settling the robust learnability of mixtures of GaussiansAllen Liu, Ankur MoitraSTOC 2021 · 14 citations
- Learning Quantum Hamiltonians at Any Temperature in Polynomial TimeAinesh Bakshi, Allen Liu, Ankur Moitra, Ewin TangSTOC 2024 · 14 citations
Builds on4
- List Decodable Learning via Sum of SquaresPrasad Raghavendra, Morris YauSODA 2020 · 44 citations
- List-Decodable Subspace Recovery: Dimension Independent Error in Polynomial TimeAinesh Bakshi, Pravesh K. KothariSODA 2021 · 17 citations
- List Decodable Mean Estimation in Nearly Linear TimeYeshwanth Cherapanamjeri, Sidhanth Mohanty, Morris YauFOCS 2020 · 13 citations
- Algorithms for heavy-tailed statistics: regression, covariance estimation, and beyondYeshwanth Cherapanamjeri, Samuel B. Hopkins, Tarun Kathuria, Prasad Raghavendra et al.STOC 2020 · 2 citations
Related papers
- Sample-Optimal Private Regression in Polynomial TimePrashanti Anderson, Ainesh Bakshi, Mahbod Majid, Stefan TiegelSTOC 2025
- Outlier-Robust Clustering of Gaussians and Other Non-Spherical MixturesAinesh Bakshi, Ilias Diakonikolas, Samuel B. Hopkins, Daniel Kane et al.FOCS 2020 · 13 citations
- Robust Sparse Regression with Non-Isotropic DesignsChih-Hung Liu, Gleb NovikovNeurIPS 2024 · 2 citations
- List-decodable covariance estimationMisha Ivkov, Pravesh K. KothariSTOC 2022 · 5 citations
- Consistent Adversarially Robust Linear Classification: Non-Parametric SettingElvis DohmatobICML 2024 · 2 citations
