arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2312.01547cs.DScs.LGstat.ML

Huber污染下高斯分布的近最优算法:均值估计与线性回归

Near-Optimal Algorithms for Gaussians with Huber Contamination: Mean Estimation and Linear Regression

University of Wisconsin-Madison(威斯康星大学麦迪逊分校);University of California, San Diego(加州大学圣迭戈分校);IBM Research(IBM研究院)

Ilias Diakonikolas, Daniel M. Kane, Ankit Pensia, Thanasis Pittas

AI总结:

针对Huber污染下的高斯均值估计和线性回归,本文首次给出样本复杂度为$\tilde{O}(d/ε^2)$、几乎线性时间且达到$O(ε)$最优$\ell_2$误差的算法,并提出了多方向滤波的几乎线性时间方法。

摘要:

我们研究在存在Huber污染的情况下,高斯均值估计和带有高斯协变量的线性回归这两个基本问题。我们的主要贡献是为这两个问题设计了首个样本近最优且几乎线性时间的算法,并具有最优误差保证。具体而言,对于在$\mathbb{R}^d$上、污染参数$ε\in (0, ε_0)$(其中$ε_0$为一个小的绝对常数)的高斯鲁棒均值估计,我们给出一个样本复杂度为$n = \tilde{O}(d/ε^2)$且运行时间几乎线性的算法,该算法在$\ell_2$-误差$O(ε)$内逼近目标均值。这改进了先前以多项式次优的样本和时间复杂度实现相同误差保证的工作。对于鲁棒线性回归,我们给出首个样本复杂度为$n = \tilde{O}(d/ε^2)$且运行时间几乎线性的算法,该算法在$\ell_2$-误差$O(ε)$内逼近目标回归量。这是首个实现最优误差保证的多项式样本和时间算法,回答了文献中的一个开放问题。在技术层面,我们开发了一种方法,可为多方向滤波生成几乎线性时间的算法,这可能具有更广泛的兴趣。

英文摘要:

We study the fundamental problems of Gaussian mean estimation and linear regression with Gaussian covariates in the presence of Huber contamination. Our main contribution is the design of the first sample near-optimal and almost linear-time algorithms with optimal error guarantees for both of these problems. Specifically, for Gaussian robust mean estimation on $\mathbb{R}^d$ with contamination parameter $ε\in (0, ε_0)$ for a small absolute constant $ε_0$, we give an algorithm with sample complexity $n = \tilde{O}(d/ε^2)$ and almost linear runtime that approximates the target mean within $\ell_2$-error $O(ε)$. This improves on prior work that achieved this error guarantee with polynomially suboptimal sample and time complexity. For robust linear regression, we give the first algorithm with sample complexity $n = \tilde{O}(d/ε^2)$ and almost linear runtime that approximates the target regressor within $\ell_2$-error $O(ε)$. This is the first polynomial sample and time algorithm achieving the optimal error guarantee, answering an open question in the literature. At the technical level, we develop a methodology that yields almost-linear time algorithms for multi-directional filtering that may be of broader interest.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑