Regularized context gates on transformer for machine translation

X Li, L Liu, R Wang, G Huang, M Meng - arXiv preprint arXiv:1908.11020, 2019 - arxiv.org
arXiv preprint arXiv:1908.11020, 2019arxiv.org
Context gates are effective to control the contributions from the source and target contexts in
the recurrent neural network (RNN) based neural machine translation (NMT). However, it is
challenging to extend them into the advanced Transformer architecture, which is more
complicated than RNN. This paper first provides a method to identify source and target
contexts and then introduce a gate mechanism to control the source and target contributions
in Transformer. In addition, to further reduce the bias problem in the gate mechanism, this …
Context gates are effective to control the contributions from the source and target contexts in the recurrent neural network (RNN) based neural machine translation (NMT). However, it is challenging to extend them into the advanced Transformer architecture, which is more complicated than RNN. This paper first provides a method to identify source and target contexts and then introduce a gate mechanism to control the source and target contributions in Transformer. In addition, to further reduce the bias problem in the gate mechanism, this paper proposes a regularization method to guide the learning of the gates with supervision automatically generated using pointwise mutual information. Extensive experiments on 4 translation datasets demonstrate that the proposed model obtains an averaged gain of 1.0 BLEU score over a strong Transformer baseline.
arxiv.org
Showing the best result for this search. See all results