一文读懂Attention:Facebook曾拿CNN秒杀谷歌,现如今谷歌拿它秒杀所有人

不久前,Google基于Attention机制的机器翻译研究“Attention is All You Need”赚足了眼球。它放弃了机器翻译上常用的CNN、RNN架构,仅采用Attention机制就拿到了业界BLEU评分上的最佳成绩,且训练过程仅需3天半就能完成。当然,这是用了8颗P100 GPU的结果。 于是,人们不免好奇,Attention到底是个什么鬼?它相对于传统LSTM的技术优势在哪里?它所注意的是什么,究竟能解决CNN、RNN所解决不了的什么难题?要知道,Facebook此前刚用CNN模型取得了9倍于Google过去的RNN模型的成绩。 一文看懂Attention机制,你想知道的都在这里了。Enjoy!

作者 | Jason Brownlee

翻译 | AI科技大本营

参与 | JeyZhang,波波

目前采用编码器-解码器 (Encode-Decode) 结构的模型非常热门,是因为它在许多领域较其他的传统模型方法都取得了更好的结果。

这种结构的模型通常将输入序列编码成一个固定长度的向量表示,对于长度较短的输入序列而言,该模型能够学习出对应合理的向量表示。然而,这种模型存在的问题在于:当输入序列非常长时,模型难以学到合理的向量表示。

在这篇博文中,我们将探索加入LSTM/RNN模型中的Attention机制是如何克服传统编码器-解码器结构存在的问题的。

阅读本文,你将学到:

  • 传统编码器-解码器结构存在的问题及如何将输入序列编码成固定的向量表示;
  • Attention机制是如何克服上述问题的,以及在模型输出时是如何考虑输出与输入序列的每一项关系的;
  • 基于Attention机制的LSTM/RNN模型的5个应用领域:机器翻译、图片描述、语义蕴涵、语音识别和文本摘要。

让我们开始吧。

LSTM/RNN模型的Attention机制,图片来自Jonas Schleske

长输入序列带来的问题

使用传统编码器-解码器的RNN模型先用一些LSTM单元来对输入序列进行学习,编码为固定长度的向量表示;然后再用一些LSTM单元来读取这种向量表示并解码为输出序列。

采用这种结构的模型在许多比较难的序列预测问题(如文本翻译)上都取得了最好的结果,因此迅速成为了目前的主流方法。

例如:

Sequence to Sequence Learning with Neural Networks, 2014 https://arxiv.org/abs/1409.3215 Learning Phrase Representations using RNN Encoder-Decoder for Statistical Machine Translation, 2014 https://arxiv.org/abs/1406.1078

这种结构在很多其他的领域上也取得了不错的结果。然而,它存在的一个问题在于:输入序列不论长短都会被编码成一个固定长度的向量表示,而解码则受限于该固定长度的向量表示

这个问题限制了模型的性能,尤其是当输入序列比较长时,模型的性能会变得很差(在文本翻译任务上表现为待翻译的原始文本长度过长时翻译质量较差)。

“一个潜在的问题是,采用编码器-解码器结构的神经网络模型需要将输入序列中的必要信息表示为一个固定长度的向量,而当输入序列很长时则难以保留全部的必要信息(因为太多),尤其是当输入序列的长度比训练数据集中的更长时。”

via:Dzmitry Bahdanau, et al., Neural machine translation by jointly learning to align and translate, 2015 - https://arxiv.org/abs/1409.0473

序列中的Attention

Attention机制的基本思想是,打破了传统编码器-解码器结构在编解码时都依赖于内部一个固定长度向量的限制

Attention机制的实现是通过保留LSTM编码器对输入序列的中间输出结果,然后训练一个模型来对这些输入进行选择性的学习并且在模型输出时将输出序列与之进行关联。

换一个角度而言,输出序列中的每一项的生成概率取决于在输入序列中选择了哪些项。

“在文本翻译任务上,使用Attention机制的模型每生成一个词时都会在输入序列中找出一个与之最相关的词集合。之后模型根据当前的上下文向量 (context vectors) 和所有之前生成出的词来预测下一个目标词。

… 它将输入序列转化为一堆向量的序列并自适应地从中选择一个子集来解码出目标翻译文本。这感觉上像是用于文本翻译的神经网络模型需要“压缩”输入文本中的所有信息为一个固定长度的向量,不论输入文本的长短。”

via:Dzmitry Bahdanau, et al., Neural machine translation by jointly learning to align and translate, 2015

虽然模型使用Attention机制之后会增加计算量,但是性能水平能够得到提升。

另外,使用Attention机制便于理解在模型输出过程中输入序列中的信息是如何影响最后生成序列的。这有助于我们更好地理解模型的内部运作机制以及对一些特定的输入-输出进行debug。

“论文提出的方法能够直观地观察到生成序列中的每个词与输入序列中一些词的对齐关系,这可以通过对标注 (annotations) 权重参数可视化来实现…每个图中矩阵的每一行表示与标注相关联的权重。由此我们可以看出在生成目标词时,源句子中的位置信息会被认为更重要。”

via:Dzmitry Bahdanau, et al., Neural machine translation by jointly learning to align and translate, 2015

大型图片带来的问题

被广泛应用于计算机视觉领域的卷积神经网络模型同样存在类似的问题: 对于特别大的图片输入,模型学习起来比较困难。

由此,一种启发式的方法是将在模型做预测之前先对大型图片进行某种近似的表示。

“人类的感知有一个重要的特性是不会立即处理外界的全部输入,相反的,人类会将Attention专注于所选择的部分来得到所需要的信息,然后结合不同时间段的局部信息来建立一个内部的场景表示,从而引导眼球的移动及做出决策。”

via: Recurrent Models of Visual Attention, 2014 - https://arxiv.org/abs/1406.6247

这种启发式方法某种程度上也可以认为是考虑了Attention,但在这篇博文中,这种方法并不是基于Attention机制的。

基于Attention机制的相关论文如下:

  • Recurrent Models of Visual Attention, 2014 - https://arxiv.org/abs/1406.6247
  • DRAW: A Recurrent Neural Network For Image Generation, 2014 - https://arxiv.org/abs/1502.04623
  • Multiple Object Recognition with Visual Attention, 2014 - https://arxiv.org/abs/1412.7755

Attention在序列预测中的5大应用

这部分将列举几个具体的应用实例,介绍Attention机制是如何用在LSTM/RNN模型来进行序列预测的。

1. Attention在文本翻译任务上的应用

文本翻译这个实例在前面已经提过了。

给定一个法语的句子作为输入序列,需要输出翻译为英语的句子。Attention机制被用在输出输出序列中的每个词时会专注考虑输入序列中的一些被认为比较重要的词。

我们对原始的编码器-解码器模型进行了改进,使其有一个模型来对输入内容进行搜索,也就是说在生成目标词时会有一个编码器来做这个事情。这打破了之前的模型是基于将整个输入序列强行编码为一个固定长度向量的限制,同时也让模型在生成下一个目标词时重点考虑输入中相关的信息。

via:Dzmitry Bahdanau, et al., Neural machine translation by jointly learning to align and translate, 2015

Attention在文本翻译任务(输入为法语文本序列,输出为英语文本序列)上的可视化

(图片来自论文Dzmitry Bahdanau, et al., Neural machine translation by jointly learning to align and translate, 2015)

2. Attention在图片描述上的应用

与之前启发式方法不同的是,基于序列生成的Attention机制可以应用在计算机视觉相关的任务上,帮助卷积神经网络重点关注图片的一些局部信息来生成相应的序列,典型的任务就是对一张图片进行文本描述。

给定一张图片作为输入,输出对应的英文文本描述。Attention机制被用在输出输出序列的每个词时会专注考虑图片中不同的局部信息。

我们提出了一种基于Attention的方法,该方法在3个标准数据集上都取得了最佳的结果……同时展现了Attention机制能够更好地帮助我们理解模型地生成过程,模型学习到的对齐关系与人类的直观认知非常的接近(如下图)。

Attention在图片描述任务(输入为图片,输出为描述的文本)上的可视化

(图片来自论文Attend and Tell: Neural Image Caption Generation with Visual Attention, 2016)— Show, Attend and Tell: Neural Image Caption Generation with Visual Attention, 2016 - https://arxiv.org/abs/1502.03044

3. Attention在语义蕴涵 (Entailment) 中的应用

给定一个用英文描述的前提和假设作为输入,输出假设与前提是否矛盾、是否相关或者是否成立。

举个例子:

前提:在一个婚礼派对上拍照

假设:有人结婚了

该例子中的假设是成立的。

Attention机制被用于关联假设和前提描述文本之间词与词的关系。

我们提出了一种基于LSTM的神经网络模型,和把每个输入文本都独立编码为一个语义向量的模型不同的是,该模型同时读取前提和假设两个描述的文本序列并判断假设是否成立。我们在模型中加入了Attention机制来找出假设和前提文本中词/短语之间的对齐关系。……加入Attention机制能够使模型在实验结果上有2.6个点的提升,这是目前数据集上取得的最好结果…

via:Reasoning about Entailment with Neural Attention, 2016 - https://arxiv.org/abs/1509.06664

Attention在语义蕴涵任务(输入是前提文本,输出是假设文本)上的可视化(图片来自论文Reasoning about Entailment with Neural Attention, 2016)

4. Attention在语音识别上的应用

给定一个英文的语音片段作为输入,输出对应的音素序列。

Attention机制被用于对输出序列的每个音素和输入语音序列中一些特定帧进行关联。

一种基于Attention机制的端到端可训练的语音识别模型,能够结合文本内容和位置信息来选择输入序列中下一个进行编码的位置。该模型有一个优点是能够识别长度比训练数据长得多的语音输入。

via:Attention-Based Models for Speech Recognition, 2015. - https://arxiv.org/abs/1506.07503

Attention在语音识别任务(输入是音帧,输出是音素的位置)上的可视化(图片来自论文Attention-Based Models for Speech Recognition, 2015)

5. Attention在文本摘要上的应用

给定一篇英文文章作为输入序列,输出一个对应的摘要序列。

Attention机制被用于关联输出摘要中的每个词和输入中的一些特定词。

在最近神经网络翻译模型的发展基础之上,提出了一个用于生成摘要任务的基于Attention的神经网络模型。通过将这个概率模型与一个生成式方法相结合来生成出准确的摘要。

via:A Neural Attention Model for Abstractive Sentence Summarization, 2015 - https://arxiv.org/abs/1509.00685

Attention在文本摘要任务(输入为文章,输出为文本摘要)上的可视化(图片来自论文A Neural Attention Model for Abstractive Sentence Summarization, 2015)

进一步阅读

如果你想进一步地学习如何在LSTM/RNN模型中加入Attention机制,可阅读以下论文:

Attention and memory in deep learning and NLP http://www.wildml.com/2016/01/attention-and-memory-in-deep-learning-and-nlp/ Attention Mechanism https://blog.heuritech.com/2016/01/20/attention-mechanism/ Survey on Attention-based Models Applied in NLP http://yanran.li/peppypapers/2015/10/07/survey-attention-model-1.html What is exactly the attention mechanism introduced to RNN https://www.quora.com/What-is-exactly-the-attention-mechanism-introduced-to-RNN-recurrent-neural-network-It-would-be-nice-if-you-could-make-it-easy-to-understand(Quora问答) What is Attention Mechanism in Neural Networks? https://www.quora.com/What-is-Attention-Mechanism-in-Neural-Networks

目前Keras官方还没有单独将Attention模型的代码开源,下面有一些第三方的实现:

Deep Language Modeling for Question Answering using Keras http://ben.bolte.cc/blog/2016/language.html Attention Model Available! https://github.com/fchollet/keras/issues/2067 Keras Attention Mechanism https://github.com/philipperemy/keras-attention-mechanism Attention and Augmented Recurrent Neural Networks http://distill.pub/2016/augmented-rnns/ How to add Attention on top of a Recurrent Layer (Text Classification) https://github.com/fchollet/keras/issues/4962 Attention Mechanism Implementation Issue https://github.com/fchollet/keras/issues/1472 Implementing simple neural attention model (for padded inputs) https://github.com/fchollet/keras/issues/2612 Attention layer requires another PR https://github.com/fchollet/keras/issues/1094 seq2seq library https://github.com/farizrahman4u/seq2seq

总结

通过这篇博文,你应该学习到了Attention机制是如何应用在LSTM/RNN模型中来解决序列预测存在的问题。

具体来说,是这三个问题:

  1. 采用传统编码器-解码器结构的LSTM/RNN模型存在一个问题,不论输入长短都将其编码成一个固定长度的向量表示,这使模型对于长输入序列的学习效果很差(解码效果很差)。
  2. 而Attention机制则克服了上述问题,原理是在模型输出时会选择性地专注考虑输入中的对应相关的信息。
  3. 使用Attention机制的方法被广泛应用在各种序列预测任务上,包括文本翻译、语音识别等。

原文地址 http://machinelearningmastery.com/attention-long-short-term-memory-recurrent-neural-networks/

原文发布于微信公众号 - AI科技大本营(rgznai100)

原文发表时间:2017-07-03

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

相关文章

来自专栏PPV课数据科学社区

干货:基于树的建模-完整教程(R & Python)

来源:“数盟社区” 原文链接:http://dataunion.org/23697.html 简介 基于树的学习算法被认为是最好的方法之一,主要用于监测学习方...

3857
来自专栏人工智能头条

深度学习之神经网络与支持向量机

1543
来自专栏红色石头的机器学习之路

台湾大学林轩田机器学习基石课程学习笔记14 -- Regularization

上节课我们介绍了过拟合发生的原因:excessive power, stochastic/deterministic noise 和limited data。并...

2480
来自专栏郭耀华‘s Blog

NLP之——Word2Vec详解

2013年,Google开源了一款用于词向量计算的工具——word2vec,引起了工业界和学术界的关注。首先,word2vec可以在百万数量级的词典和上亿的数据...

2142
来自专栏AI研习社

如何使用高大上的方法调参数

本文主要介绍作者与 Elad Hazan, Adam Klivans 合作的最新论文: Hyperparameter Optimization: A Spec...

3649
来自专栏Spark学习技巧

【深度学习】⑤--自然语言处理的相关应用

恩恩,关注我好了 1. 自然语言处理(NLP)的常见任务 先来看看在工业界对于自然语言处理的一些常见需求与任务。 自动摘要 这个在搜索引擎中非常常用。指计算机...

3818
来自专栏老秦求学

[Deep-Learning-with-Python]神经网络入手学习[上]

网络层堆叠形成网络模型,网络模型由输入数据得到预测值。损失函数比较预测值与实际值,得到损失函数值:用来评估预测结果的好坏;优化方法用损失值来更新网络模型的权...

852
来自专栏机器之心

教程 | 将注意力机制引入RNN,解决5大应用领域的序列预测问题

3724
来自专栏深度学习思考者

文献 | 2010-2016年被引用次数最多的深度学习论文(修订版)

一、书籍 Deep learning (2015) 作者:Bengio 下载地址:http://www.deeplearningbook.org/ 二、理论 1...

2467
来自专栏机器之心

一文简述ResNet及其多种变体

选自TowardsDataScience 作者:Vincent Fung 机器之心编译 参与:邹俏也、路雪 本文主要介绍了 ResNet 架构,简要阐述了其近期...

2965

扫码关注云+社区

领取腾讯云代金券