arrow
返回

Parallelizing and optimizing neural Encoder-Decoder models without padding on multi-core architecture

delete2020-07-01
delete4
PRE
AI
Y
Yuchen Qiao *
K
Kazuma Hashimoto
A
Akiko Eriguchi
王
王海霞 (Haixia Wang)
王
王东升 (Dongsheng Wang)
Y
Yoshimasa Tsuruoka
K
Kenjiro Taura
DOI:10.1016/j.future.2018.04.070delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
Scaling up Artificial Intelligence (AI) algorithms for massive datasets to improve their performance is becoming crucial. In Machine Translation (MT), one of most important research fields of AI, models based on Recurrent Neural Networks (RNN) show state-of-the-art performance in recent years, and many researchers keep working on improving RNN-based models to achieve better accuracy in translation tasks. Most implementations of Neural Machine Translation (NMT) models employ a padding strategy when processing a mini-batch to make all sentences in a mini-batch have the same length. This enables an efficient utilization of caches and GPU/SIMD parallelism but leads to a waste of computation time. In this paper, we implement and parallelize batch learning for a Sequence-to-Sequence (Seq2Seq) model, which is the most basic model of NMT, without using a padding strategy. More specifically, our approach forms vectors which represent the input words as well as the neural network's states at different time steps into matrices when it processes one sentence, and as a result, the approach makes a better use of cache and optimizes the process that adjusts weights and biases during the back-propagation phase. Our experimental evaluation shows that our implementation achieves better scalability on multi-core CPUs. We also discuss our approach's potential to be used in other implementations of RNN-based models. (C) 2018 Elsevier B.V. All rights reserved.
Keyword:
Neural machine translation
Cache optimization
Parallel programming
AI总结

AI总结

对已上传原文的论文进行重点信息的提取,主要内容包括:简要概述、研究摘要、背景介绍、关键亮点、图文解析、展望与总结。

期刊

F
Future Generation Computer Systems-The International Journal of eScience
IF:
6.1
论文数:
6.9K
被引数:
2.3W

机构

U
University of Tokyo
学者数:
7.1W
论文数: 6.5W
被引数: 2.2K
T
tsinghua university
学者数:
11.9W
论文数: 10.0W
被引数: 137
引用论文

引用论文