Deep Learning with COTS HPC Systems

Authors: Adam Coates, Brody Huval, Tao Wang, David J. Wu, Bryan Catanzaro,Andrew Y. Ng (2013)
AUTHORED BY
Adam Coates
Brody Huval
Tao Wang
David J. Wu
Bryan Catanzaro
Andrew Y. Ng

Abstract

Scaling up deep learning algorithms has been shown to lead to increased performance in benchmark tasks and to enable discovery of complex high-level features. Recent efforts to train extremely large networks (with over 1 billion parameters) have relied on cloud- like computing infrastructure and thousands of CPU cores. In this paper, we present technical details and results from our own system based on Commodity Off-The-Shelf High Performance Computing (COTS HPC) technology: a cluster of GPU servers with Infini-band interconnects and MPI. Our system is able to train 1 billion parameter networks on just 3 machines in a couple of days, and we show that it can scale to networks with over 11 billion parameters using just 16 machines. As this infrastructure is much more easily marshaled by others, the approach enables much wider-spread research with extremely large neural networks.

Download PDF

Leave a Reply

You must be logged in to post a comment