2003.02436

Total: 1

#1 Talking-Heads Attention [PDF9] [Copy] [Kimi8]

Authors: Noam Shazeer ; Zhenzhong Lan ; Youlong Cheng ; Nan Ding ; Le Hou

We introduce "talking-heads attention" - a variation on multi-head attention which includes linearprojections across the attention-heads dimension, immediately before and after the softmax operation.While inserting only a small number of additional parameters and a moderate amount of additionalcomputation, talking-heads attention leads to better perplexities on masked language modeling tasks, aswell as better quality when transfer-learning to language comprehension and question answering tasks.

Subjects: Machine Learning ; Neural and Evolutionary Computing ; Sound ; Audio and Speech Processing ; Machine Learning

Publish: 2020-03-05 05:17:17 UTC