Talking-Heads Attention
arXiv (Cornell University)2020
Citations Over Time
Abstract
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.
Related Papers
- → Improved softmax loss for deep learning‐based face and expression recognition(2019)20 cited
- → Exploring Alternatives to Softmax Function(2021)1 cited
- → Sparse-softmax: A Simpler and Faster Alternative Softmax Transformation(2021)6 cited
- → Effectiveness of MPC-friendly Softmax Replacement(2020)3 cited