Auto Byte

专注未来出行及智能汽车科技

微信扫一扫获取更多资讯

Science AI

关注人工智能与其他前沿技术、基础学科的交叉研究与融合发展

微信扫一扫获取更多资讯

基于MXNet 的神经机器翻译实现

本篇内容是使用 MXNet 对带有注意机制的神经机器翻译的实现。


项目地址https://github.com/magic282/MXNMT


数据


目前的代码使用 IWSLT 2009 中英语料库作为训练、开发与测试数据。请使用这一数据集或其他可用的并行语料库。数据统计如下:

QQ截图20161221140733.png

开发/测试数据格式


IWSLT 2009 中英数据库的参考数(reference number)是 7,例如:

QQ截图20161221140815.png

结果


根据我自己的测试,在没有 53 次迭代后的后期处理情况下,这些代码能在 IWSLT 开发数据集上达到 44.18 BLEU 得分。特别是 1gram=72.65% 2gram=49.63% 3gram=37.62% 4gram=28.08% BP = 1.0000 BLEU = 0.4418。


问题周知


  • 兼容性问题。目前的版本会要求使用 Python 3,因为使用 Python2 处理中文代码问题非常恼人。

  • 在注意(attention)部分,h.dot(U) 要提前计算(pre-computed),然而我这么做了看起来效果也不是很好。

  • BLEU 评估器是个 exe 文件也不是内藏的,以后需要用 nltk 评估器取代。

  • 该模型需要修改,从而在此数据集上能够达到 50 BLEU 得分。

入门机器翻译MXNet开源工程实现NMT
暂无评论
暂无评论~