中文

RealFormer:偏好残差注意力的 Transformer

机器学习 2021-09-14 v3

摘要

Transformer 是现代 NLP 模型的主干网络。在本文中,我们提出 RealFormer,一种简单且通用的技术,用于创建残差注意力层 Transformer 网络,该网络在广泛任务上显著优于标准 Transformer 及其变体(BERT、ETC 等),包括掩码语言建模、GLUE、SQuAD、神经机器翻译、WikiHop、HotpotQA、Natural Questions 和 OpenKP。我们还通过实证观察到 RealFormer 稳定了训练并导致具有更稀疏注意力的模型。RealFormer 的源代码和预训练检查点可在 https://github.com/google-research/google-research/tree/master/realformer 找到。

关键词

引用

@article{arxiv.2012.11747,
  title  = {RealFormer: Transformer Likes Residual Attention},
  author = {Ruining He and Anirudh Ravula and Bhargav Kanagal and Joshua Ainslie},
  journal= {arXiv preprint arXiv:2012.11747},
  year   = {2021}
}

备注

Findings of ACL-IJCNLP 2021