A Latent Variable Model Approach to PMI-based Word Embeddings (original) (raw)
Abstract:Semantic word embeddings represent the meaning of a word via a vector, and are created by diverse methods. Many use nonlinear operations on co-occurrence statistics, and have hand-tuned hyperparameters and reweighting methods.
This paper proposes a new generative model, a dynamic version of the log-linear topic model of~\citet{mnih2007three}. The methodological novelty is to use the prior to compute closed form expressions for word statistics. This provides a theoretical justification for nonlinear models like PMI, word2vec, and GloVe, as well as some hyperparameter choices. It also helps explain why low-dimensional semantic embeddings contain linear algebraic structure that allows solution of word analogies, as shown by~\citet{mikolov2013efficient} and many subsequent papers.
Experimental support is provided for the generative model assumptions, the most important of which is that latent word vectors are fairly uniformly dispersed in space.
Submission history
From: Yingyu Liang [view email]
[v1] Thu, 12 Feb 2015 02:50:08 UTC (1,281 KB)
[v2] Tue, 3 Mar 2015 00:49:42 UTC (3,735 KB)
[v3] Tue, 21 Apr 2015 21:14:43 UTC (22,970 KB)
[v4] Wed, 22 Jul 2015 23:51:24 UTC (810 KB)
[v5] Wed, 14 Oct 2015 04:27:00 UTC (1,519 KB)
[v6] Wed, 24 Feb 2016 01:28:03 UTC (1,655 KB)
[v7] Fri, 22 Jul 2016 20:09:25 UTC (5,699 KB)
[v8] Wed, 19 Jun 2019 21:54:20 UTC (5,699 KB)