Skip to content
AI Lehel Briefing
← Back to latest
Models & research Berkeley AI Research

What exactly does word2vec learn?

What exactly does word2vec learn, and how? Answering this question amounts to understanding representation learning in a minimal yet interesting language modeling task. Despite the fact that word2vec is a well-known precursor to modern language models, for many years, researchers lacked a quantitative and predictive theory describing its learning process. In our new paper, we finally provide such a theory. We prove that there are realistic, practical regimes in which the learning problem reduces to unweighted least-squares matrix factorization. We solve the gradient flow dynamics in closed f…

Excerpt supplied by the publisher’s feed

Read the original article at Berkeley AI Research Opens in a new tab