English

Statistical Query Lower Bounds for Learning Truncated Gaussians

Data Structures and Algorithms 2024-03-05 v1 Machine Learning Statistics Theory Machine Learning Statistics Theory

Abstract

We study the problem of estimating the mean of an identity covariance Gaussian in the truncated setting, in the regime when the truncation set comes from a low-complexity family C\mathcal{C} of sets. Specifically, for a fixed but unknown truncation set SRdS \subseteq \mathbb{R}^d, we are given access to samples from the distribution N(μ,I)\mathcal{N}(\boldsymbol{ \mu}, \mathbf{ I}) truncated to the set SS. The goal is to estimate μ\boldsymbol\mu within accuracy ϵ>0\epsilon>0 in 2\ell_2-norm. Our main result is a Statistical Query (SQ) lower bound suggesting a super-polynomial information-computation gap for this task. In more detail, we show that the complexity of any SQ algorithm for this problem is dpoly(1/ϵ)d^{\mathrm{poly}(1/\epsilon)}, even when the class C\mathcal{C} is simple so that poly(d/ϵ)\mathrm{poly}(d/\epsilon) samples information-theoretically suffice. Concretely, our SQ lower bound applies when C\mathcal{C} is a union of a bounded number of rectangles whose VC dimension and Gaussian surface are small. As a corollary of our construction, it also follows that the complexity of the previously known algorithm for this task is qualitatively best possible.

Keywords

Cite

@article{arxiv.2403.02300,
  title  = {Statistical Query Lower Bounds for Learning Truncated Gaussians},
  author = {Ilias Diakonikolas and Daniel M. Kane and Thanasis Pittas and Nikos Zarifis},
  journal= {arXiv preprint arXiv:2403.02300},
  year   = {2024}
}
R2 v1 2026-06-28T15:08:46.465Z