Beam Search Strategies for Neural Machine Translation

Markus Freitag, Yaser Al-Onaizan


Abstract
The basic concept in Neural Machine Translation (NMT) is to train a large Neural Network that maximizes the translation performance on a given parallel corpus. NMT is then using a simple left-to-right beam-search decoder to generate new translations that approximately maximize the trained conditional probability. The current beam search strategy generates the target sentence word by word from left-to-right while keeping a fixed amount of active candidates at each time step. First, this simple search is less adaptive as it also expands candidates whose scores are much worse than the current best. Secondly, it does not expand hypotheses if they are not within the best scoring candidates, even if their scores are close to the best one. The latter one can be avoided by increasing the beam size until no performance improvement can be observed. While you can reach better performance, this has the drawback of a slower decoding speed. In this paper, we concentrate on speeding up the decoder by applying a more flexible beam search strategy whose candidate size may vary at each time step depending on the candidate scores. We speed up the original decoder by up to 43% for the two language pairs German to English and Chinese to English without losing any translation quality.
Anthology ID:
W17-3207
Volume:
Proceedings of the First Workshop on Neural Machine Translation
Month:
August
Year:
2017
Address:
Vancouver
Editors:
Thang Luong, Alexandra Birch, Graham Neubig, Andrew Finch
Venue:
NGT
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
56–60
Language:
URL:
https://aclanthology.org/W17-3207
DOI:
10.18653/v1/W17-3207
Bibkey:
Cite (ACL):
Markus Freitag and Yaser Al-Onaizan. 2017. Beam Search Strategies for Neural Machine Translation. In Proceedings of the First Workshop on Neural Machine Translation, pages 56–60, Vancouver. Association for Computational Linguistics.
Cite (Informal):
Beam Search Strategies for Neural Machine Translation (Freitag & Al-Onaizan, NGT 2017)
Copy Citation:
PDF:
https://aclanthology.org/W17-3207.pdf