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

ENGINE: Energy-Based Inference Networks for Non-Autoregressive Machine Translation

Toyota Technological Institute at Chicago(芝加哥丰田技术研究所);University of Chicago(芝加哥大学);New York University(纽约大学)

Lifu Tu, Richard Yuanzhe Pang, Sam Wiseman, Kevin Gimpel

英文摘要:

We propose to train a non-autoregressive machine translation model to minimize the energy defined by a pretrained autoregressive model. In particular, we view our non-autoregressive translation system as an inference network (Tu and Gimpel, 2018) trained to minimize the autoregressive teacher energy. This contrasts with the popular approach of training a non-autoregressive model on a distilled corpus consisting of the beam-searched outputs of such a teacher model. Our approach, which we call ENGINE (ENerGy-based Inference NEtworks), achieves state-of-the-art non-autoregressive results on the IWSLT 2014 DE-EN and WMT 2016 RO-EN datasets, approaching the performance of autoregressive models.

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

↑