2 个月前

node2vec:网络的可扩展特征学习方法

Aditya Grover; Jure Leskovec
node2vec:网络的可扩展特征学习方法
摘要

在网络中的节点和边进行预测任务时,需要仔细设计用于学习算法的特征。近年来,表示学习领域的研究取得了显著进展,通过自动学习特征本身来简化预测过程。然而,现有的特征学习方法在捕捉网络中观察到的连接模式多样性方面还不够充分。本文提出了一种名为node2vec的算法框架,用于学习网络中节点的连续特征表示。在node2vec中,我们学习一种将节点映射到低维特征空间的方法,该方法最大化保留节点网络邻域的可能性。我们定义了一个灵活的节点网络邻域概念,并设计了一种有偏随机游走过程,能够高效地探索多样化的邻域。我们的算法推广了基于固定网络邻域概念的先前工作,并认为在探索邻域时增加灵活性是学习更丰富表示的关键。我们在多个来自不同领域的实际网络上进行了多标签分类和链接预测实验,结果表明node2vec优于现有的最先进技术。综上所述,我们的工作为在复杂网络中高效地学习最先进的任务无关表示提供了一种新的方法。