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

Bag of Tricks for Optimizing Transformer Efficiency

Northeastern University(东北大学);The Chinese University of Hong Kong(香港中文大学);NiuTrans Research(小牛翻译研究院)

Ye Lin, Yanyang Li, Tong Xiao, Jingbo Zhu

英文摘要:

Improving Transformer efficiency has become increasingly attractive recently. A wide range of methods has been proposed, e.g., pruning, quantization, new architectures and etc. But these methods are either sophisticated in implementation or dependent on hardware. In this paper, we show that the efficiency of Transformer can be improved by combining some simple and hardware-agnostic methods, including tuning hyper-parameters, better design choices and training strategies. On the WMT news translation tasks, we improve the inference efficiency of a strong Transformer system by 3.80X on CPU and 2.52X on GPU. The code is publicly available at https://github.com/Lollipop321/mini-decoder-network.

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

↑