N3LDG:一种轻量级自然语言处理深度学习库

摘要：

　　提出一种用于自然语言处理的轻量级深度学习库N3LDG.N3LDG 可以支持动态地构建计算图,并能自动批量化执行计算图.实验显示,当训练卷积神经网络、双向LSTM 和树结构LSTM 时,N3LDG 都能高效地构建与执行计算图.当使用CPU 训练上述模型时,N3LDG 的训练速度优于PyTorch.当使用GPU 训练卷积神经网络和树结构LSTM 模型时,N3LDG 的训练速度优于PyTorch.

关键词： 深度学习库 自然语言处理 轻量级 CUDA

作者: 王潜升 余南 张梅山 韩子嘉 付国宏

作者单位: 黑龙江大学计算机科学技术学院 哈尔滨 150080

会议类型: 国际会议

会议名称: 2018自然语言处理与中文计算国际会议(NLPCC2018)

会议地点: 呼和浩特

会议语种:中文

页码: 1-8

在线出版日期: 2018-08-26（万方平台首次上网日期，不代表论文的发表时间）