arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2012.15671cs.CL

Vocabulary Learning via Optimal Transport for Neural Machine Translation

ByteDance AI Lab(字节跳动人工智能实验室);University of Wisconsin–Madison(威斯康星大学麦迪逊分校);Nanjing University(南京大学)

Jingjing Xu, Hao Zhou, Chun Gan, Zaixiang Zheng, Lei Li

英文摘要:

The choice of token vocabulary affects the performance of machine translation. This paper aims to figure out what is a good vocabulary and whether one can find the optimal vocabulary without trial training. To answer these questions, we first provide an alternative understanding of the role of vocabulary from the perspective of information theory. Motivated by this, we formulate the quest of vocabularization -- finding the best token dictionary with a proper size -- as an optimal transport (OT) problem. We propose VOLT, a simple and efficient solution without trial training. Empirical results show that VOLT outperforms widely-used vocabularies in diverse scenarios, including WMT-14 English-German and TED's 52 translation directions. For example, VOLT achieves almost 70% vocabulary size reduction and 0.5 BLEU gain on English-German translation. Also, compared to BPE-search, VOLT reduces the search time from 384 GPU hours to 30 GPU hours on English-German translation. Codes are available at https://github.com/Jingjing-NLP/VOLT .

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

↑