← 返回资源分享
Talking-Heads Attention
论文
论文
发布时间2020-03-05
发表arXiv:2003.02436
作者:Noam Shazeer,Youlong Cheng,Zhenzhong Lan,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.
代码仓库 (5)
zygmuntz/hyperband官方
JunnYu/x-transformers-paddlePaddlePaddle
lucidrains/En-transformerPyTorch
tensorflow/models/tree/master/official/nlp/modeling/layersTensorFlow
lucidrains/x-transformersPyTorch
