Building an efficient neural language model over a billion words
Oct 25, 2016Armand Joulin, Edouard Grave, Justin Chiu

Summary
This post introduces adaptive softmax, a novel softmax approximation technique tailored for GPUs to efficiently train neural network based language models over very large vocabularies. It also details the open-sourcing of torch-rnnlib, a library that allows researchers to design new recurrent models and test prototypes on GPUs with minimal effort, including seamless access to fast baselines through torch.cudnn bindings. The post demonstrates achieving a perplexity of 45 on the 1-billion word dataset on a single GPU in a couple of days using these tools.
More from the AI Research and Development trail
- An Organizational Second Brain: Building an AI That Learns From Experts
- From User Sequences to Scaling Laws: A Multi-Stage Architecture for Meta’s Ads Ranking
- GEM Training: How Meta Doubled the Efficiency of Its LLM-Scale Ads Foundation Model
- Exploring Hierarchical Interest Representation For Meta Ads Deep Funnel Optimization
- Meta’s AI Storage Blueprint at Scale