通过数据驱动的节点采样提高GraphSAGE
作者: ( , ), ( ), ( )
发表在2019 ICLR研讨会表示学习上图和流形。
概述
作为一种高效且可扩展的图神经网络,GraphSAGE通过归纳二次采样的局部邻域并以小批量梯度下降的方式进行学习,已启用了归纳能力来推断看不见的节点或图。 GraphSAGE中使用的邻域采样有效地提高了并行推断一批不同程度的目标节点时的计算和存储效率。 尽管有此优势,但默认的统一采样在训练和推理上仍存在较大差异,从而导致次优准确性。 我们提出了一种新的数据驱动的采样方法,以通过非线性回归来推断邻域的实际值重要性,并使用该值作为对邻域进行二次采样的标准。 使用基于值的强化学习来学习回归者。 从GraphSAGE的负分类损失输出中归纳地提取了顶点和邻域的每种组合的隐含重要性。 结果,在使用三个数据集的归纳节点分类基准中,我们的方法使用统一
2022-04-16 22:59:33
7.29MB
Python
1