Program Highlights »
Sat Dec 9th 08:00 AM -- 06:30 PM @ 101 B
Deep Learning at Supercomputer Scale
Erich Elsen · Danijar Hafner · Zak Stone · Brennan Saeta

Workshop Home Page

Five years ago, it took more than a month to train a state-of-the-art image recognition model on the ImageNet dataset. Earlier this year, Facebook demonstrated that such a model could be trained in an hour. However, if we could parallelize this training problem across the world’s fastest supercomputers (~100 PFlops), it would be possible to train the same model in under a minute. This workshop is about closing that gap: how can we turn months into minutes and increase the productivity of machine learning researchers everywhere?

This one-day workshop will facilitate active debate and interaction across many different disciplines. The conversation will range from algorithms to infrastructure to silicon, with invited speakers from Cerebras, DeepMind, Facebook, Google, OpenAI, and other organizations. When should synchronous training be preferred over asynchronous training? Are large batch sizes the key to reach supercomputer scale, or is it possible to fully utilize a supercomputer at batch size one? How important is sparsity in enabling us to scale? Should sparsity patterns be structured or unstructured? To what extent do we expect to customize model architectures for particular problem domains, and to what extent can a “single model architecture” deliver state-of-the-art results across many different domains? How can new hardware architectures unlock even higher real-world training performance?

Our goal is bring people who are trying to answer any of these questions together in hopes that cross pollination will accelerate progress towards deep learning at true supercomputer scale.

08:10 AM Generalization Gap (Presentation)
Nitish Shirish Keskar
08:30 AM Closing the Generalization Gap (Presentation)
Itay Hubara
08:50 AM Don't Decay the Learning Rate, Increase the Batch Size (Presentation)
Sam Smith
09:10 AM ImageNet In 1 Hour (Presentation)
Priya Goyal
09:30 AM Training with TPUs (Presentation)
Chris Ying
09:50 AM Coffee Break (Break)
10:10 AM KFAC and Natural Gradients (Presentation)
Matthew Johnson, Daniel Duckworth
10:30 AM Neumann Optimizer (Presentation)
Shankar Krishnan
10:50 AM Evolutionary Strategies (Presentation)
Tim Salimans
11:15 AM Future Hardware Directions (Discussion Panel)
Gregory Diamos, Jeff Dean, Simon Knowles, Michael James, Scott Gray
01:30 PM Learning Device Placement (Presentation)
Azalia Mirhoseini
01:50 PM Scaling and Sparsity (Presentation)
Gregory Diamos
02:10 PM Small World Network Architectures (Presentation)
Scott Gray
02:30 PM Scalable RL and AlphaGo (Presentation)
Timothy Lillicrap
03:20 PM Scaling Deep Learning to 15 PetaFlops (Presentation)
Thorsten Kurth
03:40 PM Scalable Silicon Compute (Presentation)
Simon Knowles
04:00 PM Practical Scaling Techniques (Presentation)
Ujval Kapasi
04:20 PM Designing for Supercompute-Scale Deep Learning (Presentation)
Michael James
05:00 PM Adaptive Memory Networks (Poster Session)
Daniel Li
05:00 PM Supercomputers for Deep Learning (Poster Session)
Rangan Sukumar