Not all parameters are born equal: Attention is mostly what you need
Not on the map. Why?
Abstract (arXiv)
Transformers are widely used in state-of-the-art machine translation, but the key to their success is still unknown. To gain insight into this, we consider three groups of parameters: embeddings, attention, and feed forward neural network (FFN) layers. We examine the relative importance of each by performing an ablation study where we initialise them at random and freeze them, so that their weights do not change over the course of the training. Through this, we show that the attention and FFN are equally important and fulfil the same functionality in a model. We show that the decision about whether a component is frozen or allowed to train is at least as important for the final model performance as its number of parameters. At the same time, the number of parameters alone is not indicative of a component's importance. Finally, while the embedding layer is the least essential for machine translation tasks, it is the most important component for language modelling tasks.
Key facts. Cited by 6 papers in the map (data through 14 Oct 2026). First posted on arXiv 22 Oct 2020.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 2; 2022: 1; 2023: 0; 2024: 1; 2025: 0; 2026: 2.
6 papers.
- 2021Cross-Attention is All You Need: Adapting Pretrained Transformers for Machine Translationcited by 74
- 2022A Fast Transformer-based General-Purpose Lossless Compressorcited by 26
- 2021Do Multilingual Neural Machine Translation Models Contain Language Pair Specific Attention Heads?cited by 6
- 2024Analyzing the Attention Heads for Pronoun Disambiguation in Context-aware Machine Translation Modelscited by 4
- 2026Extracting Small Translation Specialists from LLMs by Aggressively Pruning Expertscited by 3
- 2026Scaling Attention Head Analysis via Gradient-Based Attribution in Context-Aware Machine Translationcited by 0
- 2026Scaling Attention Head Analysis via Gradient-Based Attribution in Context-Aware Machine Translation23 Sep 2026
- 2026Extracting Small Translation Specialists from LLMs by Aggressively Pruning Experts27 May 2026
- 2024Analyzing the Attention Heads for Pronoun Disambiguation in Context-aware Machine Translation Models15 Dec 2024
- 2022A Fast Transformer-based General-Purpose Lossless Compressor30 Mar 2022
- 2021Do Multilingual Neural Machine Translation Models Contain Language Pair Specific Attention Heads?31 May 2021
- 2021Cross-Attention is All You Need: Adapting Pretrained Transformers for Machine Translation18 Apr 2021