指针网络

Pointer Networks

奥里奥尔·维尼亚尔斯 Oriol Vinyals · Google · 2015-06-09 · arXiv:1506.03134 ↗ · 被引 3496

打开互动全文版(逐段中英对照 + 图/公式 + 论文问答)→

摘要 · Abstract

我们提出了一种新的神经架构,用于学习输出序列的条件概率,其中输出元素是离散的标记,对应于输入序列中的位置。这类问题无法通过现有的序列到序列模型或神经图灵机直接解决,因为每一步输出的目标类别数量取决于输入的长度,而输入长度是可变的。排序可变长度序列以及各种组合优化问题都属于这一类。我们的模型利用最近提出的神经注意力机制解决了可变大小输出字典的问题。与之前的注意力机制不同,它不是在解码每一步将编码器的隐藏单元混合成一个上下文向量,而是将注意力用作指针,选择输入序列中的一个成员作为输出。我们将这种架构称为指针网络(Ptr-Net)。我们展示了 Ptr-Nets 可以仅通过训练示例学习解决三个具有挑战性的几何问题的近似解——寻找平面凸包、计算德劳内三角剖分以及平面旅行商问题。Ptr-Nets 不仅改进了带有输入注意力的序列到序列模型,还允许我们推广到可变大小的输出字典。我们展示了学习模型能够泛化到超过其训练最大长度的序列。我们希望在这些任务上的结果能够鼓励对离散问题的神经学习的更广泛探索。

We introduce a new neural architecture to learn the conditional probability of an output sequence with elements that are discrete tokens corresponding to positions in an input sequence. Such problems cannot be trivially addressed by existent approaches such as sequence-to-sequence and Neural Turing Machines, because the number of target classes in each step of the output depends on the length of the input, which is variable. Problems such as sorting variable sized sequences, and various combinatorial optimization problems belong to this class. Our model solves the problem of variable size output dictionaries using a recently proposed mechanism of neural attention. It differs from the previous attention attempts in that, instead of using attention to blend hidden units of an encoder to a context vector at each decoder step, it uses attention as a pointer to select a member of the input sequence as the output. We call this architecture a Pointer Net (Ptr-Net). We show Ptr-Nets can be used to learn approximate solutions to three challenging geometric problems -- finding planar convex hulls, computing Delaunay triangulations, and the planar Travelling Salesman Problem -- using training examples alone. Ptr-Nets not only improve over sequence-to-sequence with input attention, but also allow us to generalize to variable size output dictionaries. We show that the learnt models generalize beyond the maximum lengths they were trained on. We hope our results on these tasks will encourage a broader exploration of neural learning for discrete problems.

核心贡献 · Key contributions

局限 · Limitations

论文章节 · Sections(共 15)

阅读逐段中英对照全文 →