Statistical Query Lower Bounds for Learning Truncated Gaussians

Ilias Diakonikolas,Daniel M. Kane,Thanasis Pittas,Nikos Zarifis
2024-03-05
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 $\mathcal{C}$ of sets. Specifically, for a fixed but unknown truncation set $S \subseteq \mathbb{R}^d$, we are given access to samples from the distribution $\mathcal{N}(\boldsymbol{ \mu}, \mathbf{ I})$ truncated to the set $S$. The goal is to estimate $\boldsymbol\mu$ within accuracy $\epsilon>0$ in $\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 $d^{\mathrm{poly}(1/\epsilon)}$, even when the class $\mathcal{C}$ is simple so that $\mathrm{poly}(d/\epsilon)$ samples information-theoretically suffice. Concretely, our SQ lower bound applies when $\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.
Data Structures and Algorithms,Machine Learning,Statistics Theory
What problem does this paper attempt to address?
This paper discusses how to estimate the mean of a spherically symmetric Gaussian distribution with unit covariance in high-dimensional statistical estimation when the samples are truncated (or suppressed). The focus of the study is to establish a balance between information theory and computation under the condition of an unknown truncation set belonging to a low complexity class. The paper proves that even for a simple class of truncation sets, any Statistical Query (SQ) algorithm requires a super-polynomial number of samples, while information theory only requires a polynomial number of samples. This suggests the existence of an information-computation gap in learning truncated Gaussian distributions in certain cases.