Get our free extension to see links to code for papers anywhere online!

Chrome logo Add to Chrome

Firefox logo Add to Firefox


Talking-Heads Attention

Mar 05, 2020
Noam Shazeer, Zhenzhong Lan, Youlong Cheng, Nan Ding, Le Hou


Share this with someone who'll enjoy it:


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.



   Access Paper Source



Share this with someone who'll enjoy it: