开源 | 哈佛大学NLP组开源神经机器翻译工具包OpenNMT:已达到生产可用水平
选自OpenNMT
机器之心编译
参与:吴攀
神经机器翻译是近段时间以来推动机器翻译发展的主要推动力。今天,哈佛大学自然语言处理研究组(Harvard NLP)宣布开源了其研发的神经机器翻译系统 OpenNMT,该系统使用了 Torch 数学工具包。该研究组在官网上表示该系统已经达到生产可用的水平(industrial-strength)。
-
官网: http://opennmt.net
-
代码: https://github.com/opennmt/opennmt
-
演示: https://demo-pnmt.systran.net
-
Docker 容器: https://hub.docker.com/r/harvardnlp/opennmt
OpenNMT 可以像主要的翻译服务提供商的已投入生产的系统那样使用。该系统简单易用,易于扩展,同时也能维持效率和当前最佳的翻译准确度。
其特性包括:
-
简单的通用型接口,仅需要源文件和目标文件;
-
为高性能 GPU 训练进行了速度和内存优化;
-
可以提升翻译性能的最新研究的特性;
-
有多个语言对的预训练好的模型(即将到来);
-
允许其它序列生成任务的扩展,比如归纳总结和图像到文本生成。
安装
OpenNMT 仅需要一次 vanilla torch/cutorch 安装。它要使用 nn、nngraph 和 cunn。有(CUDA)Docker 容器可选。
快速启动
OpenNMT 包含三条指令:
1)预处理数据
th preprocess.lua -train_src data/src-train.txt -train_tgt data/tgt-train.txt -valid_src data/src-val.txt -valid_tgt data/tgt-val.txt -save_data data/demo
2)训练模型
th train.lua -data data/demo-train.t7 -save_model model
3)翻译句子
th translate.lua -model model_final.t7 -src data/src-test.txt -output pred.txt
查看指南了解更多:http://opennmt.github.io/Guide
研究
其中主要的模型基于论文 Neural Machine Translation by Jointly Learning to Align and Translate Bahdanau et al. ICLR 2015 和 Effective Approaches to Attention-based Neural Machine Translation, Luong et al. EMNLP 2015。
在基本模型上,还有大量可选项,这都要感谢 SYSTRAN(http://www.systransoft.com/)的出色工作。特别地,下面是一些实现的功能:
-
Effective Approaches to Attention-based Neural Machine Translation . Luong et al., EMNLP 2015.
-
Character-based Neural Machine Translation. Costa-Jussa and Fonollosa, ACL 2016.
-
Compression of Neural Machine Translation Models via Pruning . See et al., COLING 2016.
-
Sequence-Level Knowledge Distillation . Kim and Rush., EMNLP 2016.
-
Deep Recurrent Models with Fast Forward Connections for Neural Machine Translation . Zhou et al, TACL 2016.
-
Guided Alignment Training for Topic-Aware Neural Machine Translation . Chen et al., arXiv:1607.01628.
-
Linguistic Input Features Improve Neural Machine Translation . Senrich et al., arXiv:1606.02892
声明
OpenNMT 的实现使用了以下项目的代码:
-
Andrej Karpathy 的 char-rnn: https://github.com/karpathy/char-rnn
-
Wojciech Zaremba 的 LSTM: https://github.com/wojzaremba/lstm
-
Element RNN 库: https://github.com/Element-Research/rnn
证书
MIT
©本文由机器之心编译, 转载请联系本公众号获得授权 。
?------------------------------------------------
加入机器之心(全职记者/实习生):hr@almosthuman.cn
投稿或寻求报道:editor@almosthuman.cn
广告&商务合作:bd@almosthuman.cn