arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2010.02416cs.CLcs.LG

Efficient Inference For Neural Machine Translation

Johns Hopkins University(约翰斯·霍普金斯大学);Apple Inc.(苹果公司)

Yi-Te Hsu, Sarthak Garg, Yi-Hsiu Liao, Ilya Chatsviorkin

英文摘要:

Large Transformer models have achieved state-of-the-art results in neural machine translation and have become standard in the field. In this work, we look for the optimal combination of known techniques to optimize inference speed without sacrificing translation quality. We conduct an empirical study that stacks various approaches and demonstrates that combination of replacing decoder self-attention with simplified recurrent units, adopting a deep encoder and a shallow decoder architecture and multi-head attention pruning can achieve up to 109% and 84% speedup on CPU and GPU respectively and reduce the number of parameters by 25% while maintaining the same translation quality in terms of BLEU.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑