Learning Contextualized Sentence Representations for Document-Level Neural Machine Translation

Pei Zhang,Xu Zhang,Wei Chen,Jian Yu,Yanfeng Wang,Deyi Xiong
DOI: https://doi.org/10.48550/arXiv.2003.13205
2020-03-30
Abstract:Document-level machine translation incorporates inter-sentential dependencies into the translation of a source sentence. In this paper, we propose a new framework to model cross-sentence dependencies by training neural machine translation (NMT) to predict both the target translation and surrounding sentences of a source sentence. By enforcing the NMT model to predict source context, we want the model to learn "contextualized" source sentence representations that capture document-level dependencies on the source side. We further propose two different methods to learn and integrate such contextualized sentence embeddings into NMT: a joint training method that jointly trains an NMT model with the source context prediction model and a pre-training & fine-tuning method that pretrains the source context prediction model on a large-scale monolingual document corpus and then fine-tunes it with the NMT model. Experiments on Chinese-English and English-German translation show that both methods can substantially improve the translation quality over a strong document-level Transformer baseline.
Computation and Language
What problem does this paper attempt to address?