arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2205.01546cs.AI
收藏

Learn To Remember:用于文档级机器翻译的具有循环内存的Transformer

Learn To Remember: Transformer with Recurrent Memory for Document-Level Machine Translation

Johns Hopkins University(约翰斯·霍普金斯大学);University of Illinois Urbana-Champaign(伊利诺伊大学厄巴纳-香槟分校)

Yukun Feng, Feng Li, Ziang Song, Boyuan Zheng, Philipp Koehn

摘要:

Transformer架构在机器翻译方面取得了显著的进展。然而,大多数研究只关注文本层面的翻译,而没有考虑文本内部的语境依赖,导致文本层面连贯性不足。最近的一些研究试图通过引入额外的上下文编码器或翻译多个句子甚至整个文档来缓解这个问题。这样的方法可能丢失目标侧的信息,或者随着文档变长而具有增加的计算复杂性。为了解决这些问题,我们引入了一个经常性的记忆单元的香草Transformer,它支持的信息交换之间的句子和以前的上下文。通过从句子中获取信息,并将聚合的知识传递回后续的句子状态,来循环更新存储单元。我们遵循两阶段训练策略,首先在句子级别训练模型,然后针对文档级别的翻译进行微调。我们在三个流行的文档级机器翻译数据集上进行了实验,我们的模型在文档级基线上平均提高了0.91 s-BLEU。我们还在TED和News上取得了最先进的结果,平均比以前的工作高出0.36 s-BLEU和1.49 d-BLEU。

英文摘要:

The Transformer architecture has led to significant gains in machine translation. However, most studies focus on only sentence-level translation without considering the context dependency within documents, leading to the inadequacy of document-level coherence. Some recent research tried to mitigate this issue by introducing an additional context encoder or translating with multiple sentences or even the entire document. Such methods may lose the information on the target side or have an increasing computational complexity as documents get longer. To address such problems, we introduce a recurrent memory unit to the vanilla Transformer, which supports the information exchange between the sentence and previous context. The memory unit is recurrently updated by acquiring information from sentences, and passing the aggregated knowledge back to subsequent sentence states. We follow a two-stage training strategy, in which the model is first trained at the sentence level and then finetuned for document-level translation. We conduct experiments on three popular datasets for document-level machine translation and our model has an average improvement of 0.91 s-BLEU over the sentence-level baseline. We also achieve state-of-the-art results on TED and News, outperforming the previous work by 0.36 s-BLEU and 1.49 d-BLEU on average.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑