Large Scale Language Modeling: Converging on 40GB of Text in Four Hours

  • 2018-08-03 21:44:29
  • Raul Puri, Robert Kirby, Nikolai Yakovenko, Bryan Catanzaro
  • 37

Abstract

Recent work has shown how to train Convolutional Neural Networks (CNNs)rapidly on large image datasets, then transfer the knowledge gained from thesemodels to a variety of tasks. Following [Radford 2017], in this work, wedemonstrate similar scalability and transfer for Recurrent Neural Networks(RNNs) for Natural Language tasks. By utilizing mixed precision arithmetic anda 32k batch size distributed across 128 NVIDIA Tesla V100 GPUs, we are able totrain a character-level 4096-dimension multiplicative LSTM (mLSTM) forunsupervised text reconstruction over 3 epochs of the 40 GB Amazon Reviewsdataset in four hours. This runtime compares favorably with previous worktaking one month to train the same size and configuration for one epoch overthe same dataset. Converging large batch RNN models can be challenging. Recentwork has suggested scaling the learning rate as a function of batch size, butwe find that simply scaling the learning rate as a function of batch size leadseither to significantly worse convergence or immediate divergence for thisproblem. We provide a learning rate schedule that allows our model to convergewith a 32k batch size. Since our model converges over the Amazon Reviewsdataset in hours, and our compute requirement of 128 Tesla V100 GPUs, whilesubstantial, is commercially available, this work opens up large scaleunsupervised NLP training to most commercial applications and deep learningresearchers. A model can be trained over most public or private text datasetsovernight.

 

Introduction (beta)

None

 

Conclusion (beta)

None