中文

自动调整 Transformer 网络规模:提升低资源机器翻译的速度、效率与性能

计算与语言 2019-10-16 v1 机器学习 机器学习

摘要

神经序列到序列模型,尤其是 Transformer,是机器翻译中的当前最优(state of the art)方法。然而这些神经网络对架构与超参数设置非常敏感。通过网格或随机搜索优化这些设置计算代价高昂,因为需要多次训练运行。在本文中,我们通过自动调整规模(auto-sizing)将架构搜索融入单次训练运行中,其利用正则化在训练过程中删除网络中的神经元。在极低资源语言对上,我们表明自动调整规模能在从模型中移除三分之一参数的同时,将 BLEU 分数提升多达 3.9 点。

关键词

引用

@article{arxiv.1910.06717,
  title  = {Auto-Sizing the Transformer Network: Improving Speed, Efficiency, and Performance for Low-Resource Machine Translation},
  author = {Kenton Murray and Jeffery Kinnison and Toan Q. Nguyen and Walter Scheirer and David Chiang},
  journal= {arXiv preprint arXiv:1910.06717},
  year   = {2019}
}

备注

The 3rd Workshop on Neural Generation and Translation (WNGT 2019)