中文

通过高阶矩实现批量列表可解码线性回归

机器学习 2025-03-14 v1 数据结构与算法 统计理论 机器学习 统计理论

摘要

我们研究了使用批量进行列表可解码线性回归的任务。当一批由来自未知线性回归分布的独立同分布样本组成时,称该批为干净批。对于参数 α(0,1/2)\alpha \in (0, 1/2),未知比例为 α\alpha 的批是干净的,对其余的批不做任何假设。目标是输出一个向量列表,其中至少有一个向量在 2\ell_2 范数下接近真实回归向量。[DJKS23] 在自然的分布假设下给出了一个高效算法,其保证如下:假设批量大小 nn 满足 nΩ~(α1)n \geq \tilde{\Omega}(\alpha^{-1}) 且批次数为 m=poly(d,n,1/α)m = \mathrm{poly}(d, n, 1/\alpha),其算法在多项式时间内运行并输出一个大小为 O(1/α2)O(1/\alpha^2) 的列表,其中至少有一个向量与目标回归向量的距离为 O~(α1/2/n)\tilde{O}(\alpha^{-1/2}/\sqrt{n})。在此,我们设计了一个新的多项式时间算法,在假设协变量分布的低阶矩可被 Sum-of-Squares(SoS)认证有界的前提下,具有显著更强的保证。具体而言,对于任意常数 δ>0\delta>0,只要批量大小 nΩδ(αδ)n \geq \Omega_{\delta}(\alpha^{-\delta}) 且协变量的 Θ(1/δ)\Theta(1/\delta) 阶矩可被 SoS 认证有界,我们的算法使用 m=poly((dn)1/δ,1/α)m = \mathrm{poly}((dn)^{1/\delta}, 1/\alpha) 个批,在多项式时间内运行,并输出一个大小为 O(1/α)O(1/\alpha) 的列表,其中一个向量与目标的距离为 O(αδ/2/n)O(\alpha^{-\delta/2}/\sqrt{n})。即,我们的算法实现了显著更小的最小批量大小和最终误差,同时达到了最优列表大小。我们的方法通过仔细结合 SoS 范式与迭代方法和一种新颖的列表剪枝程序来利用高阶矩信息。在此过程中,我们给出了 Marcinkiewicz-Zygmund 不等式的 SoS 证明,该证明可能具有更广泛的适用性。

关键词

引用

@article{arxiv.2503.09802,
  title  = {Batch List-Decodable Linear Regression via Higher Moments},
  author = {Ilias Diakonikolas and Daniel M. Kane and Sushrut Karmalkar and Sihan Liu and Thanasis Pittas},
  journal= {arXiv preprint arXiv:2503.09802},
  year   = {2025}
}