中文

具有局部性约束的联合源-目标自注意力

计算与语言 2019-05-17 v1 机器学习

摘要

主流的神经机器翻译模型基于编码器-解码器结构,其中许多依赖于对源序列和目标序列的无约束感受野。本文研究一种打破上述两种惯例的新架构。我们简化的架构由 transformer 模型的解码器部分构成,基于自注意力,但对注意力感受野施加局部性约束。作为训练输入,源句和目标句均被送入网络,该网络作为语言模型进行训练。在推理时,目标词以源序列作为先前词自回归地预测。所提模型在 IWSLT'14 德英翻译上取得了 35.7 BLEU 的新 SOTA,并在 WMT'14 英德和 WMT'14 英法翻译基准上匹配了文献中报道的最佳结果。

关键词

引用

@article{arxiv.1905.06596,
  title  = {Joint Source-Target Self Attention with Locality Constraints},
  author = {José A. R. Fonollosa and Noe Casas and Marta R. Costa-jussà},
  journal= {arXiv preprint arXiv:1905.06596},
  year   = {2019}
}