← Search

Noam Shazeer

15 accepted papers

2021

Do Transformer Modifications Transfer Across Implementations and Applications?

EMNLP 2021main

The research community has proposed copious modifications to the Transformer architecture since it was introduced over three years ago, relatively few of which have seen widespread adoption. In this paper, we comprehensively evaluate many of these modifications in a shared experimental setting that…

2021

GShard: Scaling Giant Models with Conditional Computation and Automatic Sharding

ICLR 2021poster

Neural network scaling has been critical for improving the model quality in many real-world machine learning applications with vast amounts of training data and compute. Although this trend of scaling is affirmed to be a sure-fire approach for better model quality, there are challenges on the path s…

Cited by 1282SourcePDFScholar
2021

Searching for Efficient Transformers for Language Modeling

NeurIPS 2021poster

Large Transformer models have been central to recent advances in natural language processing. The training and inference costs of these models, however, have grown rapidly and become prohibitively expensive. Here we aim to reduce the costs of Transformers by searching for a more efficient variant. C…

Cited by 179SourcePDFScholar
2019

Music Transformer: Generating Music with Long-Term Structure

ICLR 2019poster

Music relies heavily on repetition to build structure and meaning. Self-reference occurs on multiple timescales, from motifs to phrases to reusing of entire sections of music, such as in pieces with ABA structure. The Transformer (Vaswani et al., 2017), a sequence model based on self-attention, ha…

Cited by 0SourcePDFScholar
2018

Fast Decoding in Sequence Models Using Discrete Latent Variables

ICML 2018oral

Autoregressive sequence models based on deep neural networks, such as RNNs, Wavenet and Transformer are the state-of-the-art on many tasks. However, they lack parallelism and are thus slow for long sequences. RNNs lack parallelism both during training and decoding, while architectures like WaveNet a…

Cited by 149SourcePDFScholar
2018

Generating Wikipedia by Summarizing Long Sequences

ICLR 2018poster

We show that generating English Wikipedia articles can be approached as a multi- document summarization of source documents. We use extractive summarization to coarsely identify salient information and a neural abstractive model to generate the article. For the abstractive model, we introduce a deco…

2018

HydraNets: Specialized Dynamic Architectures for Efficient Inference

CVPR 2018poster

There is growing interest in improving the design of deep network architectures to be both accurate and low cost. This paper explores semantic specialization as a mechanism for improving the computational efficiency (accuracy-per-unit-cost) of inference in the context of image classification. Specif…

Cited by 180SourcePDFScholar
2018

Mesh-TensorFlow: Deep Learning for Supercomputers

NeurIPS 2018poster

Batch-splitting (data-parallelism) is the dominant distributed Deep Neural Network (DNN) training strategy, due to its universal applicability and its amenability to Single-Program-Multiple-Data (SPMD) programming. However, batch-splitting suffers from problems including the inability to train very…

2017

Attention is All you Need

NeurIPS 2017spotlight

The dominant sequence transduction models are based on complex recurrent orconvolutional neural networks in an encoder and decoder configuration. The best performing such models also connect the encoder and decoder through an attentionm echanisms. We propose a novel, simple network architecture bas…

2017

Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layer

ICLR 2017poster

The capacity of a neural network to absorb information is limited by its number of parameters. Conditional computation, where parts of the network are active on a per-example basis, has been proposed in theory as a way of dramatically increasing model capacity without a proportional increase in com…

Cited by 3097SourceScholar
2015

Scheduled Sampling for Sequence Prediction with Recurrent Neural Networks

NeurIPS 2015poster

Recurrent Neural Networks can be trained to produce sequences of tokens given some input, as exemplified by recent results in machine translation and image captioning. The current approach to training them consists of maximizing the likelihood of each token in the sequence given the current (recurre…

Cited by 2424SourcePDFScholar