大规模图上的归纳式表示学习

Inductive Representation Learning on Large Graphs

威廉·汉密尔顿 William L. Hamilton · Stanford · 2017-06-07 · arXiv:1706.02216 ↗ · 被引 20182

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

摘要 · Abstract

在大规模图中,节点的低维嵌入在从内容推荐到蛋白质功能识别等多种预测任务中已被证明极为有用。然而,大多数现有方法要求图中的所有节点在嵌入训练期间都存在;这些先前的方法本质上是直推式的,不能自然地泛化到未见节点。本文提出了 GraphSAGE,一个通用的归纳式框架,它利用节点特征信息(如文本属性)来高效地为未见数据生成节点嵌入。我们不是为每个节点训练单独的嵌入,而是学习一个函数,通过从节点的局部邻域采样并聚合特征来生成嵌入。我们的算法在三个归纳式节点分类基准上优于强基线:我们基于引文和 Reddit 帖子数据对演化信息图中的未见节点类别进行分类,并且我们使用一个蛋白质-蛋白质相互作用的多图数据集展示了我们的算法能够泛化到完全未见过的图。

Low-dimensional embeddings of nodes in large graphs have proved extremely useful in a variety of prediction tasks, from content recommendation to identifying protein functions. However, most existing approaches require that all nodes in the graph are present during training of the embeddings; these previous approaches are inherently transductive and do not naturally generalize to unseen nodes. Here we present GraphSAGE, a general, inductive framework that leverages node feature information (e.g., text attributes) to efficiently generate node embeddings for previously unseen data. Instead of training individual embeddings for each node, we learn a function that generates embeddings by sampling and aggregating features from a node's local neighborhood. Our algorithm outperforms strong baselines on three inductive node-classification benchmarks: we classify the category of unseen nodes in evolving information graphs based on citation and Reddit post data, and we show that our algorithm generalizes to completely unseen graphs using a multi-graph dataset of protein-protein interactions.

核心贡献 · Key contributions

局限 · Limitations

论文章节 · Sections(共 15)

阅读逐段中英对照全文 →