arrow
Return

Parallelizing Word2Vec in Shared and Distributed Memory

delete2019-09-01
delete45
delete
OA
AI
S
Shihao Ji *
N
Nadathur Satish
李
李胜 (Sheng Li)
P
Pradeep Dubey
DOI:10.1109/TPDS.2019.2904058delete
deleteOriginal
deleteShare
deleteSave
View PDF
Abstract

Abstract

En 中文
Word2vec is a widely used algorithm for extracting low-dimensional vector representations of words. State-of-the-art algorithms including those by Mikolov et al. [1], [2] have been parallelized for multi-core CPU architectures, but are based on vector-vector operations with Hogwild updates that are memory-bandwidth intensive and do not efficiently use computational resources. In this paper, we propose HogBatch by improving reuse of various data structures in the algorithm through the use of minibatching and negative sample sharing, hence allowing us to express the problem using matrix multiply operations. We also explore different techniques to distribute word2vec computation across nodes in a computer cluster, and demonstrate good strong scalability up to 32 nodes. The new algorithm is particularly suitable for modern multi-core/many-core architectures, especially Intel's latest Knights Landing processors, and allows us to scale up the computation near linearly across cores and nodes, and process hundreds of millions of words per second, which is the fastest word2vec implementation to the best of our knowledge. We released the source code for reproducible research and general usage.
Keywords:
Word2Vec
parallel algorithms
distributed computing
multi-core and many-core systems
AI Summary

AI Summary

Key information extracted from the uploaded paper, including a brief overview, abstract, background, key highlights, visual analysis, and future outlook.

Journal

IEEE Transactions on Parallel and Distributed Systems cover
IEEE Transactions on Parallel and Distributed Systems
IF:
6
Papers:
5.2K
Citations:
1.1W

Organization

U
university system of georgia
Scholars:
7.3W
Papers: 6.6W
Citations: 101
G
Georgia State University
Scholars:
5.4K
Papers: 4.4K
Citations: 9.6K
Cited Papers

Cited Papers

One-step hydrothermal fabrication of SrMoO4/MoS2 composites with strong interfacial contacts for efficient photoreduction removal of Cr(vi)
err2020-01-01
err0
PREAI
errYiming Zhang; Xiaoyan Yang; Peng Zhang; Dan Liu; Yonglin Wang; Zhouzheng Jin; Bhekie B. Mamba; Alex T. Kuvarega; Jianzhou Gui
errShare
errSave
Induced Phase Transition in BiFeO3by High-Field Electron Spin Resonance
err2004-01-01
err0
PREAI
errD. VIEHLAND; J. F. LI; S. ZVYAGIN; A. P. PYATAKOV; A. BUSH; B. RUETTE; V. I. BELOTELOV; A. K. ZVEZDIN
errShare
errSave
A high temperature variety of BiOF
err1983-09-01
err0
PREAI
errSamir Matar; Jean-Maurice Reau; Louis Rabardel; Gérard Demazeau; Paul Hagenmuller
errShare
errSave
Placing search in context: The concept revisited
err2002-01-01
err706
PREAI
errFinkelstein, L; Gabrilovich, E; Matias, Y; Rivlin, E; Solan, Z; Wolfman, G; Ruppin, E
errShare
errSave
no more