中文

注意力池化网络

计算与语言 2016-02-12 v1 机器学习

摘要

在这项工作中,我们提出注意力池化(AP),一种用于判别模型训练的双向注意力机制。在使用神经网络进行成对排序或分类的背景下,AP使池化层能够感知当前的输入对,使得来自两个输入项的信息能够直接影响彼此表示的计算。连同这种成对输入的表示,AP联合学习对该输入对投影片段(例如三元语法)的相似度度量,并随后推导出每个输入对应的注意力向量以引导池化。我们的双向注意力机制是一个独立于底层表示学习的通用框架,并在我们的研究中应用于卷积神经网络(CNNs)和循环神经网络(RNNs)。来自问答/答案选择的三个差异很大的基准任务的实证结果表明,我们提出的模型优于多种强基线,并在所有基准中取得了最先进的性能。

关键词

引用

@article{arxiv.1602.03609,
  title  = {Attentive Pooling Networks},
  author = {Cicero dos Santos and Ming Tan and Bing Xiang and Bowen Zhou},
  journal= {arXiv preprint arXiv:1602.03609},
  year   = {2016}
}

备注

10 pages