arXiv:2501.05425cs.DScs.LG2025-01被引 3

高维数据中存在噪声时,高效准确估计共同均值。

Entangled Mean Estimation in High-Dimensions

  • 通过迭代剔除异常点,逐步逼近真实均值。
  • 误差达到信息论最优水平,包含维度与样本量的合理依赖。
  • 适合处理含噪声的高维统计推断问题,尤其对鲁棒估计研究者有参考价值。

我们研究在子信号模型下的高维纠缠均值估计任务。给定 $N$ 个独立随机点 $x_1,\ldots,x_N \in \mathbb{R}^D$,每个点来自均值为 $μ$、协方差未知的高斯分布,其中未知比例 $α \in (0,1)$ 的点具有单位有界协方差。目标是估计公共均值 $μ$。一维情形在理论计算机科学与统计学中已有深入研究,近期工作 [LY20; CV24] 已获得近最优上下界。然而,多维情形的信息论理解仍不充分。本文设计了一种计算高效的算法,实现信息论近最优误差。具体而言,最优误差(忽略多项式对数因子)为 $f(α,N) + \sqrt{D/(αN)}$,其中 $f(α,N)$ 为一维问题的误差,第二项为次高斯误差率。算法采用迭代精化策略,通过一种新颖的拒绝采样过程剔除明显偏离当前估计 $\hat μ$ 的点,以过滤异常噪声样本。由于拒绝采样引入分布偏差,我们细致分析了该偏差,提出迭代降维策略,并采用受列表可解学习启发的新子程序,利用一维结果提升性能。

原文摘要 · Abstract (English)

We study the task of high-dimensional entangled mean estimation in the subset-of-signals model. Specifically, given $N$ independent random points $x_1,\ldots,x_N$ in $\mathbb{R}^D$ and a parameter $α\in (0, 1)$ such that each $x_i$ is drawn from a Gaussian with mean $μ$ and unknown covariance, and an unknown $α$-fraction of the points have identity-bounded covariances, the goal is to estimate the common mean $μ$. The one-dimensional version of this task has received significant attention in theoretical computer science and statistics over the past decades. Recent work [LY20; CV24] has given near-optimal upper and lower bounds for the one-dimensional setting. On the other hand, our understanding of even the information-theoretic aspects of the multivariate setting has remained limited. In this work, we design a computationally efficient algorithm achieving an information-theoretically near-optimal error. Specifically, we show that the optimal error (up to polylogarithmic factors) is $f(α,N) + \sqrt{D/(αN)}$, where the term $f(α,N)$ is the error of the one-dimensional problem and the second term is the sub-Gaussian error rate. Our algorithmic approach employs an iterative refinement strategy, whereby we progressively learn more accurate approximations $\hat μ$ to $μ$. This is achieved via a novel rejection sampling procedure that removes points significantly deviating from $\hat μ$, as an attempt to filter out unusually noisy samples. A complication that arises is that rejection sampling introduces bias in the distribution of the remaining points. To address this issue, we perform a careful analysis of the bias, develop an iterative dimension-reduction strategy, and employ a novel subroutine inspired by list-decodable learning that leverages the one-dimensional result.

均值估计高维统计鲁棒学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。