arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1610.08613cs.LGcs.CL
收藏

主动记忆能取代注意力吗?

Can Active Memory Replace Attention?

Google Brain(谷歌大脑)

Łukasz Kaiser, Samy Bengio

摘要:

近年来,将神经网络的注意力集中在其输入或记忆的选定部分上的几种机制已成功地用于深度学习模型中。注意力改善了图像分类、图像描述、语音识别、生成模型和学习算法任务,但它可能对神经机器翻译产生了最大的影响。最近,使用替代机制已经获得了类似的改进,这些机制不关注存储器的单个部分,而是以统一的方式并行地对所有部分进行操作。这种机制,我们称之为主动记忆,改善了算法任务、图像处理和生成建模中的过度注意力。然而,到目前为止,对于大多数自然语言处理任务,特别是对于机器翻译,主动记忆并没有改善注意力。本文分析了这一缺点,并提出了一种扩展的主动记忆模型,该模型与神经机器翻译上的现有注意力模型相匹配,并且更好地推广到较长的句子。我们调查这个模型,并解释为什么以前的主动记忆模型没有成功。最后,我们讨论了主动记忆何时带来最大的好处,以及注意力在哪里可以成为更好的选择。

英文摘要:

Several mechanisms to focus attention of a neural network on selected parts of its input or memory have been used successfully in deep learning models in recent years. Attention has improved image classification, image captioning, speech recognition, generative models, and learning algorithmic tasks, but it had probably the largest impact on neural machine translation. Recently, similar improvements have been obtained using alternative mechanisms that do not focus on a single part of a memory but operate on all of it in parallel, in a uniform way. Such mechanism, which we call active memory, improved over attention in algorithmic tasks, image processing, and in generative modelling. So far, however, active memory has not improved over attention for most natural language processing tasks, in particular for machine translation. We analyze this shortcoming in this paper and propose an extended model of active memory that matches existing attention models on neural machine translation and generalizes better to longer sentences. We investigate this model and explain why previous active memory models did not succeed. Finally, we discuss when active memory brings most benefits and where attention can be a better choice.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑