English

Pitfalls of Gaussians as a noise distribution in NCE

Machine Learning 2023-03-03 v2 Machine Learning

Abstract

Noise Contrastive Estimation (NCE) is a popular approach for learning probability density functions parameterized up to a constant of proportionality. The main idea is to design a classification problem for distinguishing training data from samples from an easy-to-sample noise distribution qq, in a manner that avoids having to calculate a partition function. It is well-known that the choice of qq can severely impact the computational and statistical efficiency of NCE. In practice, a common choice for qq is a Gaussian which matches the mean and covariance of the data. In this paper, we show that such a choice can result in an exponentially bad (in the ambient dimension) conditioning of the Hessian of the loss, even for very simple data distributions. As a consequence, both the statistical and algorithmic complexity for such a choice of qq will be problematic in practice, suggesting that more complex noise distributions are essential to the success of NCE.

Keywords

Cite

@article{arxiv.2210.00189,
  title  = {Pitfalls of Gaussians as a noise distribution in NCE},
  author = {Holden Lee and Chirag Pabbaraju and Anish Sevekari and Andrej Risteski},
  journal= {arXiv preprint arXiv:2210.00189},
  year   = {2023}
}

Comments

14 pages, 1 figure