Predicting and Using Target Length in Neural Machine Translation

Zijian Yang1, Yingbo Gao2, Weiyue Wang2, Hermann Ney2
1RWTH, 2RWTH Aachen University


Abstract

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.