arXiv · 1706.06415
THUMT: An Open Source Toolkit for Neural Machine Translation
Abstract
This paper introduces THUMT, an open-source toolkit for neural machine translation (NMT) developed by the Natural Language Processing Group at Tsinghua University. THUMT implements the standard attention-based encoder-decoder framework on top of Theano and supports three training criteria: maximum likelihood estimation, minimum risk training, and semi-supervised training. It features a visualization tool for displaying the relevance between hidden states in neural networks and contextual words, which helps to analyze the internal workings of NMT. Experiments on Chinese-English datasets show that THUMT using minimum risk training significantly outperforms GroundHog, a state-of-the-art toolkit for NMT.
Explore related subjects
Keep this discovery
Jiacheng Zhang, Yanzhuo Ding, Shiqi Shen, Yong Cheng, Maosong Sun, Huanbo Luan, Yang Liu. 2017-06-20. THUMT: An Open Source Toolkit for Neural Machine Translation. https://arxiv.org/abs/1706.06415
Cite the original work for its findings. Save a collection to share your selection of sources.