High-dimensional Robust Mean Estimation via Gradient Descent
Yu Cheng, Ilias Diakonikolas, Rong Ge, Mahdi Soltanolkotabi
Abstract
We study the problem of high-dimensional robust mean estimation in the presence of a constant fraction of adversarial outliers. A recent line of work has provided sophisticated polynomial-time algorithms for this problem with dimension-independent error guarantees for a range of natural distribution families. In this work, we show that a natural non-convex formulation of the problem can be solved directly by gradient descent. Our approach leverages a novel structural lemma, roughly showing that any approximate stationary point of our non-convex objective gives a near-optimal solution to the underlying robust estimation task. Our work establishes an intriguing connection between algorithmic high-dimensional robust statistics and non-convex optimization, which may have broader applications to other robust estimation tasks.
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 897e77e3-c342-4e6b-a902-974d36910d23Cited by top-tier papers11
- Outlier Robust Mean Estimation with Subgaussian Rates via StabilityIlias Diakonikolas, Daniel M. Kane, Ankit PensiaNeurIPS 2020 · 76 citations
- Robust and Heavy-Tailed Mean Estimation Made Simple, via Regret MinimizationSamuel B. Hopkins, Jerry Li, Fred ZhangNeurIPS 2020 · 74 citations
- Robustly learning mixtures of k arbitrary GaussiansAinesh Bakshi, Ilias Diakonikolas, He Jia, Daniel M. Kane et al.STOC 2022 · 21 citations
- Outlier-Robust Sparse Estimation via Non-Convex OptimizationYu Cheng, Ilias Diakonikolas, Rong Ge, Shivam Gupta et al.NeurIPS 2022 · 19 citations
- Near-Optimal Algorithms for Gaussians with Huber Contamination: Mean Estimation and Linear RegressionIlias Diakonikolas, Daniel Kane, Ankit Pensia, Thanasis PittasNeurIPS 2023 · 9 citations
Related papers
- List Decodable Mean Estimation in Nearly Linear TimeYeshwanth Cherapanamjeri, Sidhanth Mohanty, Morris YauFOCS 2020 · 13 citations
- Outlier-Robust Sparse Mean Estimation for Heavy-Tailed DistributionsIlias Diakonikolas, Daniel Kane, Jasper C. H. Lee, Ankit PensiaNeurIPS 2022 · 15 citations
- Streaming Algorithms for High-Dimensional Robust StatisticsIlias Diakonikolas, Daniel M. Kane, Ankit Pensia, Thanasis PittasICML 2022 · 25 citations
- List-Decodable Sparse Mean EstimationShiwei Zeng, Jie ShenNeurIPS 2022 · 13 citations
- A Subquadratic Time Algorithm for Robust Sparse Mean EstimationAnkit PensiaICML 2024 · 1 citation
