English

High-Dimensional Robust Mean Estimation via Gradient Descent

Machine Learning 2020-05-05 v1 Data Structures and Algorithms Optimization and Control Statistics Theory Machine Learning Statistics Theory

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.

Keywords

Cite

@article{arxiv.2005.01378,
  title  = {High-Dimensional Robust Mean Estimation via Gradient Descent},
  author = {Yu Cheng and Ilias Diakonikolas and Rong Ge and Mahdi Soltanolkotabi},
  journal= {arXiv preprint arXiv:2005.01378},
  year   = {2020}
}

Comments

Under submission to ICML'20

R2 v1 2026-06-23T15:17:13.978Z