Predicting and Using Target Length in Neural Machine Translation

作者: Zijian Yang, Yingbo Gao, Weiyue Wang, Hermann Ney

摘要

Attention-based encoder-decoder models have achieved great success in neural machine translation tasks. However, the lengths of the target sequences are not explicitly predicted in these models. This work proposes length prediction as an auxiliary task and set up a sub-network to obtain the length information from the encoder. Experimental results show that the length prediction sub-network brings improvements over the strong baseline system and that the predicted length can be used as an alternative to length normalization during decoding.

PDF打开原文