Synthesizer: Rethinking Self-Attention in Transformer Models

2 May 2020  ·  Yi Tay, Dara Bahri, Donald Metzler, Da-Cheng Juan, Zhe Zhao, Che Zheng ·

The dot product self-attention is known to be central and indispensable to state-of-the-art Transformer models. But is it really required? This paper investigates the true importance and contribution of the dot product-based self-attention mechanism on the performance of Transformer models. Via extensive experiments, we find that (1) random alignment matrices surprisingly perform quite competitively and (2) learning attention weights from token-token (query-key) interactions is useful but not that important after all. To this end, we propose \textsc{Synthesizer}, a model that learns synthetic attention weights without token-token interactions. In our experiments, we first show that simple Synthesizers achieve highly competitive performance when compared against vanilla Transformer models across a range of tasks, including machine translation, language modeling, text generation and GLUE/SuperGLUE benchmarks. When composed with dot product attention, we find that Synthesizers consistently outperform Transformers. Moreover, we conduct additional comparisons of Synthesizers against Dynamic Convolutions, showing that simple Random Synthesizer is not only $60\%$ faster but also improves perplexity by a relative $3.5\%$. Finally, we show that simple factorized Synthesizers can outperform Linformers on encoding only tasks.

PDF Abstract

Results from the Paper


 Ranked #1 on Dialogue Generation on Persona-Chat (BLEU-1 metric, using extra training data)

     Get a GitHub badge
Task Dataset Model Metric Name Metric Value Global Rank Uses Extra
Training Data
Result Benchmark
Document Summarization CNN / Daily Mail Synthesizer (R+V) ROUGE-1 38.57 # 23
ROUGE-2 16.24 # 21
ROUGE-L 35.95 # 23
Linguistic Acceptability CoLA Dev Synthesizer (R+V) Accuracy 53.3 # 5
Semantic Textual Similarity MRPC Dev Synthesizer (R+V) Accuracy 91.2 # 1
Dialogue Generation Persona-Chat Synthesizer (R+V) BLEU-1 14.7 # 1
ROUGE-L 14.79 # 1
METEOR 6.39 # 1
CIDr 19.09 # 1
Machine Translation WMT2014 English-French Synthesizer (Random + Vanilla) BLEU score 41.85 # 20
Hardware Burden None # 1
Operations per network pass None # 1
Machine Translation WMT2014 English-German Synthesizer (Random + Vanilla) BLEU score 28.47 # 43
Hardware Burden None # 1
Operations per network pass None # 1

Methods