atlas

Research map / Architectures and efficiency / Attention and transformers

Are Sixteen Heads Really Better than One?

NeurIPS 2019 · Architectures and efficiency › Attention and transformers · approach: Sparse and structured attention · 45 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's title only (no summary yet). Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Measured effects reported in the paper

What was changedWhereMeasuredEffect
remove individual attention heads after training (ablation)WMT large Transformer and BERT base-uncased · newstest2013; MNLI-matched validationBLEU; accuracyNo clear effect
reduce many individual layers to one attention head (ablation)WMT large Transformer and BERT base-uncased · newstest2013; MNLI-matched validationBLEU; accuracy: None of these results are statistically significant with p<0.01No clear effect
reduce layer 6 of encoder-decoder attention to one head (ablation)WMT large Transformer · newstest2013BLEU: -13.56Hurt
prune heads iteratively in increasing importance-score order (ablation)WMT large Transformer · newstest2013BLEU: 20%No clear effect
prune heads iteratively in increasing importance-score order (ablation)BERT base-uncased · MultiNLI-matched validationaccuracy: 40%No clear effect
use importance score I_h rather than individual oracle performance difference to order pruning (ablation)WMT large Transformer and BERT base-uncased · newstest2013; MultiNLI-matched validationBLEU; accuracyHelped
prune 50% of attention heads (ablation)BERT base-uncased · BERT inference, batch size 16inference speed: 114.0\pm 3.6 -> 134.0\pm 3.6; (+17.5\%)Helped
prune heads from encoder-decoder attention rather than self-attention (ablation)WMT large Transformer · WMT translationBLEU: more than 60%Hurt
prune heads from encoder or decoder self-attention (ablation)WMT large Transformer · WMT translationBLEU: 20%; around 30No clear effect
prune heads based on importance during early training rather than after heads have differentiated (ablation)IWSLT German-to-English model · newstest2013relative BLEU degradation: epochs 1 and 2; epoch 10 onwards; 40%Mixed

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy