
摘要
在网络中的节点和边进行预测任务时,需要仔细设计用于学习算法的特征。近年来,表示学习领域的研究取得了显著进展,通过自动学习特征本身来简化预测过程。然而,现有的特征学习方法在捕捉网络中观察到的连接模式多样性方面还不够充分。本文提出了一种名为node2vec的算法框架,用于学习网络中节点的连续特征表示。在node2vec中,我们学习一种将节点映射到低维特征空间的方法,该方法最大化保留节点网络邻域的可能性。我们定义了一个灵活的节点网络邻域概念,并设计了一种有偏随机游走过程,能够高效地探索多样化的邻域。我们的算法推广了基于固定网络邻域概念的先前工作,并认为在探索邻域时增加灵活性是学习更丰富表示的关键。我们在多个来自不同领域的实际网络上进行了多标签分类和链接预测实验,结果表明node2vec优于现有的最先进技术。综上所述,我们的工作为在复杂网络中高效地学习最先进的任务无关表示提供了一种新的方法。
代码仓库
Robotmurlock/Deepwalk-and-Node2vec
pytorch
GitHub 中提及
urielsinger/ctdne
GitHub 中提及
olety/TIMLinUCB
GitHub 中提及
cvg/scrstudio
jax
GitHub 中提及
thibaudmartinez/node2vec
GitHub 中提及
TheJacksonLaboratory/N2V
tf
GitHub 中提及
Nina-Konovalova/TSP-RL-Skoltech_project
pytorch
GitHub 中提及
razrLeLe/fastwalk
GitHub 中提及
icd-codex/icd-codex
GitHub 中提及
monarch-initiative/embiggen
tf
GitHub 中提及
rusty1s/pytorch_cluster
pytorch
GitHub 中提及
urielsinger/fairwalk
GitHub 中提及
eliorc/node2vec
GitHub 中提及
WiktorJ/msnode2vec
GitHub 中提及
基准测试
| 基准 | 方法 | 指标 |
|---|---|---|
| link-prediction-on-usair | N2V | AUC: 91.44 |
| link-property-prediction-on-ogbl-citation2 | Node2vec | Ext. data: No Number of params: 374911105 Test MRR: 0.6141 ± 0.0011 Validation MRR: 0.6124 ± 0.0011 |
| link-property-prediction-on-ogbl-collab | Node2vec | Ext. data: No Number of params: 30322945 Test Hits@50: 0.4888 ± 0.0054 Validation Hits@50: 0.5703 ± 0.0052 |
| link-property-prediction-on-ogbl-ddi | Node2vec | Ext. data: No Number of params: 645249 Test Hits@20: 0.2326 ± 0.0209 Validation Hits@20: 0.3292 ± 0.0121 |
| link-property-prediction-on-ogbl-ppa | Node2vec | Ext. data: No Number of params: 73878913 Test Hits@100: 0.2226 ± 0.0083 Validation Hits@100: 0.2253 ± 0.0088 |
| malware-detection-on-android-malware-dataset | node2vec | Accuracy: 81.25 |
| node-classification-on-blogcatalog | node2vec | Accuracy: 21.50% Macro-F1: 0.206 |
| node-classification-on-wikipedia | node2vec | Accuracy: 19.10% Macro-F1: 0.179 |