Auto-Sizing the Transformer Network: Improving Speed, Efficiency, and Performance for Low-Resource Machine Translation
Computation and Language
2019-10-16 v1 Machine Learning
Machine Learning
Abstract
Neural sequence-to-sequence models, particularly the Transformer, are the state of the art in machine translation. Yet these neural networks are very sensitive to architecture and hyperparameter settings. Optimizing these settings by grid or random search is computationally expensive because it requires many training runs. In this paper, we incorporate architecture search into a single training run through auto-sizing, which uses regularization to delete neurons in a network over the course of training. On very low-resource language pairs, we show that auto-sizing can improve BLEU scores by up to 3.9 points while removing one-third of the parameters from the model.
Cite
@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}
}
Comments
The 3rd Workshop on Neural Generation and Translation (WNGT 2019)