Dimension of word embedding is usually determined with heuristic.


This paper suggests new metric to evaluate the quality of word embedding and uses it to find its optimal dimensionality. Word embedding algorithms are shown to converge to implicitly factorized PMI matrix. However, L2 loss of embedding and factorized matrix cannot be used since word embedding has unitary-invariance property which indicate word embedding is invariant to rotation. So this paper suggests Pairwise Inner Product(PIP) loss which measure relative position shifts between embeddings.

Using PIP loss, optimal dimensionality can be found that maximize the quality of embedding. This paper reported that choosing dimensionality of word embedding can be explained as bias-variance trade-off. Larger dimensionality leads to low bias but can increase estimation noise and smaller dimensionality does the opposite. Mathmathical proof was provided to show that the bias-variance trade-off captures the signal-to-noise ratio.\


Using given facts, this paper provides two discovery.


Suppose , larger leads to robustness to over-fitting.


Also, optimal dimensionality can be found that minimize the PIP loss by balancing bias-variance trade-off.

Yin, Zi, and Yuanyuan Shen. “On the dimensionality of word embedding.” Advances in Neural Information Processing Systems. 2018.