on our English dataset using a 5 layer, 3 recurrent layer architecture with 2560 hidden units for all
layers. In this table, we use a minibatch of 64 per GPU, expanding the algorithmic minibatch as we
scale to more GPUs. We see that our implementation is considerably faster than OpenMPI’s when
the communication is within a node (8 GPUs or less). As we increase the number of GPUs and
increase the amount of inter-node communication, the gap shrinks, although our implementation is
still 2-4X faster.
All of our training runs use either 8 or 16 GPUs, and in this regime, our all-reduce implementation
results in 2.5faster training for the full training run, compared to using OpenMPI directly. Optimizing all-reduce has thus resulted in important productivity beneﬁts for our experiments, and has
made our simple synchronous SGD approach scalable.
13
GPU OpenMPI Our Performance
all-reduce all-reduce Gain
4 55359.1 2587.4 21.4
8 48881.6 2470.9 19.8
16 21562.6 1393.7 15.5
32 8191.8 1339.6 6.1
64 1395.2 611.0 2.3
128 1602.1 422.6 3.8
We show that an end-to-end deep learning approach can be used to recognize
either English or Mandarin Chinese speech--two vastly different languages.
Because it replaces entire pipelines of hand-engineered components with neural
networks, end-to-end learning allows us to handle a diverse variety of speech
including noisy environments, accents and different languages. Key to our
approach is our application of HPC techniques, resulting in a 7x speedup over
our previous system. Because of this efficiency, experiments that previously
took weeks now run in days. This enables us to iterate more quickly to identify
superior architectures and algorithms. As a result, in several cases, our
system is competitive with the transcription of human workers when benchmarked
on standard datasets. Finally, using a technique called Batch Dispatch with
GPUs in the data center, we show that our system can be inexpensively deployed
in an online setting, delivering low latency when serving users at scale.
Dario Amodei Rishita Anubhai Eric Battenberg Carl Case Jared Casper Bryan Catanzaro Jingdong Chen Mike Chrzanowski Adam Coates Greg Diamos Erich Elsen Jesse Engel Linxi Fan Christopher Fougner Tony Han Awni Hannun Billy Jun Patrick LeGresley Libby Lin Sharan Narang Andrew Ng Sherjil Ozair Ryan Prenger Jonathan Raiman Sanjeev Satheesh David Seetapun Shubho Sengupta Yi Wang Zhiqian Wang Chong Wang Bo Xiao Dani Yogatama Jun Zhan Zhenyao Zhu